whatismyip/internal/httputils/http.go

67 lines
1.2 KiB
Go
Raw Normal View History

2021-11-10 19:06:12 +00:00
package httputils
import (
"fmt"
"net/http"
"sort"
"strings"
"github.com/gin-gonic/gin"
)
func HeadersToSortedString(headers http.Header) string {
var output string
keys := make([]string, 0, len(headers))
for k := range headers {
keys = append(keys, k)
}
sort.Strings(keys)
for _, k := range keys {
if len(headers[k]) > 1 {
for _, h := range headers[k] {
output += k + ": " + h + "\n"
}
} else {
output += k + ": " + headers[k][0] + "\n"
}
}
return output
}
func GetLogFormatter(param gin.LogFormatterParams) string {
return fmt.Sprintf("%s - [%s] \"%s %s %s\" %d %d %d %s \"%s\" \"%s\" \"%s\"\n",
param.ClientIP,
param.TimeStamp.Format("02/Nov/2006:15:04:05 -0700"),
param.Method,
param.Path,
param.Request.Proto,
param.StatusCode,
param.BodySize,
param.Latency.Nanoseconds(),
normalizeLog(param.Request.Referer()),
normalizeLog(param.Request.UserAgent()),
normalizeLog(param.Request.Header["X-Forwarded-For"]),
normalizeLog(param.ErrorMessage),
)
}
func normalizeLog(log interface{}) interface{} {
switch v := log.(type) {
case string:
if v == "" {
return "-"
}
case []string:
if len(v) == 0 {
return "-"
} else {
return strings.Join(v, ", ")
}
}
return log
}