go_dreamfactory/modules/user/api_login.go
2022-07-20 17:56:09 +08:00

120 lines
2.7 KiB
Go

package user
import (
"go_dreamfactory/comm"
"go_dreamfactory/lego/sys/event"
"go_dreamfactory/lego/sys/log"
"go_dreamfactory/pb"
"go_dreamfactory/sys/cache"
"go_dreamfactory/utils"
"time"
"go.mongodb.org/mongo-driver/mongo"
"google.golang.org/protobuf/proto"
)
//参数校验
func (this *apiComp) LoginCheck(session comm.IUserSession, req *pb.UserLoginReq) (code pb.ErrorCode) {
if req.Account == "" {
code = pb.ErrorCode_ReqParameterError
}
return
}
//登录
func (this *apiComp) Login(session comm.IUserSession, req *pb.UserLoginReq) (code pb.ErrorCode, data proto.Message) {
if code = this.LoginCheck(session, req); code != pb.ErrorCode_Success {
return
}
var (
err error
user *pb.DBUser
)
defer func() {
if user != nil {
err = session.SendMsg(string(this.module.GetType()), UserSubTypeLogin, &pb.UserLoginResp{
Data: user,
})
if err != nil {
code = pb.ErrorCode_SystemError
return
}
event.TriggerEvent(comm.EventUserLogin, user.Uid)
}
}()
//从mgo查询user
user, err = this.module.modelUser.FindByAccount(req.Sid, req.Account)
if err != nil {
if err != mongo.ErrNoDocuments {
log.Errorf("User_FindByAccount err %v", err)
return
}
}
var isNewUser bool
if user == nil {
//如果是新玩家,创建一条基础的数据,页面会引导进入创角页面
user = &pb.DBUser{
Sid: req.Sid,
Binduid: req.Account,
Lastloginip: session.GetIP(),
}
err = this.module.modelUser.User_Create(user)
if err != nil {
log.Errorf("User_CreateUser err %v", err)
return
}
isNewUser = true
}
//bind user
err = session.Bind(user.Uid, this.service.GetId())
if err != nil {
this.module.Errorf("err:%v", err)
code = pb.ErrorCode_BindUser
return
}
//缓存user session
err = this.module.modelSession.Change(user.Uid, map[string]interface{}{
"uid": user.Uid,
"sessionId": session.GetSessionId(),
"gatewayServiceId": session.GetGatewayServiceId(),
"ip": session.GetIP(),
},
cache.WithExpire(time.Hour*12),
cache.WithDisabledMgoLog())
if err != nil {
code = pb.ErrorCode_DBError
return
}
//不是新账号
if !isNewUser {
//set user other info
user.Logintime = time.Now().Unix()
user.Lastloginip = session.GetIP()
update := utils.StructToMap(user)
err = this.module.modelUser.Change(user.Uid, update)
if err != nil {
code = pb.ErrorCode_DBError
return
}
go func() {
if this.module.modelUser.isLoginFirst(user.Logintime) {
//清空日常
this.module.ModuleTask.ResetTask(user.Uid, comm.TASK_DAILY)
//清周常
if utils.IsAfterWeek(user.Logintime) {
this.module.ModuleTask.ResetTask(user.Uid, comm.TASK_WEEKLY)
}
}
}()
}
return
}