Jelajahi Sumber

feat: capture request_id, filter by request_id, show request_conversion

Seefs 1 bulan lalu
induk
melakukan
bb54ed91dc
1 mengubah file dengan 4 tambahan dan 2 penghapusan
  1. 4 2
      model/log.go

+ 4 - 2
model/log.go

@@ -102,6 +102,7 @@ func RecordErrorLog(c *gin.Context, userId int, channelId int, modelName string,
 	isStream bool, group string, other map[string]interface{}) {
 	logger.LogInfo(c, fmt.Sprintf("record error log: userId=%d, channelId=%d, modelName=%s, tokenName=%s, content=%s", userId, channelId, modelName, tokenName, content))
 	username := c.GetString("username")
+	requestId := c.GetString(common.RequestIdKey)
 	otherStr := common.MapToJsonStr(other)
 	// 判断是否需要记录 IP
 	needRecordIp := false
@@ -132,7 +133,7 @@ func RecordErrorLog(c *gin.Context, userId int, channelId int, modelName string,
 			}
 			return ""
 		}(),
-		RequestId: c.GetString(common.RequestIdKey),
+		RequestId: requestId,
 		Other:     otherStr,
 	}
 	err := LOG_DB.Create(log).Error
@@ -162,6 +163,7 @@ func RecordConsumeLog(c *gin.Context, userId int, params RecordConsumeLogParams)
 	}
 	logger.LogInfo(c, fmt.Sprintf("record consume log: userId=%d, params=%s", userId, common.GetJsonString(params)))
 	username := c.GetString("username")
+	requestId := c.GetString(common.RequestIdKey)
 	otherStr := common.MapToJsonStr(params.Other)
 	// 判断是否需要记录 IP
 	needRecordIp := false
@@ -192,7 +194,7 @@ func RecordConsumeLog(c *gin.Context, userId int, params RecordConsumeLogParams)
 			}
 			return ""
 		}(),
-		RequestId: c.GetString(common.RequestIdKey),
+		RequestId: requestId,
 		Other:     otherStr,
 	}
 	err := LOG_DB.Create(log).Error