Merge branch 'dev'

徐近程
2 parents e0dd08f0 f1edbd75
Showing 4 changed files with 164 additions and 19 deletions
src/main/java/com/cnlive/shenhe/enums/ImageLabels.java
src/main/java/com/cnlive/shenhe/enums/TextLabels.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;
+    }
+}
+package com.cnlive.shenhe.enums;
+
+/**
+ * @Author: xujincheng
+ * @Date: 2020/6/22 14:08
+ */
+public enum TextLabels {
+
+    PORNOGRAPHIC("色情", 100), ADVERTISEMENT("广告", 200), VIOLENTFEAR("暴恐", 300),
+    CONTRABAND("违禁", 400), POLITICSINVOLVED("涉政", 500), ABUSE("谩骂", 600),
+    IRRIGATION("灌水", 700), OTHER("其他", 900);
+
+    private String name;
+    private int index;
+
+    TextLabels(String name, int index) {
+        this.name = name;
+        this.index = index;
+    }
+
+    public static String getName(int index) {
+        for (TextLabels c : TextLabels.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,8 @@ 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;
 import com.cnlive.shenhe.utils.CommonConst;
@@ -76,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 {
@@ -106,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");
@@ -115,12 +113,26 @@ public class YiDunUrlResultJob {
                                         JSONObject jsonObject1 = texts.getJSONObject(j);
                                         //检测结果，0：通过，1：嫌疑，2：不通过
                                         int action = jsonObject1.getIntValue("action");
+                                        String labelMsg="";
+                                        if(action!=0){
+                                            JSONArray labels = jsonObject1.getJSONArray("labels");
+                                            labelMsg="(";
+                                            if(labels!=null&&labels.size()>0){
+                                                for (Object obj : labels) {
+                                                    JSONObject jsonObjectone = (JSONObject) obj;
+                                                    Integer label = jsonObjectone.getInteger("label");
+                                                    String name = TextLabels.getName(label);
+                                                    labelMsg=labelMsg+name;
+
+                                                }
+                                            }
+                                        }
                                         if (action == 1) {
-                                            desc = desc + "文字嫌疑";
+                                            desc = desc + "文字嫌疑"+labelMsg+")";
                                             break;
                                         }
                                         if (action == 2) {
-                                            desc = desc + "文字违规";
+                                            desc = desc + "文字违规"+labelMsg+")";
                                             break;
                                         }
                                     }
@@ -131,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;
                                         }
                                     }
@@ -147,8 +177,6 @@ public class YiDunUrlResultJob {
                                 //拒绝原因
                                 shyContent.setShenhe_msg(desc);
                                 newYiDunMsg.setDesc_msg(desc);
-
-                                desc = "";
                             }
                         } catch (NumberFormatException e) {
                             e.printStackTrace();
@@ -4,6 +4,8 @@ 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;
 import org.slf4j.Logger;
@@ -76,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);
@@ -95,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);
@@ -140,17 +142,32 @@ public class YiDunAntiFraudUtil {
                     //检测结果，0：通过，1：嫌疑，2：不通过
                     int action = resultJsonObject.getIntValue("action");
                     String taskId = resultJsonObject.getString("taskId");
+                    String labelMsg="";
+                    if(action!=0){
+                        JSONArray labels = resultJsonObject.getJSONArray("labels");
+                        labelMsg="(";
+                        if(labels!=null&&labels.size()>0){
+                            for (Object obj : labels) {
+                                JSONObject jsonObjectone = (JSONObject) obj;
+                                Integer label = jsonObjectone.getInteger("label");
+                                String name = TextLabels.getName(label);
+                                labelMsg=labelMsg+name;
+
+                            }
+                        }
+                    }
+
                     if (action == 0) {
                         jsonObj.put("type", "PASS");
-                        jsonObj.put("desc", "易盾审核通过");
+                        jsonObj.put("desc", "易盾文本审核通过");
                     }
                     if (action == 1) {
                         jsonObj.put("type", "REJECT");
-                        jsonObj.put("desc", "文本嫌疑");
+                        jsonObj.put("desc", "易盾文本审核文本嫌疑"+labelMsg+")");
                     }
                     if (action == 2) {
                         jsonObj.put("type", "REJECT");
-                        jsonObj.put("desc", "文本违规");
+                        jsonObj.put("desc", "易盾文本审核文本违规"+labelMsg+")");
                     }
                     jsonObj.put("taskId", taskId);
                     logger.info("审核云文本识别处理后结果:{}", jsonObj.toString());
@@ -210,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;
                             }