cleaning up log handling with closures

cmd/echo/main.go

@@ -82,6 +82,6 @@ func main() {
 			log.Fatal(err)
 		}
 
-		go ts.Handle(conn)
+		go ts.Handle(wrappedSlogger)(conn)
 	}
 }

telnet/server.go

@@ -43,27 +43,29 @@ func (ts *TELNETServer) Accept() (net.Conn, error) {
 	return conn, err
 }
 
-func (ts *TELNETServer) Handle(conn net.Conn) {
-	s := bufio.NewScanner(conn)
-	defer conn.Close()
-	// i need to create an ecs Renderer that targets a iw.Writer (net.Conn)
-	// the renderer
-
-	correlationID := watermill.NewUUID()
-	ts.logger = ts.logger.With(
+func (ts *TELNETServer) Handle(wml watermill.LoggerAdapter) func(conn net.Conn) {
+	logger := wml.With(
 		watermill.LogFields{
-			"correlation_id": correlationID,
+			// "correlation_id": correlationID,
 		},
 	)
+	return func(conn net.Conn) {
+		s := bufio.NewScanner(conn)
+		defer conn.Close()
+		// i need to create an ecs Renderer that targets a iw.Writer (net.Conn)
+		// the renderer
+
+		correlationID := watermill.NewUUID()
 
-	for s.Scan() {
-		t := s.Text()
-		m := message.NewMessage(watermill.NewUUID(), []byte(t))
-		middleware.SetCorrelationID(correlationID, m)
-		err := ts.p.Publish("telnet.raw", m)
-		if err != nil {
-			log.Fatalln("couldn't write to telnet.raw")
+		for s.Scan() {
+			t := s.Text()
+			m := message.NewMessage(watermill.NewUUID(), []byte(t))
+			middleware.SetCorrelationID(correlationID, m)
+			err := ts.p.Publish("telnet.raw", m)
+			if err != nil {
+				log.Fatalln("couldn't write to telnet.raw")
+			}
+			logger.Trace("received bytes over telnet", nil)
 		}
-		ts.logger.Trace("received bytes over telnet", nil)
 	}
 }