|
|
@@ -2,6 +2,8 @@ package com.iformall.utils; |
|
|
|
|
|
|
|
import com.alibaba.fastjson.JSON; |
|
|
|
import com.alibaba.fastjson.JSONObject; |
|
|
|
import com.iformall.common.ErrorCode; |
|
|
|
import com.iformall.sm.AiBaiduCheckResult; |
|
|
|
import okhttp3.*; |
|
|
|
import org.springframework.web.multipart.MultipartFile; |
|
|
|
|
|
|
@@ -29,7 +31,7 @@ public class BaiduImageCheckUtil { |
|
|
|
|
|
|
|
public static final OkHttpClient HTTP_CLIENT = new OkHttpClient().newBuilder().build(); |
|
|
|
|
|
|
|
public static String photoCheck(MultipartFile file) { |
|
|
|
public static AiBaiduCheckResult photoCheck(MultipartFile file) { |
|
|
|
try { |
|
|
|
byte[] fileBytes = file.getBytes(); |
|
|
|
String imgStr = Base64Util.encode(fileBytes); |
|
|
@@ -44,21 +46,28 @@ public class BaiduImageCheckUtil { |
|
|
|
JSONObject jsonObject = JSON.parseObject(result); |
|
|
|
//1:合规,2:不合规,3:疑似,4:审核失败 |
|
|
|
Integer type = jsonObject.getInteger("conclusionType"); |
|
|
|
String msg; |
|
|
|
AiBaiduCheckResult checkResult = new AiBaiduCheckResult(); |
|
|
|
if (type == 1) { |
|
|
|
msg = "图片合规"; |
|
|
|
checkResult.setMsg("图片合规"); |
|
|
|
checkResult.setCode(200); |
|
|
|
} else if (type == 2 || type == 3) { |
|
|
|
msg = "图片不合规"; |
|
|
|
checkResult.setMsg("图片不合规"); |
|
|
|
checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); |
|
|
|
} else if (type == 4) { |
|
|
|
msg = "图片审核失败"; |
|
|
|
checkResult.setMsg("图片审核失败"); |
|
|
|
checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); |
|
|
|
} else { |
|
|
|
msg = "图片审核失败"; |
|
|
|
checkResult.setMsg("图片审核失败"); |
|
|
|
checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); |
|
|
|
} |
|
|
|
return msg; |
|
|
|
return checkResult; |
|
|
|
} catch (Exception e) { |
|
|
|
e.printStackTrace(); |
|
|
|
AiBaiduCheckResult checkResult = new AiBaiduCheckResult(); |
|
|
|
checkResult.setMsg("图片审核失败"); |
|
|
|
checkResult.setCode(ErrorCode.SYS_SERVER_ERROR.getCode()); |
|
|
|
return checkResult; |
|
|
|
} |
|
|
|
return "图片审核接口请求失败"; |
|
|
|
} |
|
|
|
|
|
|
|
/** |
|
|
|