winter 1 год назад
Родитель
Сommit
006907c173
5 измененных файлов: 47 добавлений и 9 удалений
  1. +13
    -2
      suimangCApi/src/main/java/com/iformall/controller/BaiduController.java
  2. +2
    -2
      suimangCApi/src/main/java/com/iformall/controller/PersonPhotoController.java
  3. +1
    -1
      suimangCApi/src/main/java/com/iformall/controller/ProductController.java
  4. +3
    -3
      suimangCApi/src/main/java/com/iformall/controller/UserMouldVideoController.java
  5. +28
    -1
      suimangCApi/src/main/java/com/iformall/utils/BaiduCheckUtil.java

+ 13
- 2
suimangCApi/src/main/java/com/iformall/controller/BaiduController.java Просмотреть файл

@@ -10,7 +10,7 @@ import com.iformall.enums.EnumMouldSendType;
import com.iformall.enums.EnumaMouldPatchStatus;
import com.iformall.service.sm.MouldPatchSignService;
import com.iformall.service.sm.PersonPhotoService;
import com.iformall.utils.BaiduImageCheckUtil;
import com.iformall.utils.BaiduCheckUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
@@ -45,7 +45,18 @@ public class BaiduController extends BaseController {
if (size > length) {
return new ResultData(ErrorCode.PICTURE_FOUR_SIZE_EXCEED);
}
return BaiduImageCheckUtil.photoCheck(file);
return BaiduCheckUtil.photoCheck(file);
}
@AuthIgnore
@ApiOperation("百度文字审核接口")
@PostMapping(value = "checkText")
@ApiImplicitParam(name = "text", value = "text", dataType = "String", paramType = "query", required = true)
public ResultData baiduCheckPhoto(@RequestBody String text) {
if (StringUtils.isBlank(text)) {
return new ResultData(ErrorCode.SYS_PARAMETER_NOT_NULL.getCode(), "内容为空");
}
return BaiduCheckUtil.textCheck(text);
}

}

+ 2
- 2
suimangCApi/src/main/java/com/iformall/controller/PersonPhotoController.java Просмотреть файл

@@ -11,7 +11,7 @@ import com.iformall.enums.EnumaMouldPatchStatus;
import com.iformall.service.sm.MouldPatchSignService;
import com.iformall.service.sm.PersonPhotoService;
import com.iformall.sm.AiBaiduCheckResult;
import com.iformall.utils.BaiduImageCheckUtil;
import com.iformall.utils.BaiduCheckUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
@@ -109,7 +109,7 @@ public class PersonPhotoController extends BaseController {
if (size > length) {
return new ResultData(ErrorCode.PICTURE_FOUR_SIZE_EXCEED);
}
return BaiduImageCheckUtil.photoCheck(file);
return BaiduCheckUtil.photoCheck(file);
}




+ 1
- 1
suimangCApi/src/main/java/com/iformall/controller/ProductController.java Просмотреть файл

@@ -14,7 +14,7 @@ import com.iformall.enums.EnumaMouldPatchStatus;
import com.iformall.service.ProductService;
import com.iformall.service.sm.MouldPatchSignService;
import com.iformall.service.sm.PersonPhotoService;
import com.iformall.utils.BaiduImageCheckUtil;
import com.iformall.utils.BaiduCheckUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;


+ 3
- 3
suimangCApi/src/main/java/com/iformall/controller/UserMouldVideoController.java Просмотреть файл

@@ -24,6 +24,7 @@ import com.iformall.service.sm.VoiceInfoService;
import com.iformall.smsdk.SmGenerateVideoDTO;
import com.iformall.smsdk.SmPreviewVideoDTO;
import com.iformall.smsdk.SmSdkUtils;
import com.iformall.utils.BaiduCheckUtil;
import com.iformall.video.VideoFactory;
import com.iformall.video.entity.VideUploadResult;
import io.swagger.annotations.Api;
@@ -228,9 +229,8 @@ public class UserMouldVideoController extends BaseController {
if(mouldVideo == null){
return new ResultData(ErrorCode.SYS_PARAMETER_NOT_NULL.getCode(),"id无效");
}
//TODO 校验
//return new ResultData(ErrorCode.SYS_PARAMETER_NOT_NULL.getCode(),"校验失败");
return new ResultData();
//校验文本
return BaiduCheckUtil.textCheck(mouldVideo.getPaperwork());
}

// @AuthIgnore


suimangCApi/src/main/java/com/iformall/utils/BaiduImageCheckUtil.java → suimangCApi/src/main/java/com/iformall/utils/BaiduCheckUtil.java Просмотреть файл

@@ -22,9 +22,10 @@ import java.util.HashMap;
*/
@Slf4j
@Component
public class BaiduImageCheckUtil {
public class BaiduCheckUtil {
// 百度图片审核接口地址
private final static String photo_check_url = "https://aip.baidubce.com/rest/2.0/solution/v1/img_censor/v2/user_defined";
private final static String text_check_url = "https://aip.baidubce.com/rest/2.0/solution/v1/text_censor/v2/user_defined";
// 获取token
private final static String auth_url = "https://aip.baidubce.com/oauth/2.0/token?";
// 百度API Key
@@ -33,6 +34,32 @@ public class BaiduImageCheckUtil {
// 百度Secret Key
private final static String secretKey = "eGmeQkP3Opzph0GB4Y2voiOkGOlwbeWd";

public static ResultData textCheck(String text) {
try {
String param = "text=" + URLEncoder.encode(text, "utf-8");;
String accessToken = getAuth();
String result = BaiDuHttpUtil.post(text_check_url, accessToken, param);
log.info("图片检测"+result);
JSONObject jsonObject = JSON.parseObject(result);
//1:合规,2:不合规,3:疑似,4:审核失败
Integer type = jsonObject.getInteger("conclusionType");
if (type == 1) {
return new ResultData();
} else if (type == 2 || type == 3) {
return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"文字内容不合规");
} else if (type == 4) {
return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"文字内容审核失败");
} else {
return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"文字内容不合规");
}
} catch (Exception e) {
e.printStackTrace();
return new ResultData(ErrorCode.SYS_SERVER_ERROR.getCode(),"文字内容审核失败");
}
}
public static ResultData photoCheck(MultipartFile file) {
try {
byte[] fileBytes = file.getBytes();

Загрузка…
Отмена
Сохранить