Go 語言微服務(wù)框架 Kratos 操作 MySQL 和 Redis
作者：frank
本文我們以 XORM 和 Go Redis 為例，介紹 Kratos 微服務(wù)框架怎么集成 XORM[1] 和 Go Redis[2] 操作 MySQL 和 Redis。
1.介紹
Go 語言微服務(wù)框架 Kratos 不限制使用任何第三方庫，我們可以根據(jù)個(gè)人喜好選擇第三方庫。
本文我們以 XORM 和 Go Redis 為例，介紹 Kratos 微服務(wù)框架怎么集成 XORM[1] 和 Go Redis[2] 操作 MySQL 和 Redis。
2.集成 XORM 和 Go Redis
創(chuàng)建項(xiàng)目
kratos new blog安裝 MySQL 驅(qū)動
示例代碼：
go get -u github.com/go-sql-driver/mysql安裝 xorm
示例代碼：
go get xorm.io/xorm安裝 go redis
示例代碼：
go get github.com/redis/go-redis/v9客戶端
編寫文件 blog/internal/data/data.go。
導(dǎo)入第三方庫
import (
_ "github.com/go-sql-driver/mysql"
"github.com/redis/go-redis/v9"
"xorm.io/xorm"
)添加 xorm 和 go redis 客戶端
// Data .
type Data struct {
// TODO wrapped database client
dbEngine *xorm.Engine
rdb *redis.Client
}
// NewData .
func NewData(c *conf.Data, logger log.Logger, dbEngin *xorm.Engine, rdb *redis.Client) (*Data, func(), error) {
cleanup := func() {
log.NewHelper(logger).Info("closing the data resources")
}
return &Data{
dbEngine: dbEngin,
rdb: rdb,
}, cleanup, nil
}創(chuàng)建函數(shù) NewDbEngine 和 NewRedisClient
func NewDbEngine(c *conf.Data) (dbEngine *xorm.Engine, err error) {
dbEngine, err = xorm.NewEngine(c.Database.Driver, c.Database.Source)
return
}
func NewRedisClient(c *conf.Data) (rdb *redis.Client) {
rdb = redis.NewClient(&redis.Options{
Addr: c.Redis.Addr,
Password: c.Redis.Password,
DB: int(c.Redis.Db),
})
return
}添加 Wire 提供者
編寫 blog/internal/data/data.go 文件，添加 NewDbEngine, NewRedisClient。
// ProviderSet is data providers.
var ProviderSet = wire.NewSet(NewData, NewGreeterRepo, NewDbEngine, NewRedisClient)3.使用 XORM 和 Go Redis
我們以操作用戶表 user 為例，介紹怎么使用 XORM 和 Go Redis。
數(shù)據(jù)庫表 user 表結(jié)構(gòu)
CREATE TABLE `user` (
`id` bigint(20) unsigned NOT NULL AUTO_INCREMENT COMMENT 'id',
`name` varchar(30) NOT NULL DEFAULT '' COMMENT 'user name',
`email` varchar(50) NOT NULL DEFAULT '' COMMENT 'user email',
`password` varchar(30) NOT NULL DEFAULT '' COMMENT 'user password',
`created` int(11) NOT NULL DEFAULT '0' COMMENT 'create time',
`updated` int(11) NOT NULL DEFAULT '0' COMMENT 'updated time',
PRIMARY KEY (`id`),
KEY `idx_user` (`name`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='user'Usecase 層
首先，我們需要在 blog/internal/biz 目錄中，創(chuàng)建文件 user.go。
在 user.go 文件中，我們定義 User 結(jié)構(gòu)體，定義 UserRepository，定義 UserUsecase。
定義 User 結(jié)構(gòu)體
type User struct {
Id int64 `xorm:"autoincr""`
Name string
Email string
Password string
Created int64 `xorm:"created""`
Updated int64 `xorm:"updated""`
}定義 UserRepository
type UserRepository interface {
Create(ctx context.Context, user *User) (int64, error)
Get(ctx context.Context, user *User) (bool, error)
RedisGetUser(ctx context.Context, id int64) (val string, err error)
RedisSetUser(ctx context.Context, id int64, val interface{}, expiration time.Duration) (err error)
}定義 UserUsecase
type UserUsecase struct {
userRepo UserRepository
}
func NewUserUsecase(userRepo UserRepository) *UserUsecase {
return &UserUsecase{
userRepo: userRepo,
}
}
func (u *UserUsecase) CreateUser(ctx context.Context, user *User) (id int64, err error) {
id, err = u.userRepo.Create(ctx, user)
if err != nil {
return
}
if id > 0 {
var b []byte
b, err = json.Marshal(user)
if err != nil {
return
}
err = u.userRepo.RedisSetUser(ctx, user.Id, string(b), time.Second*300)
if err != nil {
return
}
}
return
}添加 Wire 提供者
編寫 blog/internal/biz/biz.go 文件，添加 NewUserUsecase。
// ProviderSet is biz providers.
var ProviderSet = wire.NewSet(NewGreeterUsecase, NewUserUsecase)Repository 層
然后，我們需要在 blog/internal/data 目錄中，創(chuàng)建 user.go 文件和 redis.go 文件。
在 user.go 文件和 redis.go 文件中，我們實(shí)現(xiàn) UserRepository 接口中的方法。
MySQL
type userRepository struct {
data *Data
}
func NewUserRepository(data *Data) biz.UserRepository {
return &userRepository{
data: data,
}
}
func (u *userRepository) Create(ctx context.Context, user *biz.User) (id int64, err error) {
var affected int64
affected, err = u.data.dbEngine.InsertOne(user)
if err != nil {
return
}
if affected > 0 {
id = user.Id
}
return
}Redis
func (u *userRepository) RedisSetUser(ctx context.Context, id int64, val interface{}, expiration time.Duration) (err error) {
key := fmt.Sprintf("user:%d", id)
err = u.data.rdb.Set(ctx, key, val, expiration).Err()
if errors.Is(err, redis.Nil) {
err = nil
return
}
return
}添加 Wire 提供者
編寫 blog/internal/data/data.go 文件，添加 NewUserRepository。
// ProviderSet is data providers.
var ProviderSet = wire.NewSet(NewData, NewGreeterRepo, NewDbEngine, NewUserRepository, NewRedisClient)Delivery 層
再然后，我們編輯 blog/internal/service/user.go 文件，完善該文件中的方法。
type UserService struct {
pb.UnimplementedUserServer
userUcase *biz.UserUsecase
}
func NewUserService(userUcase *biz.UserUsecase) *UserService {
return &UserService{
userUcase: userUcase,
}
}
func (s *UserService) CreateUser(ctx context.Context, req *pb.CreateUserRequest) (*pb.CreateUserReply, error) {
user := &biz.User{
Name: req.Name,
Email: req.Email,
Password: req.Password,
}
id, err := s.userUcase.CreateUser(ctx, user)
if err != nil {
return nil, err
}
name := fmt.Sprintf("user_%d", id)
return &pb.CreateUserReply{
Name: name,
}, nil
}添加 Wire 提供者
編寫 blog/internal/service/service.go 文件，添加 NewUserService。
// ProviderSet is service providers.
var ProviderSet = wire.NewSet(NewGreeterService, NewUserService)注冊服務(wù)
最后，進(jìn)入 blog/internal/server 目錄，編寫 http.go 和 grpc.go 文件，添加函數(shù)簽名 user *service.UserService。
并且，分別在函數(shù)體中注冊服務(wù)。
示例代碼：
HTTP
v1_user.RegisterUserHTTPServer(srv, user)gRPC
v1_user.RegisterUserServer(srv, user)Wire 生成代碼
進(jìn)入目錄 blog/cmd/blog，執(zhí)行 wire 命令。
cd blog/cmd/blog
wire項(xiàng)目運(yùn)行和測試
Kratos 運(yùn)行
kratos runcurl 請求示例
curl -H "Content-Type: application/json" -X POST -d '{"name":"tony", "email":"tony@gmail.com", "password":"tony"}' http://192.168.110.209:8000/user/create4.總結(jié)
本文我們通過示例代碼，介紹 Kratos 微服務(wù)框架怎么集成第三方庫，操作 MySQL 和 Redis。
責(zé)任編輯：武曉燕
來源：
Golang語言開發(fā)棧
