mirror of
https://github.com/Hello-hao/Tbed.git
synced 2026-09-20 03:18:28 +00:00
116 lines
5.5 KiB
Java
116 lines
5.5 KiB
Java
package cn.hellohao.service.impl;
|
|
|
|
import cn.hellohao.dao.ImgMapper;
|
|
import cn.hellohao.dao.ImgreviewMapper;
|
|
import cn.hellohao.pojo.Images;
|
|
import cn.hellohao.pojo.Imgreview;
|
|
import cn.hellohao.pojo.Keys;
|
|
import cn.hellohao.pojo.Msg;
|
|
import com.alibaba.fastjson.JSON;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
import com.baidu.aip.contentcensor.AipContentCensor;
|
|
import com.baidu.aip.contentcensor.EImgType;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.scheduling.annotation.Async;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
import java.util.List;
|
|
|
|
@Service
|
|
public class ImgViolationJudgeServiceImpl {
|
|
|
|
private static Logger logger = LoggerFactory.getLogger(ImgViolationJudgeServiceImpl.class);
|
|
@Autowired
|
|
ImgreviewServiceImpl imgreviewService;
|
|
@Autowired ImgMapper imgMapper;
|
|
@Autowired deleImages deleimages;
|
|
@Autowired private ImgreviewMapper imgreviewMapper;
|
|
|
|
|
|
//synchronized
|
|
@Async("taskExecutor")
|
|
public void LegalImageCheck(Images images, Keys keys) {
|
|
Imgreview imgreview = null;
|
|
try {
|
|
imgreview = imgreviewService.selectByusing(1);
|
|
} catch (Exception e) {
|
|
logger.error("获取鉴别程序的时候发生错误");
|
|
e.printStackTrace();
|
|
}
|
|
// 判断哪个鉴别平台
|
|
if (null != imgreview) {
|
|
LegalImageCheckForBaiDu(imgreview, images, keys);
|
|
}
|
|
}
|
|
|
|
private void LegalImageCheckForBaiDu(Imgreview imgreview, Images images, Keys keys) {
|
|
// Imgreview imgreview = imgreviewService.selectByPrimaryKey(1);
|
|
if (imgreview.getUsing() == 1) {
|
|
logger.info("非法图像鉴别进程启动-BaiDu");
|
|
try {
|
|
AipContentCensor client =
|
|
new AipContentCensor(
|
|
imgreview.getAppId(),
|
|
imgreview.getApiKey(),
|
|
imgreview.getSecretKey());
|
|
client.setConnectionTimeoutInMillis(5000);
|
|
client.setSocketTimeoutInMillis(30000);
|
|
// res = client.antiPorn(images.getImgurl());
|
|
org.json.JSONObject res = client.imageCensorUserDefined(images.getImgurl(), EImgType.URL, null);
|
|
logger.info("返回的鉴黄json:" + res.toString());
|
|
com.alibaba.fastjson.JSONArray jsonArray =
|
|
JSON.parseArray("[" + res.toString() + "]");
|
|
|
|
for (Object o : jsonArray) {
|
|
JSONObject jsonObject =
|
|
(JSONObject) o;
|
|
com.alibaba.fastjson.JSONArray data = jsonObject.getJSONArray("data");
|
|
Integer conclusionType = jsonObject.getInteger("conclusionType");
|
|
if (conclusionType != null) {
|
|
if (conclusionType == 2) {
|
|
// 1：合规，2：不合规，3：疑似，4：审核失败
|
|
for (Object datum : data) {
|
|
JSONObject imgdata =
|
|
(JSONObject) datum;
|
|
if (imgdata.getInteger("type") == 1) {
|
|
logger.info("存在非法图片，进行处理操作");
|
|
// type参数
|
|
// 0:百度官方违禁图库、1：色情识别、2：暴恐识别、3：恶心图识别、4:广告检测、5：政治敏感识别、6：图像质量检测、7：用户图像黑名单、8：用户图像白名单、10：用户头像审核、11：百度官方违禁词库、12：图文审核、13:自定义文本黑名单、14:自定义文本白名单、15:EasyDL自定义模型、16：敏感旗帜标志识别、21：不良场景识别、24：直播场景审核
|
|
// 存在非法图片，进行处理操作
|
|
Images img = new Images();
|
|
img.setImgname(images.getImgname());
|
|
img.setViolation("1[1]"); // 数字是鉴别平台的主键ID，括号是非法的类型，参考上面的注释
|
|
imgMapper.setImg(img);
|
|
// 计入总数
|
|
Imgreview imgv = new Imgreview();
|
|
imgv.setId(1);
|
|
Integer count = imgreview.getCount();
|
|
imgv.setCount(count + 1);
|
|
imgreviewMapper.updateByPrimaryKeySelective(imgv);
|
|
Images imgObj =
|
|
imgMapper.selectImgUrlByImgUID(images.getImguid());
|
|
Msg dele = deleimages.dele2(Long.toString(imgObj.getId()));
|
|
List<Long> ids = (List<Long>) dele.getData();
|
|
if (!ids.contains(imgObj.getId())) {
|
|
logger.error("检测到违规图像，但是数据库删除失败");
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
logger.error("图像鉴黄线程执行过程中出现异常");
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
}
|