'commit'

QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java

@@ -324,7 +324,13 @@ public class HuiYaController extends Controller {
         var imgUrlList = source_img_url.split(",");
         var md5List = images_md5.split(",");
         for (int i = 0; i < imgUrlList.length; i++) {
-            ym.wxWriteImageMd5(md5List[i], imgUrlList[i]);
+            boolean success = ym.wxWriteImageMd5(md5List[i], imgUrlList[i]);
+            if (!success) {
+                res.set("success", false);
+                res.set("message", "发现img_url重复！无法执行！");
+                renderJson(res);
+                return;
+            }
         }
         int task_id = ym.wxAddTask(user_id, model_id, source_img_url, prompt_id);
         //计算当前任务的可能完成时间

QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Model/HuiYaModel.java

@@ -542,13 +542,19 @@ public class HuiYaModel {
      * @param image_md5
      * @param image_url
      */
-    public void wxWriteImageMd5(String image_md5, String image_url) {
+    public boolean wxWriteImageMd5(String image_md5, String image_url) {
-        String sql = "delete from t_hy_image_md5 where image_md5=?";
+        try {
-        Db.update(sql, image_md5);
+            String sql = "delete from t_hy_image_md5 where image_md5=?";
-        Record record = new Record();
+            Db.update(sql, image_md5);
-        record.set("image_md5", image_md5);
+            Record record = new Record();
-        record.set("image_url", image_url);
+            record.set("image_md5", image_md5);
-        Db.save("t_hy_image_md5", "image_md5", record);
+            record.set("image_url", image_url);
+            Db.save("t_hy_image_md5", "image_md5", record);
+            return true;
+        } catch (Exception err) {
+            System.out.println(err.toString());
+            return false;
+        }
     }
 
     /**