opt: 缩略图生成算法

api/core/app_server.go

@@ -314,10 +314,16 @@ func staticResourceMiddleware() gin.HandlerFunc {
 				return
 			}
 
-			// 生成缩略图
-			resizedImg := resize.Thumbnail(uint(with), uint(height), img, resize.Lanczos3)
+			var newImg image.Image
+			if height == 0 || with == 0 {
+				// 固定宽度，高度自适应
+				newImg = resize.Resize(uint(with), uint(height), img, resize.Lanczos3)
+			} else {
+				// 生成缩略图
+				newImg = resize.Thumbnail(uint(with), uint(height), img, resize.Lanczos3)
+			}
 			var buffer bytes.Buffer
-			err = jpeg.Encode(&buffer, resizedImg, &jpeg.Options{Quality: quality})
+			err = jpeg.Encode(&buffer, newImg, &jpeg.Options{Quality: quality})
 			if err != nil {
 				log.Fatal(err)
 			}

api/handler/mj_handler.go

@@ -120,7 +120,7 @@ func (h *MidJourneyHandler) Image(c *gin.Context) {
 		prompt += " --style raw"
 	}
 	if data.Model != "" && !strings.Contains(prompt, "--v") && !strings.Contains(prompt, "--niji") {
-		prompt += data.Model
+		prompt += fmt.Sprintf(" %s", data.Model)
 	}
 
 	idValue, _ := c.Get(types.LoginUserID)
@@ -348,8 +348,8 @@ func (h *MidJourneyHandler) JobList(c *gin.Context) {
 			continue
 		}
 		if item.Progress < 100 {
-			// 30 分钟还没完成的任务直接删除
-			if time.Now().Sub(item.CreatedAt) > time.Minute*30 {
+			// 10 分钟还没完成的任务直接删除
+			if time.Now().Sub(item.CreatedAt) > time.Minute*10 {
 				h.db.Delete(&item)
 				continue
 			}

api/service/mj/service.go

@@ -72,6 +72,8 @@ func (s *Service) Run() {
 		}
 		if err != nil {
 			logger.Error("绘画任务执行失败：", err)
+			// 删除任务
+			s.db.Delete(&model.MidJourneyJob{Id: uint(task.Id)})
 			// 推送任务到前端
 			client := s.Clients.Get(task.SessionId)
 			if client != nil {