Преглед на файлове

fix 任务结束,但未出图

zoujiajian преди 1 година
родител
ревизия
f3d22ecf6b

+ 9 - 0
netflix-service/src/main/java/com/cyksj/server/listener/ConsoleStreamListener.java

@@ -59,6 +59,15 @@ public class ConsoleStreamListener extends AbstractStreamListener {
 				miniPopularizeCommonService.recordChatEndOfAnswer(conversationId, userId, "图片生成异常", MiniPopularizeUserConversationChat.Role.assistant);
 			}
 		}
+		//任务已完成 但图未出
+		if (content.contains("[DONE]")) {
+			Boolean flag = miniPopularizeCommonService.checkImageIsCompleted(imgRecordId);
+			if (!flag) {
+				//将用户使用次数返回
+				miniPopularizeCommonService.recoverUserLimit(userId, recoverNum);
+				miniPopularizeCommonService.saveImgRecord(setImagRecord(imgRecordId, img, MiniPopularizeUserConversationChat.Status.fail, "任务结束,未出图"));
+			}
+		}
 //		SendMessageUtil.sendMessage(userId, message);
 	}
 

+ 2 - 0
netflix-service/src/main/java/com/cyksj/service/mini/MiniPopularizeCommonService.java

@@ -17,4 +17,6 @@ public interface MiniPopularizeCommonService {
 	void recordChatEndOfAnswer(String conversationId, Long userId, String content, MiniPopularizeUserConversationChat.Role role);
 
 	void incrGptTimes(long userId, Integer oriNum, Integer addNum);
+
+	Boolean checkImageIsCompleted(Long imgRecordId);
 }

+ 9 - 0
netflix-service/src/main/java/com/cyksj/service/mini/impl/MiniPopularizeCommonServiceImpl.java

@@ -75,4 +75,13 @@ public class MiniPopularizeCommonServiceImpl implements MiniPopularizeCommonServ
 			i = miniPopularizeUserMapper.incrNumSync(userId, oriNum, addNum);
 		}
 	}
+
+	@Override
+	public Boolean checkImageIsCompleted(Long imgRecordId) {
+		MiniPopularizeUserConversationChat dbChat = conversationChatMapper.selectById(imgRecordId);
+		if (dbChat == null || (dbChat.getType() == MiniPopularizeUserConversationChat.Type.image && StrUtil.isEmpty(dbChat.getImg()))) {
+			return false;
+		}
+		return true;
+	}
 }