318 lines
13 KiB
Java
318 lines
13 KiB
Java
package com.gxwebsoft.common.core.utils;
|
|
|
|
|
|
import com.alibaba.fastjson.JSON;
|
|
import com.alibaba.fastjson2.JSONObject;
|
|
import com.alibaba.fastjson2.TypeReference;
|
|
import com.aliyun.green20220302.Client;
|
|
import com.aliyun.green20220302.models.*;
|
|
import com.aliyun.teaopenapi.models.Config;
|
|
import com.aliyun.teautil.models.RuntimeOptions;
|
|
import com.gxwebsoft.common.core.enums.GreenWebType;
|
|
import com.gxwebsoft.common.system.vo.faceId.HeadPortraitResult;
|
|
import org.apache.http.HttpEntity;
|
|
import org.apache.http.HttpResponse;
|
|
import org.apache.http.util.EntityUtils;
|
|
import org.springframework.stereotype.Component;
|
|
import org.springframework.util.StringUtils;
|
|
|
|
import java.util.*;
|
|
|
|
|
|
@Component
|
|
public class GreenWebUtils {
|
|
private static Client contentGreenClient = null;
|
|
|
|
static final String accessKeyId = "LTAI5t8UTh8CTXEi2dYxobhj";
|
|
static final String accessKeySecret = "fNdJOT4KAjrVrzHNAcSJuUCy9ZljD9";
|
|
|
|
static final String[] badAudioLabel = {"violence", "political_content", "sexual_content", "sexual_sounds", "contraband", "profanity", "religion", "cyberbullying", "negative_content", "specified_speaking", "specified_lyrics"};
|
|
// static final String[] badVideoLabel = {"pornographic_adultContent","pornographic_adultToys","pornographic_artwork","pornographic_adultContent_tii","sexual_suggestiveContent","sexual_breastNudity","sexual_cleavage","sexual_femaleUnderwear","sexual_maleTopless","sexual_femaleShoulder","sexual_cartoon","sexual_pregnancy","sexual_underage","political_politicalFigure_1","political_politicalFigure_2","political_politicalFigure_3","political_politicalFigure_name_tii",""};
|
|
|
|
private static class ContentGreenClientHolder {
|
|
private static final Client INSTANCE = createContentGreenClient();
|
|
}
|
|
|
|
private static Client createContentGreenClient() {
|
|
Config config = new Config();
|
|
config.setAccessKeyId(accessKeyId);
|
|
config.setAccessKeySecret(accessKeySecret);
|
|
config.setRegionId("cn-shenzhen");
|
|
config.setEndpoint("green-cip.cn-shenzhen.aliyuncs.com");
|
|
config.setReadTimeout(6000);
|
|
config.setConnectTimeout(3000);
|
|
try {
|
|
return new Client(config);
|
|
} catch (Exception e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
}
|
|
|
|
|
|
public static Client getContentGreenClient() {
|
|
|
|
if (contentGreenClient == null) {
|
|
contentGreenClient = ContentGreenClientHolder.INSTANCE;
|
|
}
|
|
return contentGreenClient;
|
|
}
|
|
|
|
/**
|
|
* 文本审核
|
|
*
|
|
* @param text
|
|
* @param type
|
|
* @return
|
|
*/
|
|
public static boolean testText(String text, GreenWebType type) {
|
|
if (!StringUtils.hasText(text)) {
|
|
return false;
|
|
}
|
|
|
|
JSONObject serviceParameters = new JSONObject();
|
|
serviceParameters.put("content", text);
|
|
|
|
// 创建RuntimeObject实例并设置运行参数。
|
|
RuntimeOptions runtime = new RuntimeOptions();
|
|
runtime.readTimeout = 10000;
|
|
runtime.connectTimeout = 10000;
|
|
|
|
TextModerationRequest textModerationRequest = new TextModerationRequest();
|
|
textModerationRequest.setService(type.name());
|
|
textModerationRequest.setServiceParameters(serviceParameters.toJSONString());
|
|
|
|
// try {
|
|
TextModerationResponse response = null;
|
|
try {
|
|
response = getContentGreenClient().textModerationWithOptions(textModerationRequest, runtime);
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return false;
|
|
}
|
|
if (response.statusCode != 200) {
|
|
return false;
|
|
}
|
|
|
|
TextModerationResponseBody result = response.getBody();
|
|
Integer code = result.getCode();
|
|
if (code != null && code == 200) {
|
|
TextModerationResponseBody.TextModerationResponseBodyData data = result.getData();
|
|
return !StringUtils.hasText(data.getReason());
|
|
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 图片审核
|
|
*
|
|
* @param imgUrl
|
|
* @param type
|
|
* @return
|
|
*/
|
|
|
|
public static boolean testImage(String imgUrl, GreenWebType type) {
|
|
|
|
if(!imgUrl.contains("?x-oss-process=image/resize")) {
|
|
imgUrl += "?x-oss-process=image/resize,w_750/quality,Q_90";
|
|
}
|
|
|
|
Map<String, String> serviceParameters = new HashMap<>();
|
|
//公网可访问的URL。
|
|
serviceParameters.put("imageUrl", imgUrl);
|
|
serviceParameters.put("dataId", UUID.randomUUID().toString());
|
|
|
|
ImageModerationRequest request = new ImageModerationRequest();
|
|
// 图片检测service: baselineCheck通用基线检测。
|
|
request.setService(type.name());
|
|
request.setServiceParameters(JSON.toJSONString(serviceParameters));
|
|
|
|
RuntimeOptions runtime = new RuntimeOptions();
|
|
runtime.readTimeout = 10000;
|
|
runtime.connectTimeout = 10000;
|
|
ImageModerationResponse response = null;
|
|
try {
|
|
response = getContentGreenClient().imageModerationWithOptions(request, runtime);
|
|
System.out.println(response);
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return false;
|
|
}
|
|
|
|
ImageModerationResponseBody body = response.getBody();
|
|
|
|
String[] unSafeLabelArr = {"pornographic_adultContent_tii","pornographic_adultContent","sexual_suggestiveContent",""};
|
|
if (body.getCode() == 200) {
|
|
ImageModerationResponseBody.ImageModerationResponseBodyData data = body.getData();
|
|
System.out.println("dataId=" + data.getDataId());
|
|
List<ImageModerationResponseBody.ImageModerationResponseBodyDataResult> results = data.getResult();
|
|
for (ImageModerationResponseBody.ImageModerationResponseBodyDataResult result : results) {
|
|
|
|
if (!"nonLabel".equals(result.getLabel())) {
|
|
System.out.println(String.format("Label: %s,Confidence: %f",result.getLabel(),result.getConfidence()));
|
|
return false;
|
|
}
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 视频审核（异步）
|
|
*
|
|
* @param videoUrl
|
|
* @param type
|
|
* @return
|
|
*/
|
|
public static String testVideo(String videoUrl, GreenWebType type) {
|
|
|
|
JSONObject serviceParameters = new JSONObject();
|
|
serviceParameters.put("url", videoUrl);
|
|
|
|
VideoModerationRequest videoModerationRequest = new VideoModerationRequest();
|
|
// 检测类型：videoDetection
|
|
videoModerationRequest.setService(type.name());
|
|
videoModerationRequest.setServiceParameters(serviceParameters.toJSONString());
|
|
|
|
|
|
try {
|
|
VideoModerationResponse response = getContentGreenClient().videoModeration(videoModerationRequest);
|
|
if (response.getStatusCode() == 200) {
|
|
VideoModerationResponseBody result = response.getBody();
|
|
System.out.println(JSON.toJSONString(result));
|
|
System.out.println("requestId = " + result.getRequestId());
|
|
System.out.println("code = " + result.getCode());
|
|
System.out.println("msg = " + result.getMessage());
|
|
Integer code = result.getCode();
|
|
if (200 == code) {
|
|
VideoModerationResponseBody.VideoModerationResponseBodyData data = result.getData();
|
|
return data.taskId;
|
|
} else {
|
|
return null;
|
|
}
|
|
} else {
|
|
return null;
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return null;
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* 获取视频审核结果
|
|
*
|
|
* @param taskId
|
|
* @param type
|
|
* @return
|
|
*/
|
|
public static boolean getTestVideoResult(String taskId, GreenWebType type) {
|
|
|
|
|
|
JSONObject serviceParameters = new JSONObject();
|
|
// 提交任务时返回的taskId。
|
|
serviceParameters.put("taskId", taskId);
|
|
VideoModerationResultRequest videoModerationResultRequest = new VideoModerationResultRequest();
|
|
videoModerationResultRequest.setService(type.name());
|
|
videoModerationResultRequest.setServiceParameters(serviceParameters.toJSONString());
|
|
try {
|
|
VideoModerationResultResponse response = getContentGreenClient().videoModerationResult(videoModerationResultRequest);
|
|
if (response.getStatusCode() == 200) {
|
|
VideoModerationResultResponseBody result = response.getBody();
|
|
System.out.println("requestId=" + result.getRequestId());
|
|
System.out.println("code=" + result.getCode());
|
|
System.out.println("msg=" + result.getMessage());
|
|
if (200 == result.getCode()) {
|
|
VideoModerationResultResponseBody.VideoModerationResultResponseBodyData data = result.getData();
|
|
VideoModerationResultResponseBody.VideoModerationResultResponseBodyDataAudioResult audioResult = data.getAudioResult();
|
|
VideoModerationResultResponseBody.VideoModerationResultResponseBodyDataFrameResult frameResult = data.getFrameResult();
|
|
boolean hasBadAudio = audioResult.sliceDetails.stream().filter(w -> {
|
|
String labels = w.getLabels();
|
|
String[] labelArr = labels.split(",");
|
|
return Arrays.stream(labelArr).filter(l -> {
|
|
return Arrays.stream(badAudioLabel).filter(b -> b.equals(l)).findFirst().isPresent();
|
|
}).findFirst().isPresent();
|
|
}).findFirst().isPresent();
|
|
if (hasBadAudio) {
|
|
return false;
|
|
}
|
|
boolean hasBadVideo = frameResult.getFrames().stream().filter(w -> {
|
|
return w.getResults().stream().filter(r -> {
|
|
return r.getResult().stream().filter(r2 -> {
|
|
return !"nonLabel".equals(r2.label);
|
|
}).findFirst().isPresent();
|
|
}).findFirst().isPresent();
|
|
}).findFirst().isPresent();
|
|
if (hasBadVideo) {
|
|
return false;
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
} catch (Exception e) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
/**
|
|
* 鉴别头像是否是真人头像（非卡通） 2. 鉴别图片是否有性感，色情信息 3. 鉴别头像是否有广告，敏感信息，以及二维码等 4. 鉴别头像性别。
|
|
*
|
|
* @param imageBase64
|
|
* @param imageType
|
|
* @param genter
|
|
* @return
|
|
*/
|
|
public static boolean testAvatar(String imageBase64, String imageType, Integer genter) {
|
|
String host = "https://edishead.market.alicloudapi.com";
|
|
String path = "/faceId/headPortrait/getResult";
|
|
String method = "POST";
|
|
String appcode = "566b5786c5874464909d8c0b7f64cdc7";
|
|
Map<String, String> headers = new HashMap<String, String>();
|
|
//最后在header中的格式(中间是英文空格)为Authorization:APPCODE 83359fd73fe94948385f570e3c139105
|
|
headers.put("Authorization", "APPCODE " + appcode);
|
|
//根据API的要求，定义相对应的Content-Type
|
|
headers.put("Content-Type", "application/x-www-form-urlencoded; charset=UTF-8");
|
|
Map<String, String> querys = new HashMap<String, String>();
|
|
Map<String, String> bodys = new HashMap<String, String>();
|
|
bodys.put("image", imageBase64);
|
|
bodys.put("imageType", imageType);
|
|
HttpResponse response = null;
|
|
try {
|
|
response = HttpUtils.doPost(host, path, method, headers, querys, bodys);
|
|
HttpEntity entity = response.getEntity();
|
|
String string = EntityUtils.toString(entity);
|
|
HeadPortraitResult result = JSONObject.parseObject(string, new TypeReference<HeadPortraitResult>() {
|
|
});
|
|
if (!"0000".equals(result.getCode())) {
|
|
return false;
|
|
}
|
|
if (!"normal".equals(result.getPorn()) || !"normal".equals(result.getAd())) {
|
|
return false;
|
|
}
|
|
if (result.getFaceCount() != 1 || result.getCartoonImg()) {
|
|
return false;
|
|
}
|
|
|
|
if (!genter.equals(result.getGenterList()[0])) {
|
|
return false;
|
|
}
|
|
|
|
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return false;
|
|
}
|
|
System.out.println(response.toString());
|
|
return true;
|
|
}
|
|
}
|