易盾文本审核

徐近程
1 parent af4fec74
Showing 1 changed file with 56 additions and 29 deletions
src/main/java/com/cnlive/shenhe/utils/YiDunAntiFraudUtil.java
@@ -79,10 +79,10 @@ public class YiDunAntiFraudUtil {
         //getURLFilter("http://3g.huabian.com/api/hz0227/show/440549.html");
  
         //智能文本识别审核测试
-        //getTextFilter("王八蛋");
+        getTextFilter("好人");
  
         //智能图片识别审核测试
-        JSONArray jsonArray = new JSONArray();
+        /*JSONArray jsonArray = new JSONArray();
         JSONObject image1 = new JSONObject();
         image1.put("name", UUID.randomUUID());
         image1.put("type", 1);
@@ -99,7 +99,7 @@ public class YiDunAntiFraudUtil {
         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());*/
  
     }
  
@@ -178,7 +178,7 @@ public class YiDunAntiFraudUtil {
             e.printStackTrace();
         }
         jsonObj.put("type", "REJECT");
-        jsonObj.put("desc", "易盾其它问题需排查");
+        jsonObj.put("desc", "易盾图片识别其它问题需排查");
         logger.info("审核云图片识别处理后结果:{}", jsonObj.toString());
         return jsonObj;
     }
@@ -190,35 +190,62 @@ public class YiDunAntiFraudUtil {
      * @return
      */
     public static JSONObject getTextFilter(String text) {
-        Map<String, String> params = new HashMap<String, String>();
-        // 1.设置公共参数
-        params.put("secretId", TEXTANDIMAGESECRETID);
-        params.put("businessId", TEXTBUSINESSID);
-        params.put("version", "v3.1");
-        params.put("timestamp", String.valueOf(System.currentTimeMillis()));
-        params.put("nonce", String.valueOf(new Random().nextInt()));
+        JSONObject jsonObj = new JSONObject();
+        try {
+            Map<String, String> params = new HashMap<String, String>();
+            // 1.设置公共参数
+            params.put("secretId", TEXTANDIMAGESECRETID);
+            params.put("businessId", TEXTBUSINESSID);
+            params.put("version", "v3.1");
+            params.put("timestamp", String.valueOf(System.currentTimeMillis()));
+            params.put("nonce", String.valueOf(new Random().nextInt()));
  
-        // 2.设置私有参数
-        params.put("dataId", UUID.randomUUID().toString());
-        params.put("content", text);
-        // params.put("dataType", "1");
-        // params.put("ip", "123.115.77.137");
-        // params.put("account", "java@163.com");
-        // params.put("deviceType", "4");
-        // params.put("deviceId", "92B1E5AA-4C3D-4565-A8C2-86E297055088");
-        // params.put("callback", "ebfcad1c-dba1-490c-b4de-e784c2691768");
-        // params.put("publishTime", String.valueOf(System.currentTimeMillis()));
+            // 2.设置私有参数
+            params.put("dataId", UUID.randomUUID().toString());
+            params.put("content", text);
  
-        // 3.生成签名信息
-        String signature = genSignature(TEXTANDIMAGESECRETKEY, params);
-        params.put("signature", signature);
+            // 3.生成签名信息
+            String signature = genSignature(TEXTANDIMAGESECRETKEY, params);
+            params.put("signature", signature);
  
-        // 4.发送HTTP请求
-        String s = HttpClientUtils.post_str(TEXT_RESULT, params);
-        System.out.println(s);
+            // 4.发送HTTP请求
+            String result = HttpClientUtils.post_str(TEXT_RESULT, params);
+            logger.info("易盾智能文本识别接口返回信息:{}", result);
  
-        JSONObject resultObject = JSONObject.parseObject(s);
-        return resultObject;
+            if (!StringUtils.isEmpty(result)) {
+                JSONObject jsonResult = JSONObject.parseObject(result);
+                int code = jsonResult.getIntValue("code");
+                String msg = jsonResult.getString("msg");
+                JSONObject resultJsonObject = jsonResult.getJSONObject("result");
+                if (code == 200 && "ok".equals(msg) && !StringUtils.isEmpty(resultJsonObject)) {
+                    //检测结果，0：通过，1：嫌疑，2：不通过
+                    int action = resultJsonObject.getIntValue("action");
+                    String taskId = resultJsonObject.getString("taskId");
+                    if (action == 0) {
+                        jsonObj.put("type", "PASS");
+                        jsonObj.put("desc", "易盾审核通过");
+                    }
+                    if (action == 1) {
+                        jsonObj.put("type", "REJECT");
+                        jsonObj.put("desc", "文本嫌疑");
+                    }
+                    if (action == 2) {
+                        jsonObj.put("type", "REJECT");
+                        jsonObj.put("desc", "文本违规");
+                    }
+                    jsonObj.put("taskId", taskId);
+                    logger.info("审核云文本识别处理后结果:{}", jsonObj.toString());
+                    return jsonObj;
+                }
+
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        jsonObj.put("type", "REJECT");
+        jsonObj.put("desc", "易盾文本识别其它问题需排查");
+        logger.info("审核云文本识别处理后结果:{}", jsonObj.toString());
+        return jsonObj;
     }
  
     /**