netutil

type ClientInfo struct {
	ConnectionID string    // 业务用户ID
	ConnTime     time.Time //连接存活时长

}

type CloseTask struct {
	Connection *WsConnection
	Reason     string
	Done       chan struct{}
}

type HandleMsgTask struct {
	Connection  *WsConnection //回指向所属连接
	MessageType int           //消息类型
	Body        []byte        //消息载荷
	Retries     int           // 当前重试次数
}

type HttpOption = func(h *HttpServer)

type HttpServer struct {
	Server *gin.Engine
	Port   int
}

type ManagerOption = func(m *WsConnectionManager)

type NetLogger interface {
	Debug(expand *map[string]any, format string)
	Info(expand *map[string]any, format string)
	Error(expand *map[string]any, format string)
}

type OnCloseCallBack func(*WsConnection, string)

type OnMessageCallBack func(*WsConnection, int, []byte) error

type PingTask struct {
	Connection *WsConnection
	Retries    int // 当前重试次数
}

type SendMsgTask struct {
	Connection  *WsConnection //回指向所属连接
	MessageType int           //消息类型
	Body        []byte        //消息载荷
	Done        chan error    // 同步返回发送结果
	Retries     int           // 当前重试次数
}

type Task interface {
	Execute() // 执行任务
}

type WebSocketServer struct {
	Server        *HttpServer
	WsConnManager *WsConnectionManager
	WsUp          websocket.Upgrader
}

type WsConnection struct {
	Connection     *websocket.Conn      // 底层WebSocket连接
	PeerInfo       ClientInfo           // 客户端信息
	Owner          *WsConnectionManager // 关联的连接管理器
	MessageFunc    OnMessageCallBack    // 消息处理回调函数
	CloseFunc      OnCloseCallBack      // 关闭回调函数
	HeartbeatTimer *time.Timer          // 心跳定时器
	TaskPool       *ants.Pool           // 任务池
	IsClosed       atomic.Bool          // 连接关闭状态
	Wg             sync.WaitGroup       // 等待任务完成
}

type WsConnectionManager struct {
	IDMapConnection map[string]*WsConnection
	ConnectionMapID map[*WsConnection]ClientInfo
	Locker          sync.RWMutex
	MessageFunc     OnMessageCallBack // 消息处理回调函数
	CloseFunc       OnCloseCallBack   // 关闭回调函数
}

type WsOption = func(s *WebSocketServer)