支持设置日志扩展字段

client.go

@@ -38,6 +38,7 @@ type OwnClient struct {
 	instanceTable     map[string]*redisClient.Client
 	whiteCommandTable map[string]bool
 	logger            *zap.Logger
+	extraLogFieldList []string
 }
 
 func (o *OwnClient) isAllowCommand(command string) bool {
@@ -61,6 +62,9 @@ func (o *OwnClient) isAllowCommand(command string) bool {
 //
 // Date : 11:05 2024/6/19
 func (o *OwnClient) Exec(ctx context.Context, instanceFlag string, command string, args ...any) *define.RedisResult {
+	if nil == ctx {
+		ctx = context.Background()
+	}
 	var (
 		instance *redisClient.Client
 	)
@@ -90,8 +94,7 @@ func (o *OwnClient) Exec(ctx context.Context, instanceFlag string, command strin
 			// 未注入日志实例
 			return
 		}
-		o.logger.Info(
-			"Redis命令执行记录",
+		logDataList := []zap.Field{
 			zap.Int64("start_time", res.StartTime),
 			zap.Int64("finish_time", res.FinishTime),
 			zap.Int64("used_time", res.UsedTime),
@@ -99,6 +102,13 @@ func (o *OwnClient) Exec(ctx context.Context, instanceFlag string, command strin
 			zap.String("arg_list", strings.Join(res.ArgList, " ")),
 			zap.String("execute_result", res.Result),
 			zap.Error(res.Err),
+		}
+		for _, item := range o.extraLogFieldList {
+			logDataList = append(logDataList, zap.Any(item, ctx.Value(item)))
+		}
+		o.logger.Info(
+			"Redis命令执行记录",
+			logDataList...,
 		)
 	}()
 	if instance, res.Err = o.GetRealClientWithError(instanceFlag); nil != res.Err {
@@ -189,6 +199,7 @@ func (o *OwnClient) RemoveClient(instanceFlag string) {
 	delete(o.instanceTable, instanceFlag)
 }
 
-func (o *OwnClient) SetLogger(loggerInstance *zap.Logger) {
+func (o *OwnClient) SetLogger(loggerInstance *zap.Logger, extraLogFieldList []string) {
 	o.logger = loggerInstance
+	o.extraLogFieldList = extraLogFieldList
 }