xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-09-17 16:56:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

opt: add lock for mj task callback

Browse Source
This commit is contained in:
RockYang 2023-09-08 18:12:18 +08:00
parent 18c033d57f
commit ebd3ef842f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/handler/chat_handler.go
View File

@ -93,7 +93,7 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
Id: chatModel.Id, Id: chatModel.Id,
Value: chatModel.Value, Value: chatModel.Value,
Platform: types.Platform(chatModel.Platform)} 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 var chatRole model.ChatRole
res = h.db.First(&chatRole, roleId) res = h.db.First(&chatRole, roleId)
if res.Error != nil || !chatRole.Enable { if res.Error != nil || !chatRole.Enable {

2
api/handler/mj_handler.go
View File

@ -85,7 +85,7 @@ func (h *MidJourneyHandler) Notify(c *gin.Context) {
logger.Debugf("收到 MidJourney 回调请求：%+v", data) logger.Debugf("收到 MidJourney 回调请求：%+v", data)
h.lock.Lock() h.lock.Lock()
defer h.lock.Unlock() defer h.lock.Unlock()
// the job is saved // the job is saved
var job model.MidJourneyJob var job model.MidJourneyJob
res := h.db.Where("message_id = ?", data.MessageId).First(&job) res := h.db.Where("message_id = ?", data.MessageId).First(&job)