ylink/core/inner/rpc/internal/logic/csconnectplayerlogic.go

package logic

import (
	"context"
	"github.com/bytedance/sonic"
	"github.com/gookit/event"
	treemap "github.com/liyue201/gostl/ds/map"
	"github.com/pkg/errors"
	"github.com/robfig/cron/v3"
	"github.com/zeromicro/go-zero/core/logx"
	"time"
	"ylink/comm/ctxdata"
	"ylink/comm/globalkey"
	"ylink/comm/model"
	"ylink/comm/result"
	"ylink/comm/trace"
	"ylink/core/inner/rpc/internal/ext"
	"ylink/core/inner/rpc/internal/svc"
	"ylink/core/inner/rpc/pb"
)

type CsConnectPlayerLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewCsConnectPlayerLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CsConnectPlayerLogic {
	return &CsConnectPlayerLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *CsConnectPlayerLogic) CsConnectPlayer(in *pb.InnerCsConnectPlayerReq) (*pb.InnerCsConnectPlayerResp, error) {

	playerInfo := ext.GetOnlinePlayerInfo(in.GameId, in.PlayerId)
	if playerInfo == nil {
		return nil, errors.Wrapf(result.NewErrMsg("玩家不在线"), "")
	}

	playerInfo.CsId = in.CsId
	playerInfo.DequeueTs = time.Now().Unix()

	if ext.GameConnectedMap.Contains(in.GameId) {
		playerConnectedMap := ext.GameConnectedMap.Get(in.GameId).(*treemap.Map)
		playerConnectedMap.Insert(in.PlayerId, playerInfo)
	} else {
		playerConnectedMap := treemap.New(treemap.WithGoroutineSafe())
		playerConnectedMap.Insert(in.PlayerId, playerInfo)
		ext.GameConnectedMap.Insert(in.GameId, playerConnectedMap)
	}

	// 移除WaitingQueue
	uniqueId := in.GameId + "_" + in.PlayerId
	if ext.WaitingQueue.Contains(uniqueId) {
		l.Logger.Infof("remove the player from the queue, game_id: %s, player_id: %s", in.GameId, in.PlayerId)
		ext.WaitingQueue.Erase(uniqueId)

		// 广播客户端更新等待队列信息
		payload, _ := sonic.MarshalString(&model.CommandMessage{
			CmdInfo: map[string]interface{}{
				"queue_size": ext.WaitingQueue.Size(),
			},
		})
		kMsg, _ := sonic.MarshalString(&model.KqMessage{
			Opt:        model.CMD_UPDATE_WAITING_QUEUE,
			CreateTs:   time.Now().Unix(),
			Payload:    payload,
			SenderId:   uniqueId,
			ReceiverId: globalkey.AllNormalPlayer,
			GameId:     in.GameId,
			Uid:        in.PlayerId,
			Ext:        "",
		})
		l.svcCtx.KqCmdBoxProducer.SendMessage(l.ctx, kMsg, globalkey.AllNormalPlayer)
	}

	traceId := ctxdata.GetTraceIdFromCtx(l.ctx)
	trace.RunOnTracing(traceId, func(ctx context.Context) {
		var entryId cron.EntryID
		entryId, _ = l.svcCtx.TimeoutCron.AddFunc("@every 1m", func() {
			var timeoutTs int64
			if playerInfo.LastChatTs == 0 {
				timeoutTs = time.Now().Unix() - playerInfo.ConnectTs
			} else {
				timeoutTs = time.Now().Unix() - playerInfo.LastChatTs
			}
			if timeoutTs >= 3600 {
				// 释放计时器任务
				_ = event.MustFire(globalkey.EventRemoveTimeoutJob, event.M{"entry_id": entryId})
				l.Logger.Infof("trigger timeout event, remove cron job, entry id: %d", entryId)

				trace.StartTrace(ctx, "InnerServer.CountDownTimer.SendCmdMessage", func(ctx context.Context) {
					// 发踢下线的command指令
					uniqueId := in.GameId + "_" + in.PlayerId
					payload, _ := sonic.MarshalString(&model.CommandMessage{
						CmdInfo: "",
					})
					kMsg, _ := sonic.MarshalString(&model.KqMessage{
						Opt:        model.CMD_CHAT_TIMEOUT,
						CreateTs:   time.Now().Unix(),
						Payload:    payload,
						SenderId:   uniqueId,
						ReceiverId: uniqueId,
						GameId:     in.GameId,
						Uid:        in.PlayerId,
						Ext:        "",
					})
					l.svcCtx.KqCmdBoxProducer.SendMessage(ctx, kMsg, uniqueId)
				})
			}
		})
	})

	return &pb.InnerCsConnectPlayerResp{}, nil
}