Override http server's error logger

Signed-off-by: Doğukan Teber <[email protected]>

Author: dogukanteber

gateway/proxy.go

@@ -11,6 +11,7 @@ import (
 	"os"
 	"time"
 
+	"github.com/cortexproject/auth-gateway/utils"
 	"github.com/sirupsen/logrus"
 )
 
@@ -54,13 +55,6 @@ type Proxy struct {
 	reverseProxy *httputil.ReverseProxy
 }
 
-type logrusErrorWriter struct{}
-
-func (w logrusErrorWriter) Write(p []byte) (n int, err error) {
-	logrus.Errorf("%s", string(p))
-	return len(p), nil
-}
-
 func NewProxy(targetURL string, upstream Upstream, component string) (*Proxy, error) {
 	url, err := url.Parse(targetURL)
 	if err != nil {
@@ -74,7 +68,7 @@ func NewProxy(targetURL string, upstream Upstream, component string) (*Proxy, er
 	reverseProxy.Transport = customTransport(component, upstream)
 	originalDirector := reverseProxy.Director
 	reverseProxy.Director = customDirector(url, originalDirector)
-	reverseProxy.ErrorLog = log.New(logrusErrorWriter{}, "", 0)
+	reverseProxy.ErrorLog = log.New(utils.LogrusErrorWriter{}, "", 0)
 
 	if upstream.HTTPClientTimeout == 0 {
 		upstream.HTTPClientTimeout = defaultTimeoutValues[component].HTTPClientTimeout

server/server.go

@@ -3,12 +3,14 @@ package server
 import (
 	"context"
 	"fmt"
+	"log"
 	"net"
 	"net/http"
 	"net/http/pprof"
 	"time"
 
 	"github.com/cortexproject/auth-gateway/middleware"
+	"github.com/cortexproject/auth-gateway/utils"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promauto"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
@@ -102,6 +104,7 @@ func initAuthServer(cfg *Config, middlewares []middleware.Interface) (*server, e
 		ReadTimeout:  readTimeout,
 		WriteTimeout: writeTimeout,
 		IdleTimeout:  idleTimeout,
+		ErrorLog:     log.New(utils.LogrusErrorWriter{}, "", 0),
 	}
 
 	return &server{
@@ -157,6 +160,7 @@ func initUnAuthServer(cfg *Config, middlewares []middleware.Interface) (*server,
 		ReadTimeout:  readTimeout,
 		WriteTimeout: writeTimeout,
 		IdleTimeout:  idleTimeout,
+		ErrorLog:     log.New(utils.LogrusErrorWriter{}, "", 0),
 	}
 
 	return &server{
@@ -218,7 +222,7 @@ func New(cfg Config) (*Server, error) {
 }
 
 func (s *Server) Run() error {
-	fmt.Println("server has started")
+	logrus.Infof("the server has started listening on %v", s.authServer.httpServer.Addr)
 	errChan := make(chan error, 1)
 
 	go func() {

utils/utils.go

@@ -0,0 +1,10 @@
+package utils
+
+import "github.com/sirupsen/logrus"
+
+type LogrusErrorWriter struct{}
+
+func (w LogrusErrorWriter) Write(p []byte) (n int, err error) {
+	logrus.Errorf("%s", string(p))
+	return len(p), nil
+}