game/server/login/server/processor.go

package server

import (
	"encoding/json"
	"game/common/model/user"
	"game/common/proto/pb"
	"game/common/rpcName"
	"game/common/utils"
	"github.com/fox/fox/etcd"
	"github.com/fox/fox/ipb"
	"github.com/fox/fox/ksync"
	"github.com/fox/fox/log"
	"github.com/fox/fox/processor"
	"github.com/fox/fox/service"
	"time"
)

const (
	timeout = time.Second * 30
)

func (s *LoginService) initProcessor() {
	s.processor.RegisterMessages(processor.RegisterMetas{
		pb.MsgId_C2SUserLoginId: {pb.C2SUserLogin{}, s.onLoginOrRegister},
	})
}

func (s *LoginService) checkLoginOrRegister(req *pb.C2SUserLogin) (us *user.UserAccount, code pb.ErrCode, node *etcd.ServiceNode) {
	var err error
	node, err = s.bindService.RandServiceNode(pb.ServiceTypeId_STI_DB)
	if err != nil {
		log.ErrorF(s.Log("not find db service.err:%s ", err.Error()))
		return nil, pb.ErrCode_SystemErr, node
	}
	if req.Version < "20250601123030" {
		return nil, pb.ErrCode_VersionTooLow, node
	}
	us = &user.UserAccount{
		Username:    req.Username,
		Password:    req.Password,
		DeviceID:    req.DeviceId,
		LastLoginIP: req.Ip,
	}
	rpcMsg := ipb.MakeRpcMsg[user.UserAccount](rpcName.GetUserAccount, 0, us)
	rspMsg, err := s.Call(service.RpcTopicEx(node.Name), timeout, rpcMsg)
	if err != nil {
		log.ErrorF(s.Log("call rpc:%v err:%s ", rpcMsg.RpcMsgId, err.Error()))
		return nil, pb.ErrCode_SystemErr, node
	}
	_ = json.Unmarshal(rspMsg.Msg, us)
	log.DebugF("收到rcp:%v返回数据:%v", rpcName.GetUserAccount, string(rspMsg.Msg))
	if us.ID == 0 {
		// 没有帐号，创建帐号
		us = &user.UserAccount{
			Username:      req.Username,
			Password:      req.Password,
			DeviceID:      req.DeviceId,
			LastLoginIP:   req.Ip,
			LastLoginTime: time.Now(),
			RegisterIP:    req.Ip,
		}
		rpcMsg = ipb.MakeRpcMsg[user.UserAccount](rpcName.CreateUserAccount, 0, us)
		rspMsg, err = s.Call(service.RpcTopicEx(node.Name), timeout, rpcMsg)
		if err != nil {
			log.ErrorF(s.Log("call rpc:%v err:%s ", rpcMsg.RpcMsgId, err.Error()))
			return nil, pb.ErrCode_SystemErr, node
		}
		_ = json.Unmarshal(rspMsg.Msg, us)
		if us.ID == 0 {
			log.ErrorF(s.Log("call rpc:%v err", rpcMsg.RpcMsgId))
			return nil, pb.ErrCode_SystemErr, node
		}
		log.DebugF("收到rcp:%v返回数据:%v", rpcName.CreateUserAccount, string(rspMsg.Msg))
	}
	if !utils.CheckPassword(req.Password, us.Password) {
		log.ErrorF(s.Log("用户密码:%v 数据库中密码:%v", req.Password, us.Password))
		return nil, pb.ErrCode_LoginUserOrPwdErr, node
	}
	switch us.Status {
	case user.AccountFrozen:
		return nil, pb.ErrCode_AccountFrozen, node
	case user.AccountBanned:
		return nil, pb.ErrCode_AccountBanned, node
	default:
		return us, pb.ErrCode_OK, node
	}
}

// 生成JWT令牌(简化版)
func generateToken(userID uint, username string) (string, error) {
	_ = userID
	_ = username
	// 这里应该使用JWT库生成实际令牌
	// 简化实现，实际项目中请使用安全的JWT实现
	return "generated-token-placeholder", nil
}

// 登录或注册
func (s *LoginService) onLoginOrRegister(iMsg *ipb.InternalMsg, req *pb.C2SUserLogin) {
	ksync.GoSafe(func() {
		us, code, node := s.checkLoginOrRegister(req)
		userId := int64(0)
		rsp := &pb.S2CUserLogin{Code: code}
		if us != nil && code == pb.ErrCode_OK {
			rsp.UserId = int64(us.ID)
			rsp.Token, _ = generateToken(us.ID, us.Username)
			userId = rsp.UserId
		}
		s.SendServiceMsg(service.TopicEx(iMsg.ServiceName), iMsg.ConnId, userId, int32(pb.MsgId_S2CUserLoginId), rsp)

		if us != nil && us.ID > 0 {
			switch code {
			case pb.ErrCode_LoginUserOrPwdErr:
				rpcMsg := ipb.MakeRpcMsg[user.UserLoginLog](rpcName.GetUserAccount, 0, &user.UserLoginLog{
					UID:         us.ID,
					LoginIP:     us.LastLoginIP,
					LoginTime:   time.Now(),
					DeviceInfo:  us.DeviceID,
					LoginResult: code == pb.ErrCode_OK,
					FailReason:  code.String(),
				})
				_, _ = s.Call(service.RpcTopicEx(node.Name), timeout, rpcMsg)
			}
		}

	}, nil)
}