Files
bl/common/rpc/rpc.go
昔念 7a87bfa089 refactor(common): 重构 common 模块
- 移除 global.go 文件
- 更新 player.go 中的玩家数据存储方式
- 删除 session.go 文件
- 调整 rpc.go 中的 RPC 客户端方法
- 更新 ServerEvent.go 中的会话管理
- 调整 controller 中的 Maincontroller 结构
- 更新 login.go 中的用户登录逻辑
- 调整 service 中的玩家数据获取方式
- 更新 admin/login.go 和 login.go 中的会话管理
2025-07-15 12:14:17 +08:00

145 lines
3.2 KiB
Go

package rpc
import (
"context"
"fmt"
"log"
"net/http"
"github.com/filecoin-project/go-jsonrpc"
)
const rpcaddr = "127.0.0.1:40000"
var clientmap = make(map[int]*ClientHandler) //客户端map
var usermap = make(map[int]int) //用户->客户端的map
// Define the client handler interface
type ClientHandler struct {
KickPerson func(int) error //踢人,这里是返回具体的logic
QuitSelf func(int) error //关闭服务器进程
}
// Define the server handler
type ServerHandler struct{}
// 实现踢人
func (h *ServerHandler) Kick(ctx context.Context, userid int) error {
useid1, ok := usermap[userid]
if !ok {
return fmt.Errorf("user not found")
}
return clientmap[useid1].KickPerson(userid) //实现指定服务器踢人
// for _, v := range clientmap {
// err := v.KickPerson(userid)
// if err != nil {
// return fmt.Errorf("call on client: %w", err)
// }
// }
//return nil
}
// 退出指定服务器
// func (h *ServerHandler) Quit(ctx context.Context, portid int) error {
// a, ok := clientmap[portid]
// if ok && a != nil {
// a.QuitSelf(0)
// }
// //clientmap[portid].QuitSelf(0)
// return nil
// }
// 注册logic服务器
func (h *ServerHandler) RegisterLogic(ctx context.Context, port int) error {
//TODO 待修复滚动更新可能导致的玩家可以同时在旧服务器和新服务器同时在线的bug
revClient, ok := jsonrpc.ExtractReverseClient[ClientHandler](ctx)
if !ok {
return fmt.Errorf("no reverse client")
}
aa, ok := clientmap[port]
if ok && aa != nil {
aa.QuitSelf(0)
}
clientmap[port] = &revClient
return nil
}
// 用户登录事件
func (h *ServerHandler) UserLogin(ctx context.Context, port int, userid int) error {
usermap[userid] = port
return nil
}
// 用户登出事件
func (h *ServerHandler) UserLogout(ctx context.Context, port int, userid int) error {
delete(usermap, userid)
return nil
}
func StartServer() {
// create a new server instance
rpcServer := jsonrpc.NewServer(jsonrpc.WithReverseClient[ClientHandler](""))
rpcServer.Register("", &ServerHandler{})
// go time.AfterFunc(3000, func() {
// testjsonrpc()
// })
err := http.ListenAndServe(rpcaddr, rpcServer)
fmt.Println(err)
}
var closer jsonrpc.ClientCloser
func StartClient(port uint16, callback any) *struct {
Kick func(int32) error
RegisterLogic func(int32) error
} {
closer1, err := jsonrpc.NewMergeClient(context.Background(), "ws://"+rpcaddr, "", []interface{}{
&RPCClient,
}, nil, jsonrpc.WithClientHandler("", callback))
if err != nil {
log.Fatalf("Failed to create client: %v", err)
}
if port != 0 { //注册logic
RPCClient.RegisterLogic(int32(port))
}
closer = closer1
return &RPCClient
}
// 关闭客户端
func CloseClient() {
if closer != nil {
closer()
}
}
// Setup RPCClient with reverse call handler
var RPCClient struct {
Kick func(int32) error //踢人
RegisterLogic func(int32) error //注册服务器消息
// UserLogin func(int32, int32) error //用户登录事件
// UserLogout func(int32, int32) error //用户登出事件
}