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
|
|
}
|