提交 cf656fbf 编写于 作者: minghao.wu's avatar minghao.wu

fix: 图片检测功能加入

上级 a2703148
......@@ -140,9 +140,9 @@ public class AntiSpamServiceImpl implements AntiSpamService {
}
int index = -1;
int end = -1;
while (-1 != (index = content.indexOf("<imgsrc=\\\""))) {
end = content.substring(index, content.length()).indexOf("\\\">");
imageUrlList.add(content.substring(index,index + end).replace("<imgsrc=\\\"", ""));
while (-1 != (index = content.indexOf("<imgsrc=\""))) {
end = content.substring(index, content.length()).indexOf("\">");
imageUrlList.add(content.substring(index,index + end).replace("<imgsrc=\"", ""));
content = content.replaceFirst("(<imgsrc[\\s\\S]*?>)", "");
}
......@@ -221,30 +221,25 @@ public class AntiSpamServiceImpl implements AntiSpamService {
continue;
}
/**
* 100:色情,110:性感,200:广告,210:二维码,300:暴恐,400:违禁,500:涉政
*/
int maxLevel = -1;
double maxRate = 0.0;
// 产品需根据自身需求,自行解析处理,本示例只是简单判断分类级别
for (JsonElement labelElement : labelArray) {
JsonObject lObject = labelElement.getAsJsonObject();
int label = lObject.get("label").getAsInt();
int level = lObject.get("level").getAsInt();
double rate = lObject.get("rate").getAsDouble();
if (label == 100) {
if (label == 100 || label == 110 || label == 210) {
System.out.println(String.format("label:%s, level=%s, rate=%s", label, level, rate));
maxLevel = level > maxLevel ? level : maxLevel;
maxRate = rate > maxRate ? rate : maxRate;
}
}
switch (maxLevel) {
case 0:
System.out.println("#图片机器检测结果:最高等级为\"正常\"\n");
break;
case 1:
System.out.println("#图片机器检测结果:最高等级为\"嫌疑\"\n");
break;
case 2:
System.out.println("#图片机器检测结果:最高等级为\"确定\"\n");
break;
default:
break;
if (maxLevel == 2) {
return false;
}
}
} else {
......
Markdown 格式
0% or
您添加了 0 到此讨论。请谨慎行事。
先完成此消息的编辑!
想要评论请 注册