mmorpg/service-user
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat kafka 改版 2

Browse Source
This commit is contained in:
DESKTOP-V763RJ7\Administrator
2026-01-30 11:02:55 +08:00
parent 3b14697a45
commit 9781cd126f
7 changed files with 18 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/dao/repository/define.go
View File

@@ -3,6 +3,7 @@ package repository
import (
"fmt"
"git.hlsq.asia/mmorpg/service-common/db/mysql"
"git.hlsq.asia/mmorpg/service-user/internal/dao/query"
)
var dbName mysql.DBName = "user_db"
@@ -12,6 +13,10 @@ var (
userCacheByPhone = "c:user:p:%v"
)
func Query() *query.Query {
return query.Use(mysql.GetDB(dbName))
}
func keyCacheBySn(sn string, tableName string) string {
return fmt.Sprintf(cacheBySn, tableName, sn)
}

5
internal/dao/repository/users.go
View File

@@ -2,7 +2,6 @@ package repository
import (
"context"
"git.hlsq.asia/mmorpg/service-common/db/mysql"
"git.hlsq.asia/mmorpg/service-common/db/redis"
"git.hlsq.asia/mmorpg/service-user/internal/dao/model"
"git.hlsq.asia/mmorpg/service-user/internal/dao/query"
@@ -15,10 +14,10 @@ type UserDao struct {
cache *redis.CacheClient
}
func NewUserDao(ctx context.Context, cache ...*redis.CacheClient) *UserDao {
func NewUserDao(ctx context.Context, query *query.Query, cache ...*redis.CacheClient) *UserDao {
dao := &UserDao{
ctx: ctx,
query: query.Use(mysql.GetDB(dbName)),
query: query,
}
if len(cache) > 0 {
dao.cache = cache[0]

6
internal/grpc_server/server.go
View File

@@ -15,7 +15,7 @@ import (
)
func (s *Server) PhoneLogin(ctx context.Context, req *grpc_pb.PhoneLoginReq) (*grpc_pb.PhoneLoginResp, error) {
userDao := repository.NewUserDao(ctx, redis.GetCacheClient())
userDao := repository.NewUserDao(ctx, s.query, redis.GetCacheClient())
user, err := userDao.FindByPhone(req.Phone)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
@@ -42,7 +42,7 @@ func (s *Server) WxMiniLogin(ctx context.Context, req *grpc_pb.WxMiniLoginReq) (
if err != nil {
return nil, utils.ErrorsWrap(err)
}
userDao := repository.NewUserDao(ctx, redis.GetCacheClient())
userDao := repository.NewUserDao(ctx, s.query, redis.GetCacheClient())
user, err := userDao.FindByWxUnionIDOrOpenID(session.UnionID, session.OpenID)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
@@ -69,7 +69,7 @@ func (s *Server) GetUserInfo(ctx context.Context, req *grpc_pb.GetUserInfoReq) (
if !utils.ShouldBindUsn(ctx, &req.USN) {
return nil, utils.ErrorsWrap(http_resp.ParamError)
}
userDao := repository.NewUserDao(ctx, redis.GetCacheClient())
userDao := repository.NewUserDao(ctx, s.query, redis.GetCacheClient())
user, err := userDao.FindBySn(req.USN)
if err != nil {
return nil, utils.ErrorsWrap(err)

4
internal/grpc_server/server_init.go
View File

@@ -5,12 +5,15 @@ import (
"git.hlsq.asia/mmorpg/service-common/discover/common"
"git.hlsq.asia/mmorpg/service-common/net/grpc/service"
"git.hlsq.asia/mmorpg/service-common/proto/rs/grpc_pb"
"git.hlsq.asia/mmorpg/service-user/internal/dao/query"
"git.hlsq.asia/mmorpg/service-user/internal/dao/repository"
"google.golang.org/grpc"
)
type Server struct {
grpc_pb.UnimplementedUserServer
service.Base
query *query.Query
}
func NewServer(cfg *config.GrpcConfig) *Server {
@@ -33,6 +36,7 @@ func (s *Server) OnCustomGrpcServerOption() []grpc.ServerOption {
func (s *Server) OnInit(serve *grpc.Server) {
grpc_pb.RegisterUserServer(serve, s)
s.query = repository.Query()
}
func (s *Server) OnClose() {