log

cmd/link.go

@@ -2,11 +2,11 @@ package cmd
 
 import (
 	"fmt"
-	"log"
 	"net"
 	"sync"
 
 	"github.com/gorilla/websocket"
+	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -57,7 +57,7 @@ func startTCPTunnel(vhost, hostPort string) error {
 		var msg TunnelMessage
 		err := conn.ReadJSON(&msg)
 		if err != nil {
-			log.Printf("Error reading from tunnel: %v", err)
+			log.Errorf("Error reading from tunnel: %v", err)
 			break
 		}
 
@@ -71,7 +71,7 @@ func startTCPTunnel(vhost, hostPort string) error {
 				// New connection
 				localConn, err = net.Dial("tcp", hostPort)
 				if err != nil {
-					log.Printf("Failed to connect to %s: %v", hostPort, err)
+					log.Errorf("Failed to connect to %s: %v", hostPort, err)
 					continue
 				}
 
@@ -110,7 +110,7 @@ func startTCPTunnel(vhost, hostPort string) error {
 
 			// Write data to local connection
 			if _, err := localConn.Write(msg.Data); err != nil {
-				log.Printf("Error writing to local connection: %v", err)
+				log.Errorf("Error writing to local connection: %v", err)
 				localConn.Close()
 				connMutex.Lock()
 				delete(connections, msg.StreamID)