ylink/bff/rpcbff/internal/logic/connectlogic.go

package logic

import (
	"context"
	"google.golang.org/protobuf/types/known/structpb"

	"ylink/bff/rpcbff/internal/svc"
	"ylink/bff/rpcbff/pb"

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

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

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

func (l *ConnectLogic) Connect(in *pb.CommandReq, stream pb.Rpcbff_ConnectServer) error {
	// todo: 验证token
	// todo: 把stream放入资源pool

	l.Logger.Info("invoke func connect")
	l.Logger.Infof("%s", in.Token)

	data, _ := structpb.NewStruct(map[string]interface{}{})

	return stream.Send(&pb.CommandResp{
		CommandCode: 0,
		CommandMsg:  "success",
		CommandData: data,
	})
}