Add request log

Signed-off-by: Viet-Anh Duong <vietanhs0817@gmail.com>

Author: mipsel64

hander.go

@@ -2,18 +2,21 @@ package main
 
 import (
 	"encoding/json"
+	"fmt"
 	"html/template"
+	"io/ioutil"
 	"log"
 	"net/http"
 	"strconv"
+	"strings"
 )
 
 type Response struct {
 	ErrorCode   int    `json:"error_code"`
 	RequestId   string `json:"request_id"`
 	OriginalURI string `json:"original_uri"`
-	RayId       string `json:"ray_id"`   
-	ClientIp 		string `json:"client_ip"`
+	RayId       string `json:"ray_id"`
+	ClientIp    string `json:"client_ip"`
 	Message     string `json:"message"`
 }
 
@@ -27,12 +30,14 @@ var ErrorMap = map[int]string{
 
 func errorHandler(t *template.Template) func(http.ResponseWriter, *http.Request) {
 	return func(w http.ResponseWriter, r *http.Request) {
+		log.Println("Header:", formatHeader(r.Header))
+		log.Printf("Body: %s\n", formatBody(r))
+
 		w.Header().Set(FormatHeader, r.Header.Get(FormatHeader))
 		w.Header().Set(CodeHeader, r.Header.Get(CodeHeader))
 		w.Header().Set(ContentType, r.Header.Get(ContentType))
 		w.Header().Set(OriginalURI, r.Header.Get(OriginalURI))
 		w.Header().Set(RequestId, r.Header.Get(RequestId))
-
 		format := r.Header.Get(FormatHeader)
 		if format != "application/json" {
 			format = DefaultFormat
@@ -57,7 +62,7 @@ func errorHandler(t *template.Template) func(http.ResponseWriter, *http.Request)
 			RequestId:   r.Header.Get(RequestId),
 			OriginalURI: r.Header.Get(OriginalURI),
 			ClientIp:    r.Header.Get(ClientIp),
-			RayId:  		 r.Header.Get(RayId),
+			RayId:       r.Header.Get(RayId),
 			Message:     message,
 		}
 
@@ -78,3 +83,20 @@ func errorHandler(t *template.Template) func(http.ResponseWriter, *http.Request)
 		}
 	}
 }
+
+func formatHeader(header http.Header) string {
+	var pairs []string
+	for key, value := range header {
+		pairs = append(pairs, fmt.Sprintf("%s=%s", key, value))
+	}
+	return strings.Join(pairs, "; ")
+}
+
+func formatBody(r *http.Request) string {
+	b, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		log.Printf("Read body request failed with err: %v\n", err)
+		return ""
+	}
+	return string(b)
+}