go-zero 實戰（2）

go-zero 實戰（1）中，使用了go-zero 創建了order 和 user 兩個微服務。而order作為grpc的客戶端，user 作為grpc的服務端，打通了 order 到 user的調用。接下來，我們在user中，加入mysql組件。確保數據能夠寫到數據庫。

引入MySQL

1. 啟動mysql，創建數據庫 zero-mall

可以使用 DBeaver 工具，連接mysql，并創建zero-mall數據庫。
在這里插入圖片描述
并且執行如下腳本創建表：

use zero_mall;create table `user`(id bigint(0) not null auto_increment,name varchar(255) character set utf8mb4 COLLATE utf8mb4_general_ci not null,gender varchar(255) character set utf8mb4 COLLATE utf8mb4_general_ci not null,PRIMARY key (id) using btree
);

2. 在 user/internal 目錄下創建 model目錄，并創建user.sql

創建 user.sql 文件，并將上面腳本放入 user.sql 文件中。然后，在當前目錄下，執行：

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

這步操作，會生成操作數據庫相關的代碼。由于生成的代碼比較亂。我們在做數據庫連接的時候，會摘取部分代碼，按照自己的思路做數據庫相關操作。

在當前目錄下新建 user.go (user/internal/model/user.go)文件，把當前生成的 usermodel_gen.go 文件中的 User 結構體拿出來，放到user.go 文件中，并添加一個 TableName函數。然后，把生成的 usermodel.go、usermodel_gen.go、vars.go文件刪除。最終的 user.go 代碼如下：

package modeltype User struct {Id     int64  `db:"id"`Name   string `db:"name"`Gender string `db:"gender"`
}
// 返回表名
func (User) TableName() string {return "user"
}

當我們創建了model之后，就有User 實體，該實體映射數據庫的User表。接下來的我們需要創建數據庫的連接。

3. 在user下創建database目錄

創建database目錄，在該目錄下創建sqlx.go文件，主要為了使用 go-zero的orm框架。當然，這個目錄下也可以創建如 mongo、redis的連接。也可以使用不同的orm框架，如gorm等。

sqlx.go 文件

package databaseimport "github.com/zeromicro/go-zero/core/stores/sqlx"// we use go-zero sqlxtype DBConn struct {Conn sqlx.SqlConn
}func Connect(datasource string) *DBConn {return &DBConn{Conn: sqlx.NewMysql(datasource),}
}

4. 創建操作數據的接口，并提供實現

在 user/internal/ 下創建 repo 目錄，并創建 user.go 文件

package repoimport ("context""user/internal/model"
)type UserRepo interface {Save(ctx context.Context, user *model.User) error
}

該代碼提供了一個Save接口，用來保存 User。

在 user/internal/ 下創建 dao 目錄，并創建 user.go 文件，提供接口的實現。

package daoimport ("context""fmt""user/database""user/internal/model"
)type UserDao struct {*database.DBConn
}func NewUserDao(conn *database.DBConn) *UserDao {return &UserDao{conn,}
}func (d *UserDao) Save(ctx context.Context, user *model.User) error {sql := fmt.Sprintf("insert into %s (name, gender) values(?, ?)", user.TableName())result, err := d.Conn.ExecCtx(ctx, sql, user.Name, user.Gender)if err != nil {return err}id, err := result.LastInsertId()if err != nil {return err}user.Id = idreturn nil}

5. 修改 user/internal/config目錄下的 config.go文件

由于，我們需要連接 mysql 數據庫。因此，我們需要從配置文件中讀取 mysql 連接的配置。go-zero 提供了一種簡便方式，可以自動讀取配置。

首先，修改 user/etc/user.yaml中的配置，如下：

Name: user.rpc
ListenOn: 0.0.0.0:8080
Etcd:Hosts:- 127.0.0.1:2379Key: user.rpcMysql:Datasource: root:thinker@tcp(127.0.0.1:33306)/zero_mall?charset=utf8mb4&parseTime=True&loc=Asia%2FShanghai

Mysql 的配置是我自己手動添加的。

user/internal/config/config.go 文件如下：

package configimport "github.com/zeromicro/go-zero/zrpc"type Config struct {zrpc.RpcServerConfMysql MysqlConfig
}type MysqlConfig struct {DataSource string
}

該文件中添加了 MySqlConfig 結構體，并且在Config 結構體中添加了 Mysql 變量。這樣 go-zero 可以自動讀取到 user.yaml 中 Mysql連接配置。

6. 修改 user/rpc/user.proto 文件，并重新生成代碼

user.proto 文件

option go_package = "./user";message IdRequest {string id = 1;
}message UserRequest {string id = 1;string name = 2;string gender = 3;
}message UserResponse {string id = 1;string name = 2;string gender = 3;
}service User {rpc getUser(IdRequest) returns(UserResponse);rpc save(UserRequest) returns(UserResponse);
}

該代碼中，添加了 rpc save(UserRequest) returns(UserResponse); 接口。并使用如下命令重新生成代碼：

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

7. 修改userserver.go 和 getuserlogic.go 代碼

將生成的 types/user 和 userclient/下的代碼，覆蓋之前生成的代碼。并且把 internal/server/userserver.go 文件中的如下代碼（新生成的代碼）：

func (s *UserServer) Save(ctx context.Context, in *user.UserRequest) (*user.UserResponse, error) {l := logic.NewUserLogic(ctx, s.svcCtx)return l.SaveUser(in)
}

放到 user/internal/server/userserver.go（舊文件中）文件中。

修改 user/internal/logic/getuserlogic.go代碼，為了命名規范，我將getuserlogic.go 該成了 userlogic.go。

package logicimport ("context""strconv""user/internal/model""user/internal/svc""user/types/user""github.com/zeromicro/go-zero/core/logx"
)type UserLogic struct {ctx    context.ContextsvcCtx *svc.ServiceContextlogx.Logger
}func NewUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UserLogic {return &UserLogic{ctx:    ctx,svcCtx: svcCtx,Logger: logx.WithContext(ctx),}
}func (l *UserLogic) GetUser(in *user.IdRequest) (*user.UserResponse, error) {// todo: add your logic here and delete this linereturn &user.UserResponse{Id:     in.GetId(),Name:   "hello user name",Gender: "man",}, nil
}func (l *UserLogic) SaveUser(in *user.UserRequest) (*user.UserResponse, error) {data := &model.User{Name:   in.Name,Gender: in.Gender,}err := l.svcCtx.UserRepo.Save(context.Background(), data)if err != nil {return nil, err}return &user.UserResponse{Id:     strconv.FormatInt(data.Id, 10),Name:   data.Name,Gender: data.Gender,}, nil
}

userlogic 相當于業務組件，這里實現了用戶保存到數據庫的邏輯。

到此，在user服務中連接mysql數據庫，并實現通過rpc接口調用將用戶數據保存到 mysql 邏輯已經完成。

8. 調用 rpc 接口，測試 user 保存到數據庫

1. 在mall 目錄下執行如下命令，創建 userapi微服務（為了測試user rpc 保存到數據的功能）：

goctl api new userapi

2. 在 userapi 目錄下，創建一個 go.mod 文件，文件內容如下：

module userapigo 1.22.2

3. 在 mall 目錄下執行如下命令，將 userapi 加入workspace中

go work use userapi/
cd userapi/
go mod tidy

到此，生成的代碼結構如下：
在這里插入圖片描述

生成的包名稍微有點問題，建議直接手動修改一下。將user/api 改為 userapi。

4. 修改 userapi/etc/user-api.yaml 文件如下：

Name: userapi-api
Host: 0.0.0.0
Port: 8888
UserRpc:Etcd:Hosts:- 127.0.0.1:2379Key: user.rpc

該文件中增加了 UserRpc 配置，主要是為了調用rpc接口。

5. 修改 userapi/internal/config/config.go 文件

package configimport ("github.com/zeromicro/go-zero/rest""github.com/zeromicro/go-zero/zrpc"
)type Config struct {rest.RestConfUserRpc zrpc.RpcClientConf
}

增加了 UserRpc 變量，為了讀取 user-api.yaml 中的配置。

6. 修改userapi/internal/handler/routers.go文件

在 userapi/internal/handler 目錄下創建 register.go 文件和 userhandler.go文件

package handlerimport ("github.com/zeromicro/go-zero/rest/httpx""net/http""userapi/internal/logic""userapi/internal/types"
)func (u *UserHandler) register(w http.ResponseWriter, r *http.Request) {var req types.Requestif err := httpx.ParseJsonBody(r, &req); err != nil {httpx.ErrorCtx(r.Context(), w, err)return}l := logic.NewUserLogic(r.Context(), u.svcCtx)resp, err := l.Register(&req)if err != nil {httpx.ErrorCtx(r.Context(), w, err)} else {httpx.OkJsonCtx(r.Context(), w, resp)}
}

userhandler.go

package handlerimport ("userapi/internal/svc"
)type UserHandler struct {svcCtx *svc.ServiceContext
}func NewUserHandler(svcCtx *svc.ServiceContext) *UserHandler {return &UserHandler{svcCtx: svcCtx,}}

刪除自動生成的代碼 userapihandler.go 文件。

將生成的 userapi/internal/handler/routers.go 文件修改如下：

// Code generated by goctl. DO NOT EDIT.
package handlerimport ("net/http""userapi/internal/svc""github.com/zeromicro/go-zero/rest"
)func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {handler := NewUserHandler(serverCtx)server.AddRoutes([]rest.Route{{Method:  http.MethodPost,Path:    "/register",Handler: handler.register,},},)
}

7. 修改 userapi/internal/types.go 文件

// Code generated by goctl. DO NOT EDIT.
package typestype Request struct {Name string `json:"name"`Gender string `json:"gender"`
}type Response struct {Message string `json:"message"`Data any `json:"data"`
}

這里主要是為了處理 http請求過來的 json數據。

8. 為了使用 Rpc 服務，修改 userapi/internal/svc/servicecontext.go 文件

servicecontext.go 文件

package svcimport ("github.com/zeromicro/go-zero/zrpc""user/userclient""userapi/internal/config"
)type ServiceContext struct {Config  config.ConfigUserRpc userclient.User
}func NewServiceContext(c config.Config) *ServiceContext {return &ServiceContext{Config:  c,UserRpc: userclient.NewUser(zrpc.MustNewClient(c.UserRpc)),}
}

這里加入了 UserRpc 變量，為了遠程調用User服務提供Save方法。

9.修改業務代碼 userapi/internal/logic/userapilogic.go

package logicimport ("context""time""user/types/user""userapi/internal/svc""userapi/internal/types""github.com/zeromicro/go-zero/core/logx"
)type UserLogic struct {logx.Loggerctx    context.ContextsvcCtx *svc.ServiceContext
}func NewUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UserLogic {return &UserLogic{Logger: logx.WithContext(ctx),ctx:    ctx,svcCtx: svcCtx,}
}func (l *UserLogic) Register(req *types.Request) (resp *types.Response, err error) {// todo: add your logic here and delete this linectx, cancelFunc := context.WithTimeout(context.Background(), 5*time.Second)defer cancelFunc()userResponse, err := l.svcCtx.UserRpc.Save(ctx, &user.UserRequest{Name:   req.Name,Gender: req.Gender,})if err != nil {return nil, err}return &types.Response{Message: "success",Data:    userResponse,}, nil
}