bl/common/rpc/rpc.go

package rpc

import (
	"blazing/common/data/share"
	"blazing/cool"

	"context"
	"fmt"
	"log"

	config "blazing/modules/config/service"

	"github.com/filecoin-project/go-jsonrpc"
	"github.com/gogf/gf/v2/util/gconv"
)

// Define the server handler
type ServerHandler struct{}

// 实现踢人
func (*ServerHandler) Kick(_ context.Context, userid uint32) error {

	useid1, err := share.ShareManager.GetUserOnline(userid)

	if err != nil {
		return fmt.Errorf("user not found", err)
	}

	cl, ok := cool.GetClient(useid1)
	if ok {
		err := cl.KickPerson(userid) //实现指定服务器踢人
		if err != nil {
			return fmt.Errorf("踢人失败", err)
		}
	}
	return nil
}

// 注册logic服务器
func (*ServerHandler) RegisterLogic(ctx context.Context, id, port uint16) error {
	fmt.Println("注册logic服务器", id, port)

	//TODO 待修复滚动更新可能导致的玩家可以同时在旧服务器和新服务器同时在线的bug
	revClient, ok := jsonrpc.ExtractReverseClient[cool.ClientHandler](ctx)
	if !ok {
		return fmt.Errorf("no reverse client")
	}
	t := config.NewServerService().GetServerID((id))

	aa, ok := cool.GetClient(t.Port)
	if ok && aa != nil { //如果已经存在且这个端口已经被存过
		aa.QuitSelf(0)
	}
	cool.AddClient(port, &revClient)

	//Refurh()
	return nil

}

func CServer() *jsonrpc.RPCServer {
	// create a new server instance
	rpcServer := jsonrpc.NewServer(jsonrpc.WithReverseClient[cool.ClientHandler](""))

	rpcServer.Register("", &ServerHandler{})

	return rpcServer

}

var closer jsonrpc.ClientCloser

func StartClient(id, port uint16, callback any) *struct {
	Kick func(uint32) error

	RegisterLogic func(uint16, uint16) error
} {
	cool.Config.File.Domain = "127.0.0.1"
	var rpcaddr = "ws://" + cool.Config.File.Domain + gconv.String(cool.Config.Address) + "/rpc"

	closer1, err := jsonrpc.NewMergeClient(context.Background(),
		rpcaddr, "", []interface{}{
			&RPCClient,
		}, nil, jsonrpc.WithClientHandler("", callback),
		jsonrpc.WithReconnFun(func() { RPCClient.RegisterLogic(id, port) }),
	)
	if err != nil {
		log.Fatalf("Failed to create client: %v", err)
	}

	//if port != 0 { //注册logic
	defer RPCClient.RegisterLogic(id, port)

	//}

	closer = closer1

	return &RPCClient
}

// Setup RPCClient with reverse call handler
var RPCClient struct {
	Kick func(uint32) error //踢人

	RegisterLogic func(uint16, uint16) error

	//	UserLogin  func(int32, int32) error //用户登录事件
	//	UserLogout func(int32, int32) error //用户登出事件
}