xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-08 10:13:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

opt: add default extension for mj image

Browse Source
This commit is contained in:
RockYang
2024-01-30 21:46:17 +08:00
parent 8f4d94f046
commit 56a91db7e7
7 changed files with 38 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
api/service/mj/plus/service.go
View File

@@ -180,7 +180,6 @@ func (s *Service) Notify(job model.MidJourneyJob) error {
job.OrgURL = task.ImageUrl
}
}
job.UseProxy = true
job.MessageId = task.Id
tx := s.db.Updates(&job)
if tx.Error != nil {

12
api/service/mj/pool.go
View File

@@ -124,14 +124,12 @@ func (p *ServicePool) DownloadImages() {
logger.Infof("try to download image: %s", v.OrgURL)
var imgURL string
var err error
if v.UseProxy {
if servicePlus := p.getServicePlus(v.ChannelId); servicePlus != nil {
task, _ := servicePlus.Client.QueryTask(v.TaskId)
if len(task.Buttons) > 0 {
v.Hash = plus.GetImageHash(task.Buttons[0].CustomId)
}
imgURL, err = p.uploaderManager.GetUploadHandler().PutImg(v.OrgURL, false)
if servicePlus := p.getServicePlus(v.ChannelId); servicePlus != nil {
task, _ := servicePlus.Client.QueryTask(v.TaskId)
if len(task.Buttons) > 0 {
v.Hash = plus.GetImageHash(task.Buttons[0].CustomId)
}
imgURL, err = p.uploaderManager.GetUploadHandler().PutImg(v.OrgURL, false)
} else {
imgURL, err = p.uploaderManager.GetUploadHandler().PutImg(v.OrgURL, true)
}