You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
184 lines
7.5 KiB
184 lines
7.5 KiB
package golog
|
|
|
|
import (
|
|
"context"
|
|
"github.com/dtapps/go-library"
|
|
"github.com/dtapps/go-library/utils/dorm"
|
|
"github.com/dtapps/go-library/utils/goip"
|
|
"github.com/dtapps/go-library/utils/gorequest"
|
|
"github.com/dtapps/go-library/utils/gostring"
|
|
"github.com/dtapps/go-library/utils/gotime"
|
|
"github.com/dtapps/go-library/utils/gotrace_id"
|
|
"github.com/dtapps/go-library/utils/gourl"
|
|
"gopkg.in/natefinch/lumberjack.v2"
|
|
"log/slog"
|
|
"os"
|
|
"runtime"
|
|
)
|
|
|
|
type ApiSLogFun func() *ApiSLog
|
|
|
|
type ApiSLogConfig struct {
|
|
LogPath string // 日志文件路径
|
|
LogName string // 日志文件名
|
|
MaxSize int // 单位为MB,默认为512MB
|
|
MaxBackups int // 保留旧文件的最大个数
|
|
MaxAge int // 文件最多保存多少天 0=不删除
|
|
LocalTime bool // 采用本地时间
|
|
Compress bool // 是否压缩日志
|
|
ShowLine bool // 显示代码行
|
|
LogSaveFile bool // 是否保存到文件
|
|
}
|
|
|
|
type ApiSLog struct {
|
|
config *ApiSLogConfig
|
|
jsonHandler *slog.JSONHandler
|
|
logger *slog.Logger
|
|
systemConfig struct {
|
|
systemHostname string // 主机名
|
|
systemOs string // 系统类型
|
|
systemKernel string // 系统内核
|
|
systemInsideIp string // 内网ip
|
|
systemOutsideIp string // 外网ip
|
|
goVersion string // go版本
|
|
sdkVersion string // sdk版本
|
|
}
|
|
}
|
|
|
|
func NewApiSlog(ctx context.Context, config *ApiSLogConfig) *ApiSLog {
|
|
|
|
sl := &ApiSLog{config: config}
|
|
|
|
opts := slog.HandlerOptions{
|
|
AddSource: sl.config.ShowLine,
|
|
Level: slog.LevelDebug,
|
|
ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
|
|
if a.Key == slog.TimeKey {
|
|
a.Value = slog.StringValue(a.Value.Time().Format(gotime.DateTimeFormat))
|
|
//return slog.Attr{}
|
|
}
|
|
return a
|
|
},
|
|
}
|
|
|
|
// 是否保存到文件
|
|
if sl.config.LogSaveFile {
|
|
lumberjackLogger := lumberjack.Logger{
|
|
Filename: sl.config.LogPath + sl.config.LogName, // ⽇志⽂件路径
|
|
MaxSize: sl.config.MaxSize, // 单位为MB,默认为512MB
|
|
MaxAge: sl.config.MaxAge, // 文件最多保存多少天
|
|
MaxBackups: sl.config.MaxBackups, // 保留旧文件的最大个数
|
|
LocalTime: sl.config.LocalTime, // 采用本地时间
|
|
Compress: sl.config.Compress, // 是否压缩日志
|
|
}
|
|
sl.jsonHandler = slog.NewJSONHandler(&lumberjackLogger, &opts)
|
|
sl.logger = slog.New(sl.jsonHandler)
|
|
} else {
|
|
sl.jsonHandler = slog.NewJSONHandler(os.Stdout, &opts)
|
|
sl.logger = slog.New(sl.jsonHandler)
|
|
}
|
|
|
|
sl.setConfig(ctx)
|
|
|
|
return sl
|
|
}
|
|
|
|
// Middleware 中间件
|
|
func (sl *ApiSLog) Middleware(ctx context.Context, request gorequest.Response) {
|
|
jsonHandler := sl.jsonHandler.WithAttrs([]slog.Attr{
|
|
slog.String("trace_id", gotrace_id.GetTraceIdContext(ctx)),
|
|
slog.String("request_time", gotime.SetCurrent(request.RequestTime).Format()),
|
|
slog.String("request_uri", request.RequestUri),
|
|
slog.String("request_url", gourl.UriParse(request.RequestUri).Url),
|
|
slog.String("request_api", gourl.UriParse(request.RequestUri).Path),
|
|
slog.String("request_method", request.RequestMethod),
|
|
slog.String("request_params", gostring.ToString(request.RequestParams)),
|
|
slog.String("request_header", gostring.ToString(request.RequestHeader)),
|
|
slog.String("request_ip", sl.systemConfig.systemOutsideIp),
|
|
slog.String("response_header", gostring.ToString(request.ResponseHeader)),
|
|
slog.Int("response_status_code", request.ResponseStatusCode),
|
|
slog.String("response_body", gostring.ToString(dorm.JsonDecodeNoError(request.ResponseBody))),
|
|
slog.Int64("response_content_length", request.ResponseContentLength),
|
|
slog.String("response_time", gotime.SetCurrent(request.ResponseTime).Format()),
|
|
slog.String("system_host_name", sl.systemConfig.systemHostname),
|
|
slog.String("system_inside_ip", sl.systemConfig.systemInsideIp),
|
|
slog.String("system_os", sl.systemConfig.systemOs),
|
|
slog.String("system_arch", sl.systemConfig.systemKernel),
|
|
slog.String("go_version", sl.systemConfig.goVersion),
|
|
slog.String("sdk_version", sl.systemConfig.sdkVersion),
|
|
})
|
|
logger := slog.New(jsonHandler)
|
|
logger.Info("Middleware")
|
|
}
|
|
|
|
// MiddlewareXml 中间件
|
|
func (sl *ApiSLog) MiddlewareXml(ctx context.Context, request gorequest.Response) {
|
|
jsonHandler := sl.jsonHandler.WithAttrs([]slog.Attr{
|
|
slog.String("trace_id", gotrace_id.GetTraceIdContext(ctx)),
|
|
slog.String("request_time", gotime.SetCurrent(request.RequestTime).Format()),
|
|
slog.String("request_uri", request.RequestUri),
|
|
slog.String("request_url", gourl.UriParse(request.RequestUri).Url),
|
|
slog.String("request_api", gourl.UriParse(request.RequestUri).Path),
|
|
slog.String("request_method", request.RequestMethod),
|
|
slog.String("request_params", gostring.ToString(request.RequestParams)),
|
|
slog.String("request_header", gostring.ToString(request.RequestHeader)),
|
|
slog.String("request_ip", sl.systemConfig.systemOutsideIp),
|
|
slog.String("response_header", gostring.ToString(request.ResponseHeader)),
|
|
slog.Int("response_status_code", request.ResponseStatusCode),
|
|
slog.String("response_body", gostring.ToString(dorm.XmlDecodeNoError(request.ResponseBody))),
|
|
slog.Int64("response_content_length", request.ResponseContentLength),
|
|
slog.String("response_time", gotime.SetCurrent(request.ResponseTime).Format()),
|
|
slog.String("system_host_name", sl.systemConfig.systemHostname),
|
|
slog.String("system_inside_ip", sl.systemConfig.systemInsideIp),
|
|
slog.String("system_os", sl.systemConfig.systemOs),
|
|
slog.String("system_arch", sl.systemConfig.systemKernel),
|
|
slog.String("go_version", sl.systemConfig.goVersion),
|
|
slog.String("sdk_version", sl.systemConfig.sdkVersion),
|
|
})
|
|
logger := slog.New(jsonHandler)
|
|
logger.Info("MiddlewareXml")
|
|
}
|
|
|
|
// MiddlewareCustom 中间件
|
|
func (sl *ApiSLog) MiddlewareCustom(ctx context.Context, api string, request gorequest.Response) {
|
|
jsonHandler := sl.jsonHandler.WithAttrs([]slog.Attr{
|
|
slog.String("trace_id", gotrace_id.GetTraceIdContext(ctx)),
|
|
slog.String("request_time", gotime.SetCurrent(request.RequestTime).Format()),
|
|
slog.String("request_uri", request.RequestUri),
|
|
slog.String("request_url", gourl.UriParse(request.RequestUri).Url),
|
|
slog.String("request_api", api),
|
|
slog.String("request_method", request.RequestMethod),
|
|
slog.String("request_params", gostring.ToString(request.RequestParams)),
|
|
slog.String("request_header", gostring.ToString(request.RequestHeader)),
|
|
slog.String("request_ip", sl.systemConfig.systemOutsideIp),
|
|
slog.String("response_header", gostring.ToString(request.ResponseHeader)),
|
|
slog.Int("response_status_code", request.ResponseStatusCode),
|
|
slog.String("response_body", gostring.ToString(dorm.JsonDecodeNoError(request.ResponseBody))),
|
|
slog.Int64("response_content_length", request.ResponseContentLength),
|
|
slog.String("response_time", gotime.SetCurrent(request.ResponseTime).Format()),
|
|
slog.String("system_host_name", sl.systemConfig.systemHostname),
|
|
slog.String("system_inside_ip", sl.systemConfig.systemInsideIp),
|
|
slog.String("system_os", sl.systemConfig.systemOs),
|
|
slog.String("system_arch", sl.systemConfig.systemKernel),
|
|
slog.String("go_version", sl.systemConfig.goVersion),
|
|
slog.String("sdk_version", sl.systemConfig.sdkVersion),
|
|
})
|
|
logger := slog.New(jsonHandler)
|
|
logger.Info("MiddlewareCustom")
|
|
}
|
|
|
|
func (sl *ApiSLog) setConfig(ctx context.Context) {
|
|
|
|
info := getSystem()
|
|
|
|
sl.systemConfig.systemHostname = info.SystemHostname
|
|
sl.systemConfig.systemOs = info.SystemOs
|
|
sl.systemConfig.systemKernel = info.SystemKernel
|
|
|
|
sl.systemConfig.systemInsideIp = goip.GetInsideIp(ctx)
|
|
|
|
sl.systemConfig.sdkVersion = go_library.Version()
|
|
sl.systemConfig.goVersion = runtime.Version()
|
|
|
|
}
|