diff --git a/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java b/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java index e7b5373d..3daf49ae 100644 --- a/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java +++ b/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); //计算当前任务的可能完成时间 diff --git a/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Model/HuiYaModel.java b/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Model/HuiYaModel.java index 06335cd1..e46fa3ba 100644 --- a/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Model/HuiYaModel.java +++ b/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) { - String sql = "delete from t_hy_image_md5 where image_md5=?"; - Db.update(sql, image_md5); - Record record = new Record(); - record.set("image_md5", image_md5); - record.set("image_url", image_url); - Db.save("t_hy_image_md5", "image_md5", record); + public boolean wxWriteImageMd5(String image_md5, String image_url) { + try { + String sql = "delete from t_hy_image_md5 where image_md5=?"; + Db.update(sql, image_md5); + Record record = new Record(); + record.set("image_md5", image_md5); + 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; + } } /**