feat: blend and swap face function for midjourney-plus is ready

api/service/mj/plus/client.go

@@ -98,7 +98,7 @@ func (c *Client) Blend(task types.MjTask) (ImageRes, error) {
 		BotType:     "MID_JOURNEY",
 		Dimensions:  "SQUARE",
 		NotifyHook:  c.Config.NotifyURL,
-		Base64Array: make([]string, 1),
+		Base64Array: make([]string, 0),
 	}
 	// 生成图片 Base64 编码
 	if len(task.ImgArr) > 0 {
@@ -107,7 +107,7 @@ func (c *Client) Blend(task types.MjTask) (ImageRes, error) {
 			if err != nil {
 				logger.Error("error with download image: ", err)
 			} else {
-				body.Base64Array[0] = "data:image/png;base64," + base64.StdEncoding.EncodeToString(imageData)
+				body.Base64Array = append(body.Base64Array, "data:image/png;base64,"+base64.StdEncoding.EncodeToString(imageData))
 			}
 		}
 	}

api/service/mj/pool.go

@@ -167,7 +167,7 @@ func (p *ServicePool) HasAvailableService() bool {
 }
 
 func (p *ServicePool) Notify(data plus.CBReq) error {
-	logger.Infof("收到任务回调：%+v", data)
+	logger.Debugf("收到任务回调：%+v", data)
 	var job model.MidJourneyJob
 	res := p.db.Where("task_id = ?", data.Id).First(&job)
 	if res.Error != nil {
@@ -190,7 +190,7 @@ func (p *ServicePool) SyncTaskProgress() {
 	go func() {
 		var items []model.MidJourneyJob
 		for {
-			res := p.db.Where("progress < ?", 100).Find(&items)
+			res := p.db.Where("progress >= ? AND progress < ?", 0, 100).Find(&items)
 			if res.Error != nil {
 				continue
 			}
@@ -215,6 +215,11 @@ func (p *ServicePool) SyncTaskProgress() {
 					if err != nil {
 						continue
 					}
+					// 任务失败了
+					if task.FailReason != "" {
+						p.db.Model(&model.MidJourneyJob{Id: v.Id}).UpdateColumn("progress", -1)
+						continue
+					}
 					if len(task.Buttons) > 0 {
 						v.Hash = getImageHash(task.Buttons[0].CustomId)
 					}