Add logger

Author: tkc

example/health-check.json

@@ -2,14 +2,15 @@ package main
 
 import (
 	"bytes"
-	"encoding/csv"
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
 	"log"
 	"net/http"
 	"os"
 	"strconv"
+
+	"github.com/tkc/go-json-server/src/logger"
 )
 
 const (
@@ -127,7 +128,7 @@ func main() {
 func response(w http.ResponseWriter, r *http.Request) {
 
 	r.ParseForm()
-	accessLog(r)
+	logger.AccessLog(r)
 
 	w.Header().Set("Access-Control-Allow-Origin", "*")
 	w.Header().Set("Access-Control-Allow-Credentials", "true")
@@ -159,15 +160,3 @@ func path2Response(path string) string {
 	buf.ReadFrom(file)
 	return buf.String()
 }
-
-func accessLog(r *http.Request) {
-	file, err := os.OpenFile("log.csv", os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
-	if err != nil {
-		log.Fatal(err)
-	}
-	defer file.Close()
-	s := []string{r.Method, r.Host, r.Proto, r.RequestURI, r.RemoteAddr}
-	writer := csv.NewWriter(file)
-	writer.Write(s)
-	writer.Flush()
-}

go-json-server.go

@@ -0,0 +1,8 @@
+module github.com/tkc/go-json-server
+
+go 1.13
+
+require (
+	github.com/spf13/cast v1.3.1
+	github.com/stretchr/testify v1.5.1
+)

go.mod

@@ -0,0 +1,14 @@
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/spf13/cast v1.3.1 h1:nFm6S0SMdyzrzcmThSipiEubIDy8WEXKNZ0UOgiRpng=
+github.com/spf13/cast v1.3.1/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+github.com/stretchr/testify v1.5.1 h1:nOGnQDM7FYENwehXlg/kFVnos3rEvtKTjRvOWSzb6H4=
+github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
+gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

go.sum

@@ -0,0 +1,87 @@
+package logger
+
+import (
+	"encoding/csv"
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	stdlog "log"
+	"net/http"
+	"os"
+	"strconv"
+)
+
+type stdLogger struct {
+	stderr *stdlog.Logger
+	stdout *stdlog.Logger
+}
+
+func CreateLogger() *stdLogger {
+	return &stdLogger{
+		stdout: stdlog.New(os.Stdout, "", 0),
+		stderr: stdlog.New(os.Stderr, "", 0),
+	}
+}
+
+func (l *stdLogger) AccessLog(r *http.Request) {
+	file, err := os.OpenFile("log.csv", os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	defer file.Close()
+
+	jsonBody := dumpJsonBoddy(r)
+
+	s := []string{r.Method, r.Host, r.Proto, r.RequestURI, r.RemoteAddr, jsonBody}
+	writer := csv.NewWriter(file)
+	writer.Write(s)
+	writer.Flush()
+}
+
+func dumpJsonBoddy(req *http.Request) string {
+
+	if req.Method == "GET" {
+		return ""
+	}
+
+	if req.Header.Get("Content-Type") != "application/json" {
+		return ""
+	}
+
+	length, err := strconv.Atoi(req.Header.Get("Content-Length"))
+
+	if err != nil {
+		return ""
+	}
+
+	body := make([]byte, length)
+	length, err = req.Body.Read(body)
+
+	if err != nil && err != io.EOF {
+		return ""
+	}
+
+	var jsonBody map[string]interface{}
+	err = json.Unmarshal(body[:length], &jsonBody)
+
+	if err != nil {
+		return ""
+	}
+
+	s := fmt.Sprintf("%v", jsonBody)
+	return s
+}
+
+func (l *stdLogger) Printf(format string, args ...interface{}) {
+	l.stdout.Printf(format, args...)
+}
+
+func (l *stdLogger) Errorf(format string, args ...interface{}) {
+	l.stderr.Printf(format, args...)
+}
+
+func (l *stdLogger) Fatalf(format string, args ...interface{}) {
+	l.stderr.Fatalf(format, args...)
+}

src/logger/logger.go

@@ -0,0 +1,26 @@
+package logger
+
+import (
+	"bytes"
+	"net/http"
+	"testing"
+
+	"github.com/spf13/cast"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetNonReadRecord(t *testing.T) {
+
+	url := "example.com"
+	content := `{"integer":1,"string":"xyz", "object": { "element": 1 } , "array": [1, 2, 3]}`
+	byte := []byte(content)
+
+	body := bytes.NewReader(byte)
+	req, _ := http.NewRequest("POST", url, body)
+
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Content-Length", cast.ToString(len(content)))
+
+	result := dumpJsonBoddy(req)
+	assert.Equal(t, result, "map[array:[1 2 3] integer:1 object:map[element:1] string:xyz]")
+}

src/logger/logger_test.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"strconv"
+)
+
+func dumpJsonRequestHandlerFunc(w http.ResponseWriter, req *http.Request) {
+
+	if req.Header.Get("Content-Type") != "application/json" {
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	length, err := strconv.Atoi(req.Header.Get("Content-Length"))
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	log.Print(length)
+
+	body := make([]byte, length)
+	length, err = req.Body.Read(body)
+
+	if err != nil && err != io.EOF {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	var jsonBody map[string]interface{}
+	err = json.Unmarshal(body[:length], &jsonBody)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	fmt.Printf("%v\n", jsonBody)
+	// w.WriteHeader(http.StatusOK)
+}
+
+func main() {
+	http.HandleFunc("/json", dumpJsonRequestHandlerFunc)
+	http.ListenAndServe(":8080", nil)
+}