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

package logic

import (
	"context"
	"github.com/pkg/errors"
	"google.golang.org/protobuf/types/known/structpb"
	"time"
	"ylink/comm/model"
	"ylink/comm/result"
	"ylink/core/inner/rpc/internal/ext"
	"ylink/core/inner/rpc/internal/svc"
	"ylink/core/inner/rpc/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

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

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

func (l *CsFetchPlayerQueueLogic) CsFetchPlayerQueue(in *pb.InnerCsFetchPlayerQueueReq) (*pb.InnerCsFetchPlayerQueueResp, error) {
	queueSize := int32(ext.WaitingQueue.Size())
	if queueSize == 0 {
		// 等待队列为空直接返回
		return &pb.InnerCsFetchPlayerQueueResp{
			List: nil,
		}, nil
	}

	var index int32 = 0
	if in.Limit != 0 && in.Limit < queueSize {
		queueSize = in.Limit
	}

	queue := make([]interface{}, queueSize)

	for iter := ext.WaitingQueue.Begin(); iter.IsValid(); iter.Next() {
		if index >= queueSize {
			break
		}
		info := iter.Value().(*model.PlayerInfo)
		queue[index] = map[string]interface{}{
			"game_id":   info.GameId,
			"player_id": info.PlayerId,
			"wait_time": time.Now().Unix() - info.EnqueueTs,
		}
		index += 1
	}

	//for node := ext.WaitingQueue.FrontNode(); node != nil && index < queueLen; node = node.Next() {
	//	info := node.Value.(*model.PlayerInfo)
	//	queue[index] = map[string]interface{}{
	//		"game_id":   info.GameId,
	//		"player_id": info.PlayerId,
	//		"wait_time": time.Now().Unix() - info.EnqueueTs,
	//	}
	//	index += 1
	//}

	list, err := structpb.NewList(queue)
	if err != nil {
		return nil, errors.Wrap(result.NewErrMsg("fetch player wait queue error"), "")
	}

	return &pb.InnerCsFetchPlayerQueueResp{
		List: list,
	}, nil
}