xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-17 16:06:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: print user id when error

Browse Source
This commit is contained in:
CalciumIon 2024-07-27 17:47:30 +08:00
parent d4a5df7373
commit ab1d61d910
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
common/utils.go
View File

@ -179,8 +179,8 @@ func Max(a int, b int) int {
} }
} }
func MessageWithRequestId(message string, id string) string { func MessageWithRequestId(message string, id string, userId int) string {
return fmt.Sprintf("%s (request id: %s)", message, id) return fmt.Sprintf("%s (request id: %s, user id: %d)", message, id, userId)
} }
func RandomSleep() { func RandomSleep() {

3
controller/relay.go
View File

@ -82,7 +82,8 @@ func Relay(c *gin.Context) {
if openaiErr.StatusCode == http.StatusTooManyRequests { if openaiErr.StatusCode == http.StatusTooManyRequests {
openaiErr.Error.Message = "当前分组上游负载已饱和，请稍后再试" openaiErr.Error.Message = "当前分组上游负载已饱和，请稍后再试"
} }
openaiErr.Error.Message = common.MessageWithRequestId(openaiErr.Error.Message, requestId) userId := c.GetInt("id")
openaiErr.Error.Message = common.MessageWithRequestId(openaiErr.Error.Message, requestId, userId)
c.JSON(openaiErr.StatusCode, gin.H{ c.JSON(openaiErr.StatusCode, gin.H{
"error": openaiErr.Error, "error": openaiErr.Error,
}) })

3
middleware/utils.go
View File

@ -6,9 +6,10 @@ import (
) )
func abortWithOpenAiMessage(c *gin.Context, statusCode int, message string) { func abortWithOpenAiMessage(c *gin.Context, statusCode int, message string) {
userId := c.GetInt("id")
c.JSON(statusCode, gin.H{ c.JSON(statusCode, gin.H{
"error": gin.H{ "error": gin.H{
"message": common.MessageWithRequestId(message, c.GetString(common.RequestIdKey)), "message": common.MessageWithRequestId(message, c.GetString(common.RequestIdKey), userId),
"type": "new_api_error", "type": "new_api_error",
}, },
}) })