From ebd3ef842f4491209ca290a99c0e91768ac53ff2 Mon Sep 17 00:00:00 2001 From: RockYang Date: Fri, 8 Sep 2023 18:12:18 +0800 Subject: [PATCH] opt: add lock for mj task callback --- api/handler/chat_handler.go | 2 +- api/handler/mj_handler.go | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/handler/chat_handler.go b/api/handler/chat_handler.go index 0fae34f4..baaa1e86 100644 --- a/api/handler/chat_handler.go +++ b/api/handler/chat_handler.go @@ -93,7 +93,7 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) { Id: chatModel.Id, Value: chatModel.Value, Platform: types.Platform(chatModel.Platform)} - logger.Infof("New websocket connected, IP: %s, Username: %s", c.Request.RemoteAddr, session.Username) + logger.Infof("New websocket connected, IP: %s, Username: %s", c.ClientIP(), session.Username) var chatRole model.ChatRole res = h.db.First(&chatRole, roleId) if res.Error != nil || !chatRole.Enable { diff --git a/api/handler/mj_handler.go b/api/handler/mj_handler.go index 82550be3..1c446952 100644 --- a/api/handler/mj_handler.go +++ b/api/handler/mj_handler.go @@ -85,7 +85,7 @@ func (h *MidJourneyHandler) Notify(c *gin.Context) { logger.Debugf("收到 MidJourney 回调请求：%+v", data) h.lock.Lock() defer h.lock.Unlock() - + // the job is saved var job model.MidJourneyJob res := h.db.Where("message_id = ?", data.MessageId).First(&job)