83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
package grpc_server
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"git.hlsq.asia/mmorpg/service-common/db/redis"
|
|
"git.hlsq.asia/mmorpg/service-common/net/http/http_resp"
|
|
"git.hlsq.asia/mmorpg/service-common/proto/rs/grpc_pb"
|
|
"git.hlsq.asia/mmorpg/service-common/utils"
|
|
"git.hlsq.asia/mmorpg/service-user/internal/dao/model"
|
|
"git.hlsq.asia/mmorpg/service-user/internal/dao/repository"
|
|
"git.hlsq.asia/mmorpg/service-user/internal/wechat"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func (s *Server) PhoneLogin(ctx context.Context, req *grpc_pb.PhoneLoginReq) (*grpc_pb.PhoneLoginResp, error) {
|
|
userDao := repository.NewUserDao(ctx, redis.GetCacheClient())
|
|
user, err := userDao.FindByPhone(req.Phone)
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
if user, err = userDao.Create(&model.User{
|
|
Phone: req.Phone,
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
user.Name = fmt.Sprintf("user_%v", user.Sn)
|
|
_ = userDao.Updates(user)
|
|
} else {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return &grpc_pb.PhoneLoginResp{
|
|
USN: user.Sn,
|
|
Name: user.Name,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Server) WxMiniLogin(ctx context.Context, req *grpc_pb.WxMiniLoginReq) (*grpc_pb.WxMiniLoginResp, error) {
|
|
session, err := wechat.MiniCode2Session(req.Code)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
userDao := repository.NewUserDao(ctx, redis.GetCacheClient())
|
|
user, err := userDao.FindByWxUnionIDOrOpenID(session.UnionID, session.OpenID)
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
if user, err = userDao.Create(&model.User{
|
|
WxUnionID: session.UnionID,
|
|
WxMiniOpenID: session.OpenID,
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
user.Name = fmt.Sprintf("user_%v", user.Sn)
|
|
_ = userDao.Updates(user)
|
|
} else {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return &grpc_pb.WxMiniLoginResp{
|
|
USN: user.Sn,
|
|
Name: user.Name,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Server) GetUserInfo(ctx context.Context, req *grpc_pb.GetUserInfoReq) (*grpc_pb.GetUserInfoResp, error) {
|
|
if !utils.ShouldBindUsn(ctx, &req.USN) {
|
|
return nil, http_resp.ParamError
|
|
}
|
|
userDao := repository.NewUserDao(ctx, redis.GetCacheClient())
|
|
user, err := userDao.FindBySn(req.USN)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &grpc_pb.GetUserInfoResp{
|
|
USN: user.Sn,
|
|
Name: user.Name,
|
|
}, nil
|
|
}
|