Просмотр исходного кода

Merge pull request #1513 from justincormack/no-logrus-proxy

Do not use logrus for proxy
Jana Radhakrishnan 8 лет назад
Родитель
Сommit
3f28b09ad4
2 измененных файлов с 7 добавлено и 9 удалено
  1. 3 4
      libnetwork/cmd/proxy/tcp_proxy.go
  2. 4 5
      libnetwork/cmd/proxy/udp_proxy.go

+ 3 - 4
libnetwork/cmd/proxy/tcp_proxy.go

@@ -2,11 +2,10 @@ package main
 
 
 import (
 import (
 	"io"
 	"io"
+	"log"
 	"net"
 	"net"
 	"sync"
 	"sync"
 	"syscall"
 	"syscall"
-
-	"github.com/Sirupsen/logrus"
 )
 )
 
 
 // TCPProxy is a proxy for TCP connections. It implements the Proxy interface to
 // TCPProxy is a proxy for TCP connections. It implements the Proxy interface to
@@ -35,7 +34,7 @@ func NewTCPProxy(frontendAddr, backendAddr *net.TCPAddr) (*TCPProxy, error) {
 func (proxy *TCPProxy) clientLoop(client *net.TCPConn, quit chan bool) {
 func (proxy *TCPProxy) clientLoop(client *net.TCPConn, quit chan bool) {
 	backend, err := net.DialTCP("tcp", nil, proxy.backendAddr)
 	backend, err := net.DialTCP("tcp", nil, proxy.backendAddr)
 	if err != nil {
 	if err != nil {
-		logrus.Printf("Can't forward traffic to backend tcp/%v: %s\n", proxy.backendAddr, err)
+		log.Printf("Can't forward traffic to backend tcp/%v: %s\n", proxy.backendAddr, err)
 		client.Close()
 		client.Close()
 		return
 		return
 	}
 	}
@@ -79,7 +78,7 @@ func (proxy *TCPProxy) Run() {
 	for {
 	for {
 		client, err := proxy.listener.Accept()
 		client, err := proxy.listener.Accept()
 		if err != nil {
 		if err != nil {
-			logrus.Printf("Stopping proxy on tcp/%v for tcp/%v (%s)", proxy.frontendAddr, proxy.backendAddr, err)
+			log.Printf("Stopping proxy on tcp/%v for tcp/%v (%s)", proxy.frontendAddr, proxy.backendAddr, err)
 			return
 			return
 		}
 		}
 		go proxy.clientLoop(client.(*net.TCPConn), quit)
 		go proxy.clientLoop(client.(*net.TCPConn), quit)

+ 4 - 5
libnetwork/cmd/proxy/udp_proxy.go

@@ -2,13 +2,12 @@ package main
 
 
 import (
 import (
 	"encoding/binary"
 	"encoding/binary"
+	"log"
 	"net"
 	"net"
 	"strings"
 	"strings"
 	"sync"
 	"sync"
 	"syscall"
 	"syscall"
 	"time"
 	"time"
-
-	"github.com/Sirupsen/logrus"
 )
 )
 
 
 const (
 const (
@@ -112,7 +111,7 @@ func (proxy *UDPProxy) Run() {
 			// ECONNREFUSED like Read do (see comment in
 			// ECONNREFUSED like Read do (see comment in
 			// UDPProxy.replyLoop)
 			// UDPProxy.replyLoop)
 			if !isClosedError(err) {
 			if !isClosedError(err) {
-				logrus.Printf("Stopping proxy on udp/%v for udp/%v (%s)", proxy.frontendAddr, proxy.backendAddr, err)
+				log.Printf("Stopping proxy on udp/%v for udp/%v (%s)", proxy.frontendAddr, proxy.backendAddr, err)
 			}
 			}
 			break
 			break
 		}
 		}
@@ -123,7 +122,7 @@ func (proxy *UDPProxy) Run() {
 		if !hit {
 		if !hit {
 			proxyConn, err = net.DialUDP("udp", nil, proxy.backendAddr)
 			proxyConn, err = net.DialUDP("udp", nil, proxy.backendAddr)
 			if err != nil {
 			if err != nil {
-				logrus.Printf("Can't proxy a datagram to udp/%s: %s\n", proxy.backendAddr, err)
+				log.Printf("Can't proxy a datagram to udp/%s: %s\n", proxy.backendAddr, err)
 				proxy.connTrackLock.Unlock()
 				proxy.connTrackLock.Unlock()
 				continue
 				continue
 			}
 			}
@@ -134,7 +133,7 @@ func (proxy *UDPProxy) Run() {
 		for i := 0; i != read; {
 		for i := 0; i != read; {
 			written, err := proxyConn.Write(readBuf[i:read])
 			written, err := proxyConn.Write(readBuf[i:read])
 			if err != nil {
 			if err != nil {
-				logrus.Printf("Can't proxy a datagram to udp/%s: %s\n", proxy.backendAddr, err)
+				log.Printf("Can't proxy a datagram to udp/%s: %s\n", proxy.backendAddr, err)
 				break
 				break
 			}
 			}
 			i += written
 			i += written