1 生成模型

(1) 由sql脚本生成go模型

  1. CREATE TABLE `user` (
  2.                         `id` bigint unsigned NOT NULL AUTO_INCREMENT,
  3.                         `name` varchar(255)  NOT NULL DEFAULT '' COMMENT '用户姓名',
  4.                         `gender` tinyint(3) unsigned NOT NULL DEFAULT '0' COMMENT '用户性别',
  5.                         `mobile` varchar(255)  NOT NULL DEFAULT '' COMMENT '用户电话',
  6.                         `password` varchar(255)  NOT NULL DEFAULT '' COMMENT '用户密码',
  7.                         `create_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
  8.                         `update_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
  9.                         PRIMARY KEY (`id`),
  10.                         UNIQUE KEY `idx_mobile_unique` (`mobile`)
  11. ) ENGINE=InnoDB  DEFAULT CHARSET=utf8mb4;

goctl model mysql ddl -src ./model/user.sql -dir ./model -c

(2) 由数据库生成go模型

goctl model mysql datasource —url “root:password@tcp(127.0.0.1:3306)/database” -dir ./model

2 api生成go文件

(1) 编写api文件

  1. type (
  2.     // 用户登录
  3.     LoginRequest {
  4.         Mobile   string `json:"mobile"`
  5.         Password string `json:"password"`
  6.     }
  7.     LoginResponse {
  8.         AccessToken  string `json:"accessToken"`
  9.         AccessExpire int64  `json:"accessExpire"`
  10.     }
  11.     // 用户登录
  13.     // 用户注册
  14.     RegisterRequest {
  15.         Name     string `json:"name"`
  16.         Gender   int64  `json:"gender"`
  17.         Mobile   string `json:"mobile"`
  18.         Password string `json:"password"`
  19.     }
  20.     RegisterResponse {
  21.         Id     int64  `json:"id"`
  22.         Name   string `json:"name"`
  23.         Gender int64  `json:"gender"`
  24.         Mobile string `json:"mobile"`
  25.     }
  26.     // 用户注册
  28.     // 用户信息
  29.     UserInfoResponse {
  30.         Id     int64  `json:"id"`
  31.         Name   string `json:"name"`
  32.         Gender int64  `json:"gender"`
  33.         Mobile string `json:"mobile"`
  34.     }
  35.     // 用户信息
  36. )
  38. service User {
  39.     @handler Login
  40.     post /api/user/login(LoginRequest) returns (LoginResponse)
  42.     @handler Register
  43.     post /api/user/register(RegisterRequest) returns (RegisterResponse)
  44. }
  46. @server(
  47.     jwt: Auth
  48. )
  49. service User {
  50.     @handler UserInfo
  51.     post /api/user/userinfo returns (UserInfoResponse)
  52. }

(2) 根据api生成go代码

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

3 proto生成go文件

(1) 编写proto文件

  1. syntax = "proto3";
  3. package userclient;
  5. option go_package = "./user";
  7. // 用户登录
  8. message LoginRequest {
  9.     string Mobile = 1;
  10.     string Password = 2;
  11. }
  12. message LoginResponse {
  13.     int64 Id = 1;
  14.     string Name = 2;
  15.     int64 Gender = 3;
  16.     string Mobile = 4;
  17. }
  18. // 用户登录
  20. // 用户注册
  21. message RegisterRequest {
  22.     string Name = 1;
  23.     int64 Gender = 2;
  24.     string Mobile = 3;
  25.     string Password = 4;
  26. }
  27. message RegisterResponse {
  28.     int64 Id = 1;
  29.     string Name = 2;
  30.     int64 Gender = 3;
  31.     string Mobile = 4;
  32. }
  33. // 用户注册
  35. // 用户信息
  36. message UserInfoRequest {
  37.     int64 Id = 1;
  38. }
  39. message UserInfoResponse {
  40.     int64 Id = 1;
  41.     string Name = 2;
  42.     int64 Gender = 3;
  43.     string Mobile = 4;
  44. }
  45. // 用户信息
  47. service User {
  48.     rpc Login(LoginRequest) returns(LoginResponse);
  49.     rpc Register(RegisterRequest) returns(RegisterResponse);
  50.     rpc UserInfo(UserInfoRequest) returns(UserInfoResponse);
  51. }

(2) 根据proto生成go代码

goctl rpc protoc user.proto —go_out=./ —go-grpc_out=./ —zrpc_out=./

4 编写rpc服务

(1) 修改配置文件

rpc/etc/user.yaml

  1. Name: user.rpc
  2. ListenOn: 0.0.0.0:9000
  4. Etcd:
  5.   Hosts:
  6.   - etcd:2379
  7.   Key: user.rpc
  9. Mysql:
  10.   DataSource: root:123456@tcp(mysql:3306)/mall?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai
  12. CacheRedis:
  13. - Host: redis:6379
  14.   Type: node

(2) 添加 model 依赖

vim rpc/internal/config/config.go 

  1. package config
  3. import (
  4.     "github.com/zeromicro/go-zero/core/stores/cache"
  5.     "github.com/zeromicro/go-zero/zrpc"
  6. )
  8. type Config struct {
  9.     zrpc.RpcServerConf
  11.     Mysql struct {
  12.         DataSource string
  13.     }
  15.     CacheRedis cache.CacheConf
  16. }

vim rpc/internal/svc/servicecontext.go

  1. package svc
  3. import (
  4.     "mall/service/user/model"
  5.     "mall/service/user/rpc/internal/config"
  7.     "github.com/zeromicro/go-zero/core/stores/sqlx"
  8. )
  10. type ServiceContext struct {
  11.     Config config.Config
  13.     UserModel model.UserModel
  14. }
  16. func NewServiceContext(c config.Config) *ServiceContext {
  17.     conn := sqlx.NewMysql(c.Mysql.DataSource)
  18.     return &ServiceContext{
  19.         Config:    c,
  20.         UserModel: model.NewUserModel(conn, c.CacheRedis),
  21.     }
  22. }

(3) 添加业务逻辑

1) 添加工具库

vim common/cryptx/crypt.go

  1. package cryptx
  3. import (
  4.   "fmt"
  6.   "golang.org/x/crypto/scrypt"
  7. )
  9. func PasswordEncrypt(salt, password string) string {
  10.   dk, _ := scrypt.Key([]byte(password), []byte(salt), 32768, 8, 1, 32)
  11.   return fmt.Sprintf("%x", string(dk))
  12. }

2) 添加配置信息

vim rpc/etc/user.yaml

  1. Name: user.rpc
  2. ListenOn: 0.0.0.0:9000
  4. ...
  6. Salt: HWVOFkGgPTryzICwd7qnJaZR9KQ2i8xe

vim rpc/internal/config/config.go

  1. package config
  3. import (
  4.   "github.com/zeromicro/go-zero/core/stores/cache"
  5.   "github.com/zeromicro/go-zero/zrpc"
  6. )
  8. type Config struct {
  9.   ...
  10.   Salt string
  11. }

3) 添加逻辑

vim rpc/internal/logic/registerlogic.go

  1. package logic
  3. import (
  4.   "context"
  6.   "mall/common/cryptx"
  7.   "mall/service/user/model"
  8.   "mall/service/user/rpc/internal/svc"
  9.   "mall/service/user/rpc/user"
  11.   "github.com/zeromicro/go-zero/core/logx"
  12.   "google.golang.org/grpc/status"
  13. )
  15. type RegisterLogic struct {
  16.   ctx    context.Context
  17.   svcCtx *svc.ServiceContext
  18.   logx.Logger
  19. }
  21. func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterLogic {
  22.   return &RegisterLogic{
  23.     ctx:    ctx,
  24.     svcCtx: svcCtx,
  25.     Logger: logx.WithContext(ctx),
  26.   }
  27. }
  29. func (l *RegisterLogic) Register(in *user.RegisterRequest) (*user.RegisterResponse, error) {
  30.   // 判断手机号是否已经注册
  31.   _, err := l.svcCtx.UserModel.FindOneByMobile(in.Mobile)
  32.   if err == nil {
  33.     return nil, status.Error(100, "该用户已存在")
  34.   }
  36.   if err == model.ErrNotFound {
  38.     newUser := model.User{
  39.       Name:     in.Name,
  40.       Gender:   in.Gender,
  41.       Mobile:   in.Mobile,
  42.       Password: cryptx.PasswordEncrypt(l.svcCtx.Config.Salt, in.Password),
  43.     }
  45.     res, err := l.svcCtx.UserModel.Insert(&newUser)
  46.     if err != nil {
  47.       return nil, status.Error(500, err.Error())
  48.     }
  50.     newUser.Id, err = res.LastInsertId()
  51.     if err != nil {
  52.       return nil, status.Error(500, err.Error())
  53.     }
  55.     return &user.RegisterResponse{
  56.       Id:     newUser.Id,
  57.       Name:   newUser.Name,
  58.       Gender: newUser.Gender,
  59.       Mobile: newUser.Mobile,
  60.     }, nil
  62.   }
  64.   return nil, status.Error(500, err.Error())
  65. }

6 编写api服务

(1) 修改配置文件

vim api/etc/user.yaml

  1. Name: User
  2. Host: 0.0.0.0
  3. Port: 8000
  5. ...
  7. Auth:
  8.   AccessSecret: uOvKLmVfztaXGpNYd4Z0I1SiT7MweJhl
  9.   AccessExpire: 86400
  11. UserRpc:
  12.   Etcd:
  13.     Hosts:
  14.     - etcd:2379
  15.     Key: user.rpc

(2) 添加 rpc依赖

vim api/internal/config/config.go

  1. package config
  3. import (
  4.     "github.com/zeromicro/go-zero/rest"
  5.     "github.com/zeromicro/go-zero/zrpc"
  6. )
  8. type Config struct {
  9.     rest.RestConf
  11.     Auth struct {
  12.         AccessSecret string
  13.         AccessExpire int64
  14.     }
  16.     UserRpc zrpc.RpcClientConf
  17. }

vim api/internal/svc/servicecontext.go

  1. package svc
  3. import (
  4.   "mall/service/user/api/internal/config"
  5.   "mall/service/user/rpc/userclient"
  7.   "github.com/zeromicro/go-zero/zrpc"
  8. )
  10. type ServiceContext struct {
  11.   Config config.Config
  13.   UserRpc userclient.User
  14. }
  16. func NewServiceContext(c config.Config) *ServiceContext {
  17.   return &ServiceContext{
  18.     Config:  c,
  19.     UserRpc: userclient.NewUser(zrpc.MustNewClient(c.UserRpc)),
  20.   }
  21. }

(3) 添加业务逻辑

vim api/internal/logic/registerlogic.go

  1. package logic
  3. import (
  4.   "context"
  6.   "mall/service/user/api/internal/svc"
  7.   "mall/service/user/api/internal/types"
  8.   "mall/service/user/rpc/userclient"
  10.   "github.com/zeromicro/go-zero/core/logx"
  11. )
  13. type RegisterLogic struct {
  14.   logx.Logger
  15.   ctx    context.Context
  16.   svcCtx *svc.ServiceContext
  17. }
  19. func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) RegisterLogic {
  20.   return RegisterLogic{
  21.     Logger: logx.WithContext(ctx),
  22.     ctx:    ctx,
  23.     svcCtx: svcCtx,
  24.   }
  25. }
  27. func (l *RegisterLogic) Register(req types.RegisterRequest) (resp *types.RegisterResponse, err error) {
  28.   res, err := l.svcCtx.UserRpc.Register(l.ctx, &userclient.RegisterRequest{
  29.     Name:     req.Name,
  30.     Gender:   req.Gender,
  31.     Mobile:   req.Mobile,
  32.     Password: req.Password,
  33.   })
  34.   if err != nil {
  35.     return nil, err
  36.   }
  38.   return &types.RegisterResponse{
  39.     Id:     res.Id,
  40.     Name:   res.Name,
  41.     Gender: res.Gender,
  42.     Mobile: res.Mobile,
  43.   }, nil
  44. }

7 启动服务

(1) 启动rpc服务

$ cd mall/service/user/rpc $ go run user.go -f etc/user.yaml Starting rpc server at 127.0.0.1:9000…

(2) 启动api服务

$ cd mall/service/user/api $ go run user.go -f etc/user.yaml Starting server at 0.0.0.0:8000…