go-zero 微服务应用实践（二）

服务调用

完成rpc访问功能后，还需要进行客户端调用，会用到goctl 的api功能；创建api目录编写api文件；api相关语法可以参考官方文档：api语法介绍

1、API文件编写

其中 登录注册增加了参数验证，使用 validator 包进行验证详细使用方法可以到 github.com/go-playground/validator 查看

syntax = "v1"
type (
	LoginRequest {
		Email    string `json:"Email" validate:"required,email"`
		Password string `json:"password" validate:"required,gte=8"`
	}
	LoginResponse {
		AccessToken  string `json:"accessToken"`
		AccessExpire int64  `json:"accessExpire"`
	}
	RegisterRequest {
		Name     string `json:"name" validate:"required,gte=2,lte=20"`
		Gender   int64  `json:"gender" validate:"oneof=1 2"`
		Email    string `json:"Email" validate:"required,email"`
		Password string `json:"password" validate:"required,gte=8"`
	}
	RegisterResponse {
		Id     int64  `json:"id"`
		Name   string `json:"name"`
		Gender int64  `json:"gender"`
		Email  string `json:"Email"`
	}

	UserInfoRequest {
	}
	UserInfoResponse {
		Id     int64  `json:"id"`
		Name   string `json:"name"`
		Gender int64  `json:"gender"`
		Email  string `json:"Email"`
	}
)

service User {
	@handler Login
	post /api/user/login(LoginRequest) returns (LoginResponse)
	
	@handler Register
	post /api/user/register(RegisterRequest) returns (RegisterResponse)
}

@server(
	jwt: Auth
)
service User {
	@handler UserInfo
	post /api/user/userinfo(UserInfoRequest) returns (UserInfoResponse)
}

2、执行goctl命令

goctl api go -api ./api/user.api -dir ./api

执行后生成目录如下

api
├─etc
└─internal
    ├─config
    ├─handler
    ├─logic
    ├─svc
    └─types

3、Service 层增加 UserRpc

package svc

import (
	"github.com/zeromicro/go-zero/zrpc"
	"slowdown/user/api/internal/config"
	"slowdown/user/rpc/user"
)

type ServiceContext struct {
	Config config.Config

	UserRpc user.User
}

func NewServiceContext(c config.Config) *ServiceContext {
	return &ServiceContext{
		Config:  c,
		UserRpc: user.NewUser(zrpc.MustNewClient(c.UserRpc)),
	}
}

4、logic 业务逻辑层

注册逻辑

func (l *RegisterLogic) Register(req *types.RegisterRequest) (resp *types.RegisterResponse, err error) {
	validate := validator.New()
	err = validate.Struct(req)
	if err != nil {
		return nil, err
	}
	res, err := l.svcCtx.UserRpc.Register(l.ctx, &user.RegisterRequest{
		Name:     req.Name,
		Email:    req.Email,
		Gender:   req.Gender,
		Password: req.Password,
	})
	if err != nil {
		return nil, err
	}

	return &types.RegisterResponse{
		Id:     res.Id,
		Name:   res.Name,
		Gender: res.Gender,
		Email:  res.Email,
	}, nil
}

登录逻辑

func (l *LoginLogic) Login(req *types.LoginRequest) (resp *types.LoginResponse, err error) {
	res, err := l.svcCtx.UserRpc.Login(l.ctx, &user.LoginRequest{
		Email:    req.Email,
		Password: req.Password,
	})
	if err != nil {
		return nil, err
	}
	now := time.Now().Unix()

	accessToken, err := jwtx.GetToken(l.svcCtx.Config.Auth.AccessSecret, now, l.svcCtx.Config.Auth.AccessExpire, res.Id)
	if err != nil {
		return nil, err
	}
	return &types.LoginResponse{
		AccessExpire: l.svcCtx.Config.Auth.AccessExpire,
		AccessToken:  accessToken,
	}, nil
}

获取用户信息

func (l *UserInfoLogic) UserInfo() (resp *types.UserInfoResponse, err error) {
	uid, _ := l.ctx.Value("uid").(json.Number).Int64()
	res, err := l.svcCtx.UserRpc.UserInfo(l.ctx, &user.UserInfoRequest{
		Id: uid,
	})
	if err != nil {
		return nil, err
	}
	return &types.UserInfoResponse{
		Id:     res.Id,
		Name:   res.Name,
		Gender: res.Gender,
		Email:  res.Email,
	}, nil
}

5、启动服务测试

使用apifox进行测试