ws logger 更改为全局注册

This commit is contained in:
2024-07-23 15:38:23 +08:00
parent dab0b5e0b9
commit d978913b67
3 changed files with 38 additions and 34 deletions

View File

@ -11,8 +11,6 @@ import (
"io"
"os"
"git.zhangdeman.cn/zhangdeman/websocket/storage"
"github.com/pkg/errors"
"go.uber.org/zap"
@ -61,7 +59,7 @@ func NewLogger(inputLoggerLevel string, splitConfig *RotateLogConfig, optionFunc
fileHandlerList := []zapcore.Core{
zapcore.NewCore(o.Encoder, zapcore.AddSync(loggerWriter), loggerLevelDeal),
zapcore.NewCore(o.Encoder, zapcore.AddSync(&wsWriter{o.WsLoggerConnect}), loggerLevelDeal), // 设置ws日志输出
zapcore.NewCore(o.Encoder, zapcore.AddSync(&wsWriter{}), loggerLevelDeal), // 设置ws日志输出
}
// 设置控制台输出
@ -141,7 +139,6 @@ func (l *Logger) getWriter() (io.Writer, error) {
}
type wsWriter struct {
connection storage.IConnection
}
// Write ws的writer
@ -150,11 +147,11 @@ type wsWriter struct {
//
// Date : 22:24 2024/7/22
func (w *wsWriter) Write(p []byte) (n int, err error) {
if nil == w.connection {
if nil == wsLoggerConnect {
return 0, nil
}
allConnList := w.connection.GetCtxList()
allConnList := wsLoggerConnect.GetCtxList()
for _, conn := range allConnList {
_ = conn.Session.Write(p)
}