Update log.go

xmidt-org · Nov 2, 2022 · 5d399a3 · 5d399a3
1 parent e9e49e6
commit 5d399a3
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/sallusthttp/log.go b/sallusthttp/log.go
@@ -8,14 +8,17 @@ import (
 	"go.uber.org/zap/zapcore"
 )
 
-func NewConnStateLogger(logger *zap.Logger, serverName string, lvl zapcore.Level) func(net.Conn, http.ConnState) {
+func NewConnStateLogger(logger *zap.Logger, lvl zapcore.Level, fs ...zap.Field) func(net.Conn, http.ConnState) {
 	return func(c net.Conn, cs http.ConnState) {
+		fs = append(fs, zap.String("connState", cs.String()))
+		if addr := c.LocalAddr(); addr != nil {
+			fs = append(fs, zap.String("localAddress", addr.String()))
+
+		}
 		logger.Log(
 			lvl,
 			"connState",
-			zap.String("serverName", serverName),
-			zap.String("localAddress", c.LocalAddr().String()),
-			zap.String("state", cs.String()),
+			fs...,
 		)
 	}
 }