Http logger

utils/http/logger/logger.go

@@ -0,0 +1,56 @@
+package logger
+
+import (
+	"log"
+	"net/http"
+	"strings"
+	"time"
+)
+
+type ResponseWriter struct {
+	http.ResponseWriter
+	Status int
+}
+
+func (r *ResponseWriter) WriteHeader(status int) {
+	r.Status = status
+	r.ResponseWriter.WriteHeader(status)
+}
+
+func ClientIP(r *http.Request) string {
+	ips := ClientIPs(r)
+	if len(ips) >= 1 {
+		return ips[0]
+	}
+	return ""
+}
+
+func ClientIPs(r *http.Request) []string {
+	ra := r.RemoteAddr
+	if xff := strings.Trim(r.Header.Get("X-Forwarded-For"), " "); xff != "" {
+		ra = strings.Join([]string{xff, ra}, ",")
+	}
+	res := []string{}
+	ips := strings.Split(ra, ",")
+	for _, ip := range ips {
+		res = append(res, strings.Trim(ip, " "))
+	}
+	return res
+}
+
+func LogRequests(handler http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		start := time.Now()
+		nw := &ResponseWriter{
+			ResponseWriter: w,
+			Status:         http.StatusOK,
+		}
+		handler.ServeHTTP(nw, r)
+		duration := time.Since(start)
+
+		log.Printf(
+			"\"%s\" \"%s %s\" %d \"%.3f ms\"\n",
+			ClientIP(r), r.Method, r.URL, nw.Status, duration.Seconds(),
+		)
+	})
+}

utils/http/logger/logger_test.go

@@ -0,0 +1,113 @@
+package logger_test
+
+import (
+	"bytes"
+	"fmt"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	"github.com/vladimirok5959/golang-utils/utils/http/logger"
+)
+
+var _ = Describe("logger", func() {
+	Context("ClientIP", func() {
+		It("return client IP", func() {
+			Expect(logger.ClientIP(&http.Request{
+				RemoteAddr: "127.0.0.1",
+			})).To(Equal("127.0.0.1"))
+
+			Expect(logger.ClientIP(&http.Request{
+				RemoteAddr: "192.168.0.1,127.0.0.1",
+			})).To(Equal("192.168.0.1"))
+
+			Expect(logger.ClientIP(&http.Request{
+				RemoteAddr: "192.168.0.1, 127.0.0.1",
+			})).To(Equal("192.168.0.1"))
+
+			Expect(logger.ClientIP(&http.Request{
+				RemoteAddr: "192.168.0.50,192.168.0.1,127.0.0.1",
+			})).To(Equal("192.168.0.50"))
+
+			Expect(logger.ClientIP(&http.Request{
+				RemoteAddr: "192.168.0.50, 192.168.0.1, 127.0.0.1",
+			})).To(Equal("192.168.0.50"))
+		})
+	})
+
+	Context("ClientIPs", func() {
+		It("return array of client IPs", func() {
+			Expect(logger.ClientIPs(&http.Request{
+				RemoteAddr: "127.0.0.1",
+			})).To(ConsistOf(
+				"127.0.0.1",
+			))
+
+			Expect(logger.ClientIPs(&http.Request{
+				RemoteAddr: "192.168.0.1,127.0.0.1",
+			})).To(ConsistOf(
+				"192.168.0.1", "127.0.0.1",
+			))
+
+			Expect(logger.ClientIPs(&http.Request{
+				RemoteAddr: "192.168.0.1, 127.0.0.1",
+			})).To(ConsistOf(
+				"192.168.0.1", "127.0.0.1",
+			))
+
+			Expect(logger.ClientIPs(&http.Request{
+				RemoteAddr: "192.168.0.50,192.168.0.1,127.0.0.1",
+			})).To(ConsistOf(
+				"192.168.0.50", "192.168.0.1", "127.0.0.1",
+			))
+
+			Expect(logger.ClientIPs(&http.Request{
+				RemoteAddr: "192.168.0.50, 192.168.0.1, 127.0.0.1",
+			})).To(ConsistOf(
+				"192.168.0.50", "192.168.0.1", "127.0.0.1",
+			))
+		})
+	})
+
+	Context("LogRequests", func() {
+		var srv *httptest.Server
+		var client *http.Client
+
+		var getTestHandler = func() http.HandlerFunc {
+			return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				if _, err := w.Write([]byte("Index")); err != nil {
+					fmt.Printf("%s\n", err.Error())
+				}
+			})
+		}
+
+		BeforeEach(func() {
+			srv = httptest.NewServer(logger.LogRequests(getTestHandler()))
+			client = srv.Client()
+		})
+
+		AfterEach(func() {
+			srv.Close()
+		})
+
+		It("log http requests", func() {
+			buf := new(bytes.Buffer)
+			log.SetOutput(buf)
+
+			resp, err := client.Get(srv.URL + "/")
+			Expect(err).To(Succeed())
+			defer resp.Body.Close()
+
+			Expect(resp.StatusCode).To(Equal(http.StatusOK))
+			Expect(buf.String()).To(ContainSubstring("\"GET /\" 200"))
+		})
+	})
+})
+
+func TestSuite(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "logger")
+}