Kratos战神微服务框架（二）

项目结构

api编写

protobuf编写

syntax = "proto3";
package realworld.v1;
import "google/api/annotations.proto";
option go_package = "realworld/api/realworld/v1;v1";// The greeting service definition.
service RealWorld {rpc Login(LoginRequest) returns (UserReply) {option (google.api.http) = {post: "/api/users/login",body: "*", // 注意post请求一定交加body项};}}message LoginRequest{message User {string email = 1;string password = 2;}User user = 1;
}message UserReply {message User {string email = 1;string token = 2;string username = 3;string bio = 4;string image = 5;}User user = 1;
}

使用makefile

注：需要安装make指令

GOPATH:=$(shell go env GOPATH)
VERSION=$(shell git describe --tags --always)
INTERNAL_PROTO_FILES=$(shell find internal -name *.proto)
API_PROTO_FILES=$(shell find api -name *.proto).PHONY: init
# init env
init:go install google.golang.org/protobuf/cmd/protoc-gen-go@latestgo install google.golang.org/grpc/cmd/protoc-gen-go-grpc@latestgo install github.com/go-kratos/kratos/cmd/kratos/v2@latestgo install github.com/go-kratos/kratos/cmd/protoc-gen-go-http/v2@latestgo install github.com/google/gnostic/cmd/protoc-gen-openapi@latest.PHONY: config
# generate internal proto
config:protoc --proto_path=./internal \--proto_path=./third_party \--go_out=paths=source_relative:./internal \./internal/conf/*.proto
#         $(INTERNAL_PROTO_FILES).PHONY: api
# generate api proto
api:protoc --proto_path=./api \--proto_path=./third_party \--go_out=paths=source_relative:./api \--go-http_out=paths=source_relative:./api \--go-grpc_out=paths=source_relative:./api \--openapi_out=fq_schema_naming=true,default_response=false:. \./api/realworld/v1/*.proto
#         $(API_PROTO_FILES).PHONY: build
# build
build:mkdir -p bin/ && go build -ldflags "-X main.Version=$(VERSION)" -o ./bin/ ./....PHONY: generate
# generate
generate:go mod tidygo get github.com/google/wire/cmd/wire@latestgo generate ./....PHONY: wire
# wire
wire:cd cmd/realworld/ && wire.PHONY: run
# run
run:kratos run.PHONY: all
# generate all
all:make api;make config;make generate;# show help
help:@echo ''@echo 'Usage:'@echo ' make [target]'@echo ''@echo 'Targets:'@awk '/^[a-zA-Z\-\_0-9]+:/ { \helpMessage = match(lastLine, /^# (.*)/); \if (helpMessage) { \helpCommand = substr($$1, 0, index($$1, ":")-1); \helpMessage = substr(lastLine, RSTART + 2, RLENGTH); \printf "\033[36m%-22s\033[0m %s\n", helpCommand,helpMessage; \} \} \{ lastLine = $$0 }' $(MAKEFILE_LIST).DEFAULT_GOAL := help

注：默认是linux命令,windwos需要修改路径

service层接口实现

进入internal/service目录

service.go

package serviceimport ("github.com/go-kratos/kratos/v2/log""github.com/google/wire"v1 "helloworld/api/realworld/v1""helloworld/internal/biz"
)// ProviderSet is service providers.
var ProviderSet = wire.NewSet(NewRealWorldService) //依赖注入type RealWorldService struct {v1.UnimplementedRealWorldServerpu  *biz.SocialUsecaseuc  *biz.UserUsecaselog *log.Helper
}func NewRealWorldService(uc *biz.UserUsecase, logger log.Logger) *RealWorldService {return &RealWorldService{uc: uc, log: log.NewHelper(logger)}
}//实现方法
func (s *RealWorldService) Login(ctx context.Context, req *v1.LoginRequest) (*v1.UserReply, error) {return &v1.UserReply{User: &v1.UserReply_User{Username: "jtyyds",},}, nil
}

biz层

package bizimport "github.com/google/wire"// ProviderSet is biz providers.
var ProviderSet = wire.NewSet(NewSocialUsecase, NewUserUsecase) //依赖注入type User struct {Email        stringUsername     stringBio          stringImage        stringPasswordHash string
}type UserLogin struct {Email    stringUsername stringToken    stringBio      stringImage    string
}// bcrypt方法加密
func hashPassword(pwd string) string {password, err := bcrypt.GenerateFromPassword([]byte(pwd), bcrypt.DefaultCost)if err != nil {panic(err)}fmt.Printf("%v", password)return string(password)
}
// 密码进行比较
func verifyPassword(hashed, input string) bool {err := bcrypt.CompareHashAndPassword([]byte(hashed), []byte(input))if err != nil {return false}return true
}// 接口方法
type UserRepo interface {CreateUser(ctx context.Context, user *User) errorGetUserByEmail(ctx context.Context, email string) (*User, error)
}type ProfileRepo interface {
}type UserUsecase struct {ur   UserRepopr   ProfileRepojwtc *conf.JWTlog  *log.Helper
}func NewUserUsecase(ur UserRepo, pr ProfileRepo, jwtc *conf.JWT, logger log.Logger) *UserUsecase {return &UserUsecase{ur: ur, pr: pr, jwtc: jwtc, log: log.NewHelper(logger)}
}func (uc *UserUsecase) generateToken(username string) string {return auth.GenerateToken(uc.jwtc.Token, username)
}// 注册的业务逻辑
func (uu *UserUsecase) Register(ctx context.Context, username, email, password string) (*UserLogin, error) {u := &User{Email:        email,Username:     username,PasswordHash: hashPassword(password),}if err := uu.ur.CreateUser(ctx, u); err != nil {return nil, err}return &UserLogin{Email:    email,Username: username,Token:    uu.generateToken(username),}, nil
}// 登录的业务逻辑实现
func (uu *UserUsecase) Login(ctx context.Context, email, password string) (*UserLogin, error) {u, err := uu.ur.GetUserByEmail(ctx, email)if err != nil {return nil, err}b := verifyPassword(u.PasswordHash, password)if b == true {return nil, errors.New("Loin")}return &UserLogin{Email:    u.Email,Username: u.Username,Bio:      u.Bio,Image:    u.Image,Token:    uu.generateToken(u.Username),}, nil
}

data层

data.go

package dataimport ("fmt""github.com/go-kratos/kratos/v2/log""github.com/google/wire""gorm.io/driver/mysql""gorm.io/gorm""helloworld/internal/conf"
)// ProviderSet is data providers.
var ProviderSet = wire.NewSet(NewData, NewDB, NewUserRepo, NewProfileRepo) //依赖注入// Data .
type Data struct {// TODO wrapped database clientdb *gorm.DB
}// NewData .
func NewData(c *conf.Data, logger log.Logger, db *gorm.DB) (*Data, func(), error) {cleanup := func() {log.NewHelper(logger).Info("closing the data resources")}return &Data{db: db}, cleanup, nil
}// 连接mysql数据库
func NewDB(c *conf.Data) *gorm.DB {fmt.Println(c.Database.Dsn)db, err := gorm.Open(mysql.Open(c.Database.Dsn), &gorm.Config{})if err != nil {panic("failed")}if err := db.AutoMigrate(); err != nil {panic(err)}return db
}

package dataimport ("context""github.com/go-kratos/kratos/v2/log""helloworld/internal/biz"
)type userRepo struct {data *Datalog  *log.Helper
}func NewUserRepo(data *Data, logger log.Logger) biz.UserRepo {return &userRepo{data: data,log:  log.NewHelper(logger),}
}func (r *userRepo) CreateUser(ctx context.Context, g *biz.User) error { // 可以实现操作数据库，这个并没有实现return nil
}func (r *userRepo) GetUserByEmail(ctx context.Context, email string) (*biz.User, error) {// 可以实现操作数据库，这个并没有实现return nil, nil
}

configs配置文件

server:http:addr: 0.0.0.0:8000timeout: 1sgrpc:addr: 0.0.0.0:9000timeout: 1sdata:database:driver: mysqldsn: "root:123456@tcp(127.0.0.1:3306)/realworld?charset=utf8mb4&parseTime=True&loc=Local"jwt:secret : "hello"

修改配置文件后，需要修改internal/conf下的conf.proto

syntax = "proto3";
package kratos.api;option go_package = "helloworld/internal/conf;conf";import "google/protobuf/duration.proto";message Bootstrap {Server server = 1;Data data = 2;JWT jwt = 3;
}message Server {message HTTP {string network = 1;string addr = 2;google.protobuf.Duration timeout = 3;}message GRPC {string network = 1;string addr = 2;google.protobuf.Duration timeout = 3;}HTTP http = 1;GRPC grpc = 2;
}message Data {message Database {string driver = 1;string dsn = 2;}Database database = 1;
}
message JWT {string token = 1;
}

使用make config生成conf.pb.go

注：是否需要修改config的makefile