rename ngin to rest

rest/internal/log.go

@@ -0,0 +1,83 @@
+package internal
+
+import (
+	"bytes"
+	"fmt"
+	"net/http"
+	"sync"
+
+	"zero/core/logx"
+)
+
+const LogContext = "request_logs"
+
+type LogCollector struct {
+	Messages []string
+	lock     sync.Mutex
+}
+
+func (lc *LogCollector) Append(msg string) {
+	lc.lock.Lock()
+	lc.Messages = append(lc.Messages, msg)
+	lc.lock.Unlock()
+}
+
+func (lc *LogCollector) Flush() string {
+	var buffer bytes.Buffer
+
+	start := true
+	for _, message := range lc.takeAll() {
+		if start {
+			start = false
+		} else {
+			buffer.WriteByte('\n')
+		}
+		buffer.WriteString(message)
+	}
+
+	return buffer.String()
+}
+
+func (lc *LogCollector) takeAll() []string {
+	lc.lock.Lock()
+	messages := lc.Messages
+	lc.Messages = nil
+	lc.lock.Unlock()
+
+	return messages
+}
+
+func Error(r *http.Request, v ...interface{}) {
+	logx.ErrorCaller(1, format(r, v...))
+}
+
+func Errorf(r *http.Request, format string, v ...interface{}) {
+	logx.ErrorCaller(1, formatf(r, format, v...))
+}
+
+func Info(r *http.Request, v ...interface{}) {
+	appendLog(r, format(r, v...))
+}
+
+func Infof(r *http.Request, format string, v ...interface{}) {
+	appendLog(r, formatf(r, format, v...))
+}
+
+func appendLog(r *http.Request, message string) {
+	logs := r.Context().Value(LogContext)
+	if logs != nil {
+		logs.(*LogCollector).Append(message)
+	}
+}
+
+func format(r *http.Request, v ...interface{}) string {
+	return formatWithReq(r, fmt.Sprint(v...))
+}
+
+func formatf(r *http.Request, format string, v ...interface{}) string {
+	return formatWithReq(r, fmt.Sprintf(format, v...))
+}
+
+func formatWithReq(r *http.Request, v string) string {
+	return fmt.Sprintf("(%s - %s) %s", r.RequestURI, GetRemoteAddr(r), v)
+}