mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-20 02:06:38 +08:00
opt: add lock for mj task callback
This commit is contained in:
parent
0a9b325360
commit
153b2bfa53
@ -13,6 +13,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -41,6 +42,7 @@ type MidJourneyHandler struct {
|
|||||||
db *gorm.DB
|
db *gorm.DB
|
||||||
mjFunc function.FuncMidJourney
|
mjFunc function.FuncMidJourney
|
||||||
uploaderManager *oss.UploaderManager
|
uploaderManager *oss.UploaderManager
|
||||||
|
lock sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewMidJourneyHandler(
|
func NewMidJourneyHandler(
|
||||||
@ -53,6 +55,7 @@ func NewMidJourneyHandler(
|
|||||||
leveldb: leveldb,
|
leveldb: leveldb,
|
||||||
db: db,
|
db: db,
|
||||||
uploaderManager: manager,
|
uploaderManager: manager,
|
||||||
|
lock: sync.Mutex{},
|
||||||
mjFunc: functions[types.FuncMidJourney].(function.FuncMidJourney)}
|
mjFunc: functions[types.FuncMidJourney].(function.FuncMidJourney)}
|
||||||
h.App = app
|
h.App = app
|
||||||
return &h
|
return &h
|
||||||
@ -80,7 +83,9 @@ func (h *MidJourneyHandler) Notify(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
logger.Debugf("收到 MidJourney 回调请求:%+v", data)
|
logger.Debugf("收到 MidJourney 回调请求:%+v", data)
|
||||||
|
h.lock.Lock()
|
||||||
|
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)
|
||||||
|
Loading…
Reference in New Issue
Block a user