From d841342d20a75585bb0e780ca777e0ea5083fb3c Mon Sep 17 00:00:00 2001 From: lrh <1585126058@qq.com> Date: Wed, 5 Jul 2023 10:14:55 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=A7=86=E9=A2=91=E5=88=9B?= =?UTF-8?q?=E5=BB=BA=E9=80=BB=E8=BE=91=E3=80=81=E6=96=B0=E5=A2=9E=E5=9B=BE?= =?UTF-8?q?=E7=89=87=E6=A3=80=E6=9F=A5=E6=8E=A5=E5=8F=A3=E3=80=81=E6=96=B0?= =?UTF-8?q?=E5=A2=9E=E7=99=BE=E5=BA=A6API=E6=8E=A5=E5=8F=A3=E3=80=81?= =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=B8=8A=E4=BC=A0=E9=98=BF=E9=87=8C=E4=BA=91?= =?UTF-8?q?=E6=A3=80=E6=9F=A5=E3=80=81=E6=96=B0=E5=A2=9E=E5=9B=9E=E8=B0=83?= =?UTF-8?q?=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../controller/PersonPhotoController.java | 3 +-- .../iformall/utils/BaiduImageCheckUtil.java | 22 ++++++------------- 2 files changed, 8 insertions(+), 17 deletions(-) diff --git a/suimangCApi/src/main/java/com/iformall/controller/PersonPhotoController.java b/suimangCApi/src/main/java/com/iformall/controller/PersonPhotoController.java index 6526fc5..3ed4fe7 100644 --- a/suimangCApi/src/main/java/com/iformall/controller/PersonPhotoController.java +++ b/suimangCApi/src/main/java/com/iformall/controller/PersonPhotoController.java @@ -109,8 +109,7 @@ public class PersonPhotoController extends BaseController { if (size > length) { return new ResultData(ErrorCode.PICTURE_FOUR_SIZE_EXCEED); } - AiBaiduCheckResult checkResult = BaiduImageCheckUtil.photoCheck(file); - return new ResultData(checkResult); + return BaiduImageCheckUtil.photoCheck(file); } diff --git a/suimangCApi/src/main/java/com/iformall/utils/BaiduImageCheckUtil.java b/suimangCApi/src/main/java/com/iformall/utils/BaiduImageCheckUtil.java index efe41d9..1e167f0 100644 --- a/suimangCApi/src/main/java/com/iformall/utils/BaiduImageCheckUtil.java +++ b/suimangCApi/src/main/java/com/iformall/utils/BaiduImageCheckUtil.java @@ -3,6 +3,7 @@ package com.iformall.utils; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; import com.iformall.common.ErrorCode; +import com.iformall.common.ResultData; import com.iformall.sm.AiBaiduCheckResult; import okhttp3.*; import org.springframework.web.multipart.MultipartFile; @@ -31,7 +32,7 @@ public class BaiduImageCheckUtil { public static final OkHttpClient HTTP_CLIENT = new OkHttpClient().newBuilder().build(); - public static AiBaiduCheckResult photoCheck(MultipartFile file) { + public static ResultData photoCheck(MultipartFile file) { try { byte[] fileBytes = file.getBytes(); String imgStr = Base64Util.encode(fileBytes); @@ -46,27 +47,18 @@ public class BaiduImageCheckUtil { JSONObject jsonObject = JSON.parseObject(result); //1:合规,2:不合规,3:疑似,4:审核失败 Integer type = jsonObject.getInteger("conclusionType"); - AiBaiduCheckResult checkResult = new AiBaiduCheckResult(); if (type == 1) { - checkResult.setMsg("图片合规"); - checkResult.setCode(200); + return new ResultData(); } else if (type == 2 || type == 3) { - checkResult.setMsg("图片不合规"); - checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); + return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"图片不合规"); } else if (type == 4) { - checkResult.setMsg("图片审核失败"); - checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); + return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"图片审核失败"); } else { - checkResult.setMsg("图片审核失败"); - checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); + return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"图片审核失败"); } - return checkResult; } catch (Exception e) { e.printStackTrace(); - AiBaiduCheckResult checkResult = new AiBaiduCheckResult(); - checkResult.setMsg("图片审核失败"); - checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); - return checkResult; + return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"图片审核失败"); } }