remove ws log
This commit is contained in:
parent
77bf0b2a00
commit
c472c9c078
@ -1,120 +0,0 @@
|
||||
// Package instance ...
|
||||
//
|
||||
// Description : instance ...
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 2025-04-25 18:16
|
||||
package instance
|
||||
|
||||
import (
|
||||
"git.zhangdeman.cn/zhangdeman/websocket"
|
||||
"git.zhangdeman.cn/zhangdeman/websocket/abstract"
|
||||
"git.zhangdeman.cn/zhangdeman/websocket/config"
|
||||
"git.zhangdeman.cn/zhangdeman/websocket/context"
|
||||
"git.zhangdeman.cn/zhangdeman/websocket/storage"
|
||||
"github.com/gin-gonic/gin"
|
||||
"io"
|
||||
)
|
||||
|
||||
// NewWebsocketLog 初始化长连接日志
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 18:12 2024/7/22
|
||||
func NewWebsocketLog(serverPort int, route *gin.Engine) (*WsLog, error) {
|
||||
wsLog := &WsLog{}
|
||||
if err := websocket.NewWebsocketServeWithGinRouter(serverPort, route, wsLog); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
wsServer, _ := websocket.GetWsServer(serverPort, wsLog.GetModuleFlag())
|
||||
// logger.SetWsLoggerConnect(wsServer.GetConfig().ConnectionManager)
|
||||
return &WsLog{
|
||||
port: serverPort,
|
||||
connectManager: wsServer.GetConfig().ConnectionManager,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// CloseWebsocketLog ...
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 13:27 2024/7/25
|
||||
func CloseWebsocketLog(wsLog *WsLog) {
|
||||
if nil == wsLog {
|
||||
return
|
||||
}
|
||||
wsServer, _ := websocket.GetWsServer(wsLog.port, wsLog.GetModuleFlag())
|
||||
allConnect := wsServer.GetConfig().ConnectionManager.GetCtxList()
|
||||
for _, conn := range allConnect {
|
||||
_ = conn.Session.Close()
|
||||
}
|
||||
}
|
||||
|
||||
type WsLog struct {
|
||||
port int
|
||||
connectManager storage.IConnection
|
||||
}
|
||||
|
||||
func (e WsLog) Connect(ctx *context.WSContext) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (e WsLog) Disconnect(ctx *context.WSContext) {
|
||||
}
|
||||
|
||||
func (e WsLog) Close(ctx *context.WSContext, code int, message string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (e WsLog) HandshakeURL() []string {
|
||||
return []string{
|
||||
"/system/logger/trace",
|
||||
}
|
||||
}
|
||||
|
||||
func (e WsLog) GetCommandList() []abstract.ICommand {
|
||||
return []abstract.ICommand{}
|
||||
}
|
||||
|
||||
func (e WsLog) GetModuleFlag() string {
|
||||
return "ws-log"
|
||||
}
|
||||
|
||||
func (e WsLog) GetServerPort() int {
|
||||
return e.port
|
||||
}
|
||||
|
||||
func (e WsLog) GetWSServerConfig() []config.SetWSServerConfig {
|
||||
return []config.SetWSServerConfig{
|
||||
config.SetPingPeriod(5),
|
||||
config.SetPongWait(60),
|
||||
config.SetWriteWait(60),
|
||||
config.SetMaxMessageBufferSize(81920),
|
||||
config.SetMaxMessageSize(81920),
|
||||
}
|
||||
}
|
||||
|
||||
func (e WsLog) Writer() io.Writer {
|
||||
return &wsWriter{
|
||||
connectionManager: e.connectManager,
|
||||
}
|
||||
}
|
||||
|
||||
type wsWriter struct {
|
||||
connectionManager storage.IConnection
|
||||
}
|
||||
|
||||
// Write ws的writer
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 22:24 2024/7/22
|
||||
func (w *wsWriter) Write(p []byte) (n int, err error) {
|
||||
p = []byte(" " + string(p))
|
||||
allConnList := w.connectionManager.GetCtxList()
|
||||
for _, conn := range allConnList {
|
||||
_ = conn.Session.Write(p)
|
||||
}
|
||||
return 0, nil
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user