昔念
666557416d
- 在 ClientData 中添加 sync.Mutex 以确保并发安全 - 实现 SetPlayer 和 GetPlayer 等方法来封装 player 字段的操作 - 更新相关代码以使用新的 ClientData 结构和方法
140 lines
3.2 KiB
Go
140 lines
3.2 KiB
Go
package socket
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
"blazing/common/data/entity"
|
|
|
|
"github.com/gogf/gf/v2/os/glog"
|
|
"github.com/panjf2000/gnet/v2"
|
|
"github.com/panjf2000/gnet/v2/pkg/logging"
|
|
)
|
|
|
|
func (s *Server) Boot() error {
|
|
err := gnet.Run(s, s.addr,
|
|
gnet.WithMulticore(true),
|
|
gnet.WithSocketRecvBuffer(s.bufferSize))
|
|
if err != nil {
|
|
return 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, _ error) (action gnet.Action) {
|
|
//fmt.Println(glog.GetStack(), c.Context().(*entity.ClientData).Player.UserID, "断开连接")
|
|
//fmt.Println("远程地址", c.RemoteAddr(), c)
|
|
fmt.Printf("关闭函数地址: %p\n", c.Context()) // 输出类似: 0x1040a1390
|
|
//if c.Context() != nil {
|
|
v := c.Context().(*entity.ClientData)
|
|
t := v.GetPlayer()
|
|
if t != nil {
|
|
glog.Debug(context.Background(), t.UserID, "断开连接")
|
|
|
|
}
|
|
|
|
//}
|
|
//关闭连接
|
|
return
|
|
}
|
|
func (s *Server) OnBoot(eng gnet.Engine) gnet.Action {
|
|
s.eng = eng
|
|
|
|
logging.Infof("syslog server is listening on %s\n", s.addr)
|
|
|
|
return gnet.None
|
|
}
|
|
|
|
func (s *Server) OnTraffic(conn gnet.Conn) (action gnet.Action) {
|
|
if conn.Context() == nil {
|
|
conn.SetContext(entity.NewClientData()) //注入data
|
|
}
|
|
|
|
if s.network == "tcp" {
|
|
return s.handleTcp(conn)
|
|
}
|
|
|
|
return gnet.None
|
|
}
|
|
|
|
func (s *Server) handleTcp(conn gnet.Conn) (action gnet.Action) {
|
|
|
|
for {
|
|
if !s.discorse {
|
|
handle(conn)
|
|
}
|
|
data, err := s.codec.Decode(conn)
|
|
if err != nil {
|
|
break
|
|
}
|
|
|
|
//client := conn.RemoteAddr().String()
|
|
_ = s.workerPool.Submit(func() { //TODO 这里可能存在顺序执行问题,待修复
|
|
s.parser(conn, data)
|
|
})
|
|
|
|
return gnet.None
|
|
}
|
|
|
|
if conn.InboundBuffered() > 0 {
|
|
if err := conn.Wake(nil); err != nil { // wake up the connection manually to avoid missing the leftover data
|
|
logging.Errorf("failed to wake up the connection, %v", err)
|
|
return gnet.Close
|
|
}
|
|
}
|
|
|
|
return gnet.None
|
|
|
|
}
|
|
|
|
func (s *Server) parser(c gnet.Conn, line []byte) {
|
|
//todo 这里待实现注入player实体
|
|
s.handler.Handle(c, line)
|
|
}
|
|
func (s *Server) Start() {
|
|
|
|
err := gnet.Run(s, s.network+"://"+s.addr, gnet.WithMulticore(s.multicore))
|
|
logging.Infof("server exits with error: %v", err)
|
|
}
|
|
|
|
// 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) {
|
|
clientdata := c.Context().(*entity.ClientData)
|
|
if clientdata.GetIsCrossDomain() {
|
|
return
|
|
|
|
}
|
|
// 读取数据并检查是否为跨域请求
|
|
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))
|
|
|
|
clientdata.SetCrossDomain(true) //= true //TODO 待修复未成功切换bug
|
|
return
|
|
}
|
|
|
|
//return
|
|
}
|