31d9eb3f9e
All checks were successful
ci/woodpecker/push/my-first-workflow Pipeline was successful
feat(config): 添加超时空地图配置和时间地图查询功能 新增IsTimeSpace字段用于标识地图是否为超时空地图， 添加TimeMap API接口支持查询超时空地图配置 perf(socket): 优化XORDecryptU解密函数减少内存分配 基于bytebufferpool实现缓冲区池化，大幅降低高频调用下的 内存分配和GC压力，提升性能表现 refactor(utils): 优化packVal序列化函数提升性能和稳定性 减少反射开销，
347 lines
8.9 KiB
Go
347 lines
8.9 KiB
Go
package socket
|
||
|
||
import (
|
||
"context"
|
||
"encoding/binary"
|
||
"errors"
|
||
"fmt"
|
||
"io"
|
||
"log"
|
||
"os"
|
||
"sync/atomic"
|
||
"time"
|
||
|
||
"blazing/cool"
|
||
"blazing/logic/service/common"
|
||
"blazing/logic/service/player"
|
||
"blazing/modules/config/service"
|
||
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/os/gtime"
|
||
"github.com/panjf2000/gnet/v2"
|
||
"github.com/valyala/bytebufferpool"
|
||
)
|
||
|
||
func (s *Server) Boot(serverid, port uint16) error {
|
||
// go s.bootws()
|
||
s.serverid = serverid
|
||
s.port = port
|
||
|
||
err := gnet.Run(s, s.network+"://"+s.addr,
|
||
gnet.WithMulticore(true),
|
||
gnet.WithTicker(true),
|
||
|
||
// 其他调优配置↓
|
||
gnet.WithTCPNoDelay(gnet.TCPNoDelay), // 禁用Nagle算法（降低延迟，适合小数据包场景）
|
||
//gnet.WithReusePort(true), // 开启SO_REUSEPORT（多核下提升并发）
|
||
//gnet.WithReadBufferCap(1024*64), // 读缓冲区64KB（根据业务调整，默认太小）
|
||
// gnet.WithWriteBufferCap(1024*64), // 写缓冲区64KB
|
||
|
||
//gnet.WithLockOSThread(true), // 绑定goroutine到OS线程（减少上下文切换）
|
||
)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func (s *Server) Stop() error {
|
||
_ = s.eng.Stop(context.Background())
|
||
s.workerPool.Release()
|
||
|
||
return nil
|
||
}
|
||
|
||
func (s *Server) OnClose(c gnet.Conn, err error) (action gnet.Action) {
|
||
defer func() {
|
||
if err := recover(); err != nil { // 恢复 panic，err 为 panic 错误值
|
||
// 1. 打印错误信息
|
||
if t, ok := c.Context().(*player.ClientData); ok {
|
||
if t.Player != nil {
|
||
if t.Player.Info != nil {
|
||
cool.Logger.Error(context.TODO(), "OnClose 错误：", cool.Config.ServerInfo.OnlineID, t.Player.Info.UserID, err)
|
||
t.Player.Service.Info.Save(*t.Player.Info)
|
||
}
|
||
|
||
}
|
||
|
||
} else {
|
||
cool.Logger.Error(context.TODO(), "OnClose 错误：", cool.Config.ServerInfo.OnlineID, err)
|
||
|
||
}
|
||
|
||
}
|
||
}()
|
||
// 识别 RST 导致的连接中断（错误信息含 "connection reset"）
|
||
// if err != nil && (strings.Contains(err.Error(), "connection reset") || strings.Contains(err.Error(), "reset by peer")) {
|
||
// remoteIP := c.RemoteAddr().(*net.TCPAddr).IP.String()
|
||
|
||
// log.Printf("RST 攻击检测: 来源 %s, 累计攻击次数 %d", remoteIP)
|
||
|
||
// // 防护逻辑：临时封禁异常 IP（可扩展为 IP 黑名单）
|
||
// // go s.tempBlockIP(remoteIP, 5*time.Minute)
|
||
// }
|
||
//fmt.Println(err, c.RemoteAddr().String(), "断开连接")
|
||
atomic.AddInt64(&cool.Connected, -1)
|
||
|
||
//logging.Infof("conn[%v] disconnected", c.RemoteAddr().String())
|
||
v, _ := c.Context().(*player.ClientData)
|
||
|
||
v.LF.Close()
|
||
if v.Player != nil {
|
||
v.Player.Save() //保存玩家数据
|
||
|
||
}
|
||
|
||
//}
|
||
//关闭连接
|
||
return
|
||
}
|
||
func (s *Server) OnTick() (delay time.Duration, action gnet.Action) {
|
||
g.Log().Async().Info(context.Background(), gtime.Now().ISO8601(), "服务器ID", cool.Config.ServerInfo.OnlineID, "链接数", atomic.LoadInt64(&cool.Connected))
|
||
if s.quit && atomic.LoadInt64(&cool.Connected) == 0 {
|
||
//执行正常退出逻辑
|
||
os.Exit(0)
|
||
}
|
||
return 30 * time.Second, gnet.None
|
||
}
|
||
func (s *Server) OnBoot(eng gnet.Engine) gnet.Action {
|
||
s.eng = eng
|
||
|
||
service.NewServerService().SetServerID(s.serverid, s.port) //设置当前服务器端口
|
||
return gnet.None
|
||
}
|
||
|
||
func (s *Server) OnOpen(conn gnet.Conn) (out []byte, action gnet.Action) {
|
||
if s.network != "tcp" {
|
||
return nil, gnet.Close
|
||
}
|
||
|
||
if conn.Context() == nil {
|
||
conn.SetContext(player.NewClientData(conn)) //注入data
|
||
}
|
||
|
||
atomic.AddInt64(&cool.Connected, 1)
|
||
|
||
return nil, gnet.None
|
||
}
|
||
|
||
func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
|
||
defer func() {
|
||
if err := recover(); err != nil { // 恢复 panic，err 为 panic 错误值
|
||
// 1. 打印错误信息
|
||
if t, ok := c.Context().(*player.ClientData); ok {
|
||
if t.Player != nil {
|
||
if t.Player.Info != nil {
|
||
cool.Logger.Error(context.TODO(), "OnTraffic 错误：", cool.Config.ServerInfo.OnlineID, t.Player.Info.UserID, err)
|
||
t.Player.Service.Info.Save(*t.Player.Info)
|
||
|
||
}
|
||
|
||
}
|
||
|
||
}
|
||
|
||
}
|
||
}()
|
||
|
||
ws := c.Context().(*player.ClientData).Wsmsg
|
||
if ws.Tcp { //升级失败时候防止缓冲区溢出
|
||
return s.handleTCP(c)
|
||
|
||
}
|
||
|
||
tt, len1 := ws.ReadBufferBytes(c)
|
||
if tt == gnet.Close {
|
||
|
||
return gnet.Close
|
||
}
|
||
|
||
ok, action := ws.Upgrade(c)
|
||
if action != gnet.None { //连接断开
|
||
return action
|
||
}
|
||
if !ok { //升级失败,说明是tcp连接
|
||
ws.Tcp = true
|
||
|
||
return s.handleTCP(c)
|
||
|
||
}
|
||
// fmt.Println(ws.Buf.Bytes())
|
||
c.Discard(len1)
|
||
|
||
messages, err := ws.Decode(c)
|
||
if err != nil {
|
||
return gnet.Close
|
||
}
|
||
if messages == nil {
|
||
return
|
||
}
|
||
|
||
for _, msg := range messages {
|
||
|
||
s.onevent(c, msg.Payload)
|
||
//t.OnEvent(msg.Payload)
|
||
}
|
||
|
||
return gnet.None
|
||
}
|
||
|
||
const maxBodyLen = 10 * 1024 // 业务最大包体长度，按需调整
|
||
func (s *Server) handleTCP(conn gnet.Conn) (action gnet.Action) {
|
||
|
||
conn.Context().(*player.ClientData).IsCrossDomain.Do(func() { //跨域检测
|
||
handle(conn)
|
||
})
|
||
|
||
// handle(c)
|
||
// 先读取4字节的包长度
|
||
lenBuf, err := conn.Peek(4)
|
||
|
||
if err != nil {
|
||
if errors.Is(err, io.ErrShortBuffer) {
|
||
return
|
||
}
|
||
return gnet.Close
|
||
}
|
||
|
||
bodyLen := binary.BigEndian.Uint32(lenBuf)
|
||
|
||
if bodyLen > maxBodyLen {
|
||
return gnet.Close
|
||
}
|
||
|
||
if conn.InboundBuffered() < int(bodyLen) {
|
||
return
|
||
}
|
||
// 提取包体
|
||
body, err := conn.Next(int(bodyLen))
|
||
if err != nil {
|
||
if errors.Is(err, io.ErrShortBuffer) {
|
||
return
|
||
}
|
||
return gnet.Close
|
||
}
|
||
|
||
s.onevent(conn, body)
|
||
|
||
if conn.InboundBuffered() > 0 {
|
||
if err := conn.Wake(nil); err != nil { // wake up the connection manually to avoid missing the leftover data
|
||
|
||
return gnet.Close
|
||
}
|
||
}
|
||
return action
|
||
|
||
}
|
||
|
||
// CROSS_DOMAIN 定义跨域策略文件内容
|
||
const CROSS_DOMAIN = "<?xml version=\"1.0\"?><!DOCTYPE cross-domain-policy><cross-domain-policy><allow-access-from domain=\"*\" to-ports=\"*\" /></cross-domain-policy>\x00"
|
||
|
||
// TEXT 定义跨域请求的文本格式
|
||
const TEXT = "<policy-file-request/>\x00"
|
||
|
||
func handle(c gnet.Conn) {
|
||
|
||
// 读取数据并检查是否为跨域请求
|
||
data, err := c.Peek(len(TEXT))
|
||
if err != nil {
|
||
log.Printf("Error reading cross-domain request: %v", err)
|
||
return
|
||
}
|
||
|
||
if string(data) == TEXT { //判断是否是跨域请求
|
||
//log.Printf("Received cross-domain request from %s", c.RemoteAddr())
|
||
// 处理跨域请求
|
||
c.Write([]byte(CROSS_DOMAIN))
|
||
c.Discard(len(TEXT))
|
||
|
||
return
|
||
}
|
||
|
||
//return
|
||
}
|
||
|
||
func (s *Server) onevent(c gnet.Conn, v []byte) {
|
||
if t, ok := c.Context().(*player.ClientData); ok {
|
||
var header common.TomeeHeader
|
||
// 解析Len（0-3字节）
|
||
header.Len = binary.BigEndian.Uint32(v[0:4])
|
||
// 解析Version（第4字节）
|
||
//header.Version = v[4]
|
||
// 解析CMD（5-8字节）
|
||
header.CMD = binary.BigEndian.Uint32(v[5:9])
|
||
// 解析UserID（9-12字节）
|
||
header.UserID = binary.BigEndian.Uint32(v[9:13])
|
||
// 解析Result（13-16字节）
|
||
//header.Result = binary.BigEndian.Uint32(v[13:17])
|
||
// 解析数据部分（17字节之后）
|
||
header.Data = make([]byte, 0)
|
||
if len(v) > 17 {
|
||
header.Data = v[17:]
|
||
|
||
}
|
||
if header.CMD > 1001 {
|
||
if t.Player == nil {
|
||
fmt.Println(header.UserID, "账号未注册")
|
||
return
|
||
}
|
||
if t.Player.Info == nil {
|
||
fmt.Println(header.UserID, "未创建角色")
|
||
return
|
||
}
|
||
if len(header.Data) > 0 {
|
||
header.Data = XORDecryptU(header.Data, t.Player.Hash)
|
||
}
|
||
|
||
}
|
||
if cool.Config.ServerInfo.IsDebug != 0 {
|
||
fmt.Println("接收数据", header.UserID, header.CMD)
|
||
}
|
||
|
||
t.LF.Producer().Write(header)
|
||
|
||
}
|
||
}
|
||
|
||
// XORDecryptU 优化后的异或解密：减少内存分配，支持复用缓冲区
|
||
// XORDecryptU 基于bytebufferpool优化的异或解密函数
|
||
// 保留原有接口，无侵入式优化，高频调用下大幅减少内存分配和GC
|
||
func XORDecryptU(encryptedData []byte, key uint32) []byte {
|
||
if len(encryptedData) == 0 {
|
||
return []byte{}
|
||
}
|
||
|
||
// 1. 栈上分配密钥字节数组（无GC压力，保留原优化）
|
||
var keyBytes [4]byte
|
||
binary.BigEndian.PutUint32(keyBytes[:], key)
|
||
keyLen := len(keyBytes)
|
||
|
||
// 2. 从bytebufferpool获取池化缓冲区（替代make分配）
|
||
buf := bytebufferpool.Get()
|
||
defer bytebufferpool.Put(buf) // 函数结束自动归还缓冲区到池
|
||
|
||
// 3. 调整缓冲区长度，匹配待解密数据（避免扩容）
|
||
buf.B = buf.B[:0] // 清空原有数据，保留底层数组
|
||
if cap(buf.B) < len(encryptedData) {
|
||
// 若缓冲区容量不足，直接扩容（bytebufferpool会自动管理）
|
||
buf.B = make([]byte, len(encryptedData))
|
||
} else {
|
||
// 容量足够，直接调整长度
|
||
buf.B = buf.B[:len(encryptedData)]
|
||
}
|
||
|
||
// 4. 核心异或解密逻辑（直接操作buf.B，无额外内存分配）
|
||
decrypted := buf.B
|
||
for i, b := range encryptedData {
|
||
decrypted[i] = b ^ keyBytes[i%keyLen]
|
||
}
|
||
|
||
// 5. 拷贝结果（关键：避免返回池化缓冲区，防止被后续调用覆盖）
|
||
result := make([]byte, len(decrypted))
|
||
copy(result, decrypted)
|
||
|
||
return result
|
||
}
|