昔念
666557416d
- 在 ClientData 中添加 sync.Mutex 以确保并发安全 - 实现 SetPlayer 和 GetPlayer 等方法来封装 player 字段的操作 - 更新相关代码以使用新的 ClientData 结构和方法
147 lines
3.2 KiB
Go
147 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
|
|
UserLogin func(int32, int32) error
|
|
UserLogout func(int32, 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 //用户登出事件
|
|
}
|