Add debug logging for client connection accounting

IngmarStein · May 24, 2024 · b582e51 · b582e51
1 parent 053534c
commit b582e51
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/pkg/multiplexer/multiplexer.go b/pkg/multiplexer/multiplexer.go
@@ -93,6 +93,9 @@ L:
 
 func (mux *Multiplexer) handleConnection(conn net.Conn, sender chan<- *reqContainer) {
 	defer func(c net.Conn) {
+		if logrus.IsLevelEnabled(logrus.DebugLevel) {
+			logrus.Debugf("Closing client connecion: %v", c.RemoteAddr())
+		}
 		err := c.Close()
 		sender <- &reqContainer{typ: Disconnection}
 		if err != nil {
@@ -115,7 +118,7 @@ func (mux *Multiplexer) handleConnection(conn net.Conn, sender chan<- *reqContai
 		}
 
 		if logrus.IsLevelEnabled(logrus.DebugLevel) {
-			logrus.Debug("Message from Client...")
+			logrus.Debug("Message from client...")
 			spew.Dump(msg)
 		}
 
@@ -166,10 +169,17 @@ func (mux *Multiplexer) targetConnLoop(requestQueue <-chan *reqContainer) {
 		switch container.typ {
 		case Connection:
 			clients++
+			if logrus.IsLevelEnabled(logrus.DebugLevel) {
+				logrus.Debugf("Connected clients: %d", clients)
+			}
 			continue
 		case Disconnection:
 			clients--
+			if logrus.IsLevelEnabled(logrus.DebugLevel) {
+				logrus.Debugf("Connected clients: %d", clients)
+			}
 			if clients == 0 && conn != nil {
+				logrus.Info("closing target connection")
 				err := conn.Close()
 				if err != nil {
 					logrus.Error(err)