PR #321: Cleanup TCP proxy connection

This patch updates the conns map when the connection is closed.

proxy/tcp/server.go

@@ -80,7 +80,20 @@ func (s *Server) Serve(l net.Listener) error {
 		}
 		s.conns[c] = true
 		s.mu.Unlock()
-		go s.Handler.ServeTCP(c)
+
+		go func() {
+			defer func() {
+				// Explicitly close the connection
+				c.Close()
+
+				// Delete the connection from the conns map
+				s.mu.Lock()
+				delete(s.conns, c)
+				s.mu.Unlock()
+			}()
+
+			s.Handler.ServeTCP(c)
+		}()
 	}
 }