97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
/*
|
||
* Copyright 2025 coze-dev Authors
|
||
*
|
||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
|
||
package middleware
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net/http"
|
||
"path/filepath"
|
||
"strings"
|
||
"time"
|
||
"unsafe"
|
||
|
||
"github.com/cloudwego/hertz/pkg/app"
|
||
"github.com/google/uuid"
|
||
|
||
"github.com/coze-dev/coze-studio/backend/pkg/i18n"
|
||
"github.com/coze-dev/coze-studio/backend/pkg/logs"
|
||
)
|
||
|
||
func AccessLogMW() app.HandlerFunc {
|
||
return func(c context.Context, ctx *app.RequestContext) {
|
||
start := time.Now()
|
||
ctx.Next(c)
|
||
|
||
status := ctx.Response.StatusCode()
|
||
path := bytesToString(ctx.Request.URI().PathOriginal())
|
||
latency := time.Since(start)
|
||
method := bytesToString(ctx.Request.Header.Method())
|
||
clientIP := ctx.ClientIP()
|
||
|
||
handlerPkgPath := strings.Split(ctx.HandlerName(), "/")
|
||
handleName := ""
|
||
if len(handlerPkgPath) > 0 {
|
||
handleName = handlerPkgPath[len(handlerPkgPath)-1]
|
||
}
|
||
|
||
requestType := ctx.GetInt32(RequestAuthTypeStr)
|
||
baseLog := fmt.Sprintf("| %s | %s | %d | %v | %s | %s | %v | %s | %d ｜ %s",
|
||
string(ctx.GetRequest().Scheme()), ctx.Host(), status,
|
||
latency, clientIP, method, path, handleName, requestType, i18n.GetLocale(c))
|
||
|
||
switch {
|
||
case status >= http.StatusInternalServerError:
|
||
logs.CtxErrorf(c, "%s", baseLog)
|
||
case status >= http.StatusBadRequest:
|
||
logs.CtxWarnf(c, "%s", baseLog)
|
||
default:
|
||
urlQuery := ctx.Request.URI().QueryString()
|
||
reqBody := bytesToString(ctx.Request.Body())
|
||
respBody := bytesToString(ctx.Response.Body())
|
||
maxPrintLen := 3 * 1024
|
||
if len(respBody) > maxPrintLen {
|
||
respBody = respBody[:maxPrintLen]
|
||
}
|
||
if len(reqBody) > maxPrintLen {
|
||
reqBody = reqBody[:maxPrintLen]
|
||
}
|
||
|
||
requestAuthType := ctx.GetInt32(RequestAuthTypeStr)
|
||
if requestAuthType != int32(RequestAuthTypeStaticFile) && filepath.Ext(path) == "" {
|
||
logs.CtxInfof(c, "%s ", baseLog)
|
||
logs.CtxDebugf(c, "query : %s \nreq : %s \nresp: %s",
|
||
urlQuery, reqBody, respBody)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
func SetLogIDMW() app.HandlerFunc {
|
||
return func(ctx context.Context, c *app.RequestContext) {
|
||
logID := uuid.New().String()
|
||
ctx = context.WithValue(ctx, "log-id", logID)
|
||
|
||
c.Header("X-Log-ID", logID)
|
||
c.Next(ctx)
|
||
}
|
||
}
|
||
|
||
func bytesToString(b []byte) string {
|
||
return *(*string)(unsafe.Pointer(&b)) // nolint
|
||
}
|