易盾图片解析update

徐近程
1 parent 1fe20bb4
Showing 3 changed files with 87 additions and 14 deletions
src/main/java/com/cnlive/shenhe/enums/ImageLabels.java
src/main/java/com/cnlive/shenhe/job/YiDunUrlResultJob.java
src/main/java/com/cnlive/shenhe/utils/YiDunAntiFraudUtil.java
+package com.cnlive.shenhe.enums;
+
+/**
+ * @Author: xujincheng
+ * @Date: 2020/6/22 15:30
+ */
+public enum ImageLabels {
+    PORNOGRAPHIC("色情", 100), SEXYANDVULGAR("性感低俗", 110), ADVERTISEMENT("广告", 200),
+    QRCODE("二维码", 210), POLITICSINVOLVED("涉政", 300), ABUSE("谩骂", 400),
+    IRRIGATION("灌水", 500), OTHER("其他", 900);
+
+    private String name;
+    private int index;
+
+    ImageLabels(String name, int index) {
+        this.name = name;
+        this.index = index;
+    }
+
+    public static String getName(int index) {
+        for (ImageLabels c : ImageLabels.values()) {
+            if (c.getIndex() == index) {
+                return c.name;
+            }
+        }
+        return null;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public int getIndex() {
+        return index;
+    }
+}
@@ -4,6 +4,7 @@ import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.cnlive.shenhe.entity.ShyContent;
 import com.cnlive.shenhe.entity.ShyYidun;
+import com.cnlive.shenhe.enums.ImageLabels;
 import com.cnlive.shenhe.enums.TextLabels;
 import com.cnlive.shenhe.serviceImpl.ContentServiceImpl;
 import com.cnlive.shenhe.serviceImpl.YiDunServiceImpl;
@@ -77,14 +78,11 @@ public class YiDunUrlResultJob {
             JSONObject resultObject = JSONObject.parseObject(s1);
             int code = resultObject.getIntValue("code");
             String msg = resultObject.getString("msg");
-            //易盾返回结果的类型
-            String type;
-            //易盾失败结果的详情
-            String desc = "易盾网站审核";
             if (code == 200 && "ok".equals(msg)) {
                 JSONArray resultJSONArray = resultObject.getJSONArray("result");
                 if (resultJSONArray != null && resultJSONArray.size() > 0) {
                     for (int i = 0; i < resultJSONArray.size(); i++) {
+                        String desc = "易盾网站审核";
                         ShyYidun newYiDunMsg = new ShyYidun();
                         ShyContent shyContent = new ShyContent();
                         try {
@@ -107,7 +105,6 @@ public class YiDunUrlResultJob {
                                 newYiDunMsg.setDesc_msg(desc);
                                 //修改图文状态
                                 contentServiceImpl.callback(yiDunMsg.getContent_id(), desc, CommonConst.AUDIT_SUCCESS, "", "机审通过");
-                                desc="易盾网站审核";
                             } else {
                                 JSONObject evidences = jsonObject.getJSONObject("evidences");
                                 JSONArray texts = evidences.getJSONArray("texts");
@@ -146,12 +143,30 @@ public class YiDunUrlResultJob {
                                         JSONObject jsonObject1 = images.getJSONObject(k);
                                         //0：正常，1：不确定，2：确定
                                         int level = jsonObject1.getIntValue("level");
+                                        String labelMsg="";
+                                        if(level!=0){
+                                            JSONArray labels = jsonObject1.getJSONArray("labels");
+                                            labelMsg="(";
+                                            if(labels!=null&&labels.size()>0){
+                                                for (Object obj : labels) {
+                                                    JSONObject jsonObjectone = (JSONObject) obj;
+                                                    int labelsLevel = jsonObjectone.getIntValue("level");
+                                                    //分类级别  0-正常 1-不确定 2-确定
+                                                    if(labelsLevel!=0){
+                                                        Integer label = jsonObjectone.getInteger("label");
+                                                        String name = ImageLabels.getName(label);
+                                                        labelMsg=labelMsg+name;
+                                                    }
+                                                }
+                                            }
+
+                                        }
                                         if (level == 1) {
-                                            desc = desc + "第" + (i + 1) + "张图片嫌疑";
+                                            desc = desc + "第" + (i + 1) + "张图片嫌疑"+labelMsg+")";
                                             break;
                                         }
                                         if (level == 2) {
-                                            desc = desc + "第" + (i + 1) + "张图片违规";
+                                            desc = desc + "第" + (i + 1) + "张图片违规"+labelMsg+")";
                                             break;
                                         }
                                     }
@@ -162,8 +177,6 @@ public class YiDunUrlResultJob {
                                 //拒绝原因
                                 shyContent.setShenhe_msg(desc);
                                 newYiDunMsg.setDesc_msg(desc);
-
-                                desc = "";
                             }
                         } catch (NumberFormatException e) {
                             e.printStackTrace();
@@ -4,6 +4,7 @@ import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.cnlive.shenhe.api.AvsampleCallBackControllerApi;
 import com.cnlive.shenhe.entity.ShyYidun;
+import com.cnlive.shenhe.enums.ImageLabels;
 import com.cnlive.shenhe.enums.TextLabels;
 import com.cnlive.shenhe.serviceImpl.YiDunServiceImpl;
 import org.apache.commons.codec.digest.DigestUtils;
@@ -77,10 +78,10 @@ public class YiDunAntiFraudUtil {
     public static void main(String[] args) {
         //智能文本识别审核测试
-        getTextFilter("王八蛋");
+        //getTextFilter("王八蛋");
         //智能图片识别审核测试
-        /*JSONArray jsonArray = new JSONArray();
+        JSONArray jsonArray = new JSONArray();
         JSONObject image1 = new JSONObject();
         image1.put("name", UUID.randomUUID());
         image1.put("type", 1);
@@ -96,7 +97,7 @@ public class YiDunAntiFraudUtil {
         image3.put("type", 1);
         image3.put("data", "https://wjj.ys2.cnliveimg.com/769/img/0/2020/03/21/158474856886077030277.jpg");
         jsonArray.add(image3);
-        getImagesFilter(jsonArray.toJSONString());*/
+        getImagesFilter(jsonArray.toJSONString());
         //智能网页识别审核测试
         //getURLFilter(FILE_URL);
@@ -226,13 +227,32 @@ public class YiDunAntiFraudUtil {
                             int status = jsonObject.getIntValue("status");
                             //建议动作，2：建议删除，1：建议审核，0：建议通过
                             int action = jsonObject.getIntValue("action");
+
+                            String labelMsg="";
+                            if(action!=0){
+                                JSONArray labels = jsonObject.getJSONArray("labels");
+                                if(labels!=null&&labels.size()>0){
+                                    for (Object obj : labels) {
+                                        JSONObject jsonObjectone = (JSONObject) obj;
+                                        int level = jsonObjectone.getIntValue("level");
+                                        //分类级别  0-正常 1-不确定 2-确定
+                                        if(level!=0){
+                                            Integer label = jsonObjectone.getInteger("label");
+                                            String name = ImageLabels.getName(label);
+                                            labelMsg=labelMsg+name;
+                                        }
+                                    }
+                                }
+
+                            }
+
                             if (status == 0 && action == 1) {
-                                strBuffer.append("(第" + (i + 1) + "张图片嫌疑)");
+                                strBuffer.append("(第" + (i + 1) + "张图片嫌疑"+labelMsg+")");
                                 count++;
                                 continue;
                             }
                             if (status == 0 && action == 2) {
-                                strBuffer.append("(第" + (i + 1) + "张图片违规)");
+                                strBuffer.append("(第" + (i + 1) + "张图片违规"+labelMsg+")");
                                 count++;
                                 continue;
                             }