update

徐近程
1 parent 380513a4
Showing 2 changed files with 50 additions and 28 deletions
src/main/java/com/cnlive/shenhe/api/AudioControllerApi.java
src/main/java/com/cnlive/shenhe/utils/YiDunAntiFraudUtil.java
@@ -5,7 +5,10 @@ import com.cnlive.shenhe.bean.ErrorEnum;
 import com.cnlive.shenhe.bean.ResponseBean;
 import com.cnlive.shenhe.entity.ShyAudio;
 import com.cnlive.shenhe.serviceImpl.AudioServiceImpl;
+import com.cnlive.shenhe.utils.YiDunAntiFraudUtil;
 import com.google.common.collect.EvictingQueue;
+import com.google.common.collect.Maps;
+import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -15,6 +18,8 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.ResponseBody;
 import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
 /**
  * @Author: xujincheng
@@ -98,6 +103,12 @@ public class AudioControllerApi {
     @ResponseBody
     public void callbackYiDunAudio(String secretId,String businessId,String signature,String callbackData) {
         logger.info("易盾音频检测结果回调信息secretId:{},businessId:{},signature:{},callbackData:{}",secretId,businessId,signature,callbackData);
+        boolean flag = YiDunAntiFraudUtil.verifySignature(secretId, businessId, signature, callbackData);
+        System.out.println(flag);
+
     }
+
+
+
 }
 \ No newline at end of file
@@ -51,7 +51,6 @@ public class YiDunAntiFraudUtil {
     private final static String URL_RESULT = "https://as.dun.163yun.com/v1/crawler/callback/results";
-
     /**
      * 文本识别 图片识别 点播音频识别产品密钥ID，产品标识
      */
@@ -160,19 +159,6 @@ public class YiDunAntiFraudUtil {
     }
-
-
-
-
-
-
-
-
-
-
-
-
-
     /**
      * 智能文本审核
      *
@@ -211,16 +197,16 @@ public class YiDunAntiFraudUtil {
                     //检测结果，0：通过，1：嫌疑，2：不通过
                     int action = resultJsonObject.getIntValue("action");
                     String taskId = resultJsonObject.getString("taskId");
-                    String labelMsg="";
-                    if(action!=0){
+                    String labelMsg = "";
+                    if (action != 0) {
                         JSONArray labels = resultJsonObject.getJSONArray("labels");
-                        labelMsg="(";
-                        if(labels!=null&&labels.size()>0){
+                        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;
+                                labelMsg = labelMsg + name;
                             }
                         }
@@ -232,11 +218,11 @@ public class YiDunAntiFraudUtil {
                     }
                     if (action == 1) {
                         jsonObj.put("type", "REJECT");
-                        jsonObj.put("desc", "易盾文本审核文本嫌疑"+labelMsg+")");
+                        jsonObj.put("desc", "易盾文本审核文本嫌疑" + labelMsg + ")");
                     }
                     if (action == 2) {
                         jsonObj.put("type", "REJECT");
-                        jsonObj.put("desc", "易盾文本审核文本违规"+labelMsg+")");
+                        jsonObj.put("desc", "易盾文本审核文本违规" + labelMsg + ")");
                     }
                     jsonObj.put("taskId", taskId);
                     logger.info("审核云文本识别处理后结果:{}", jsonObj.toString());
@@ -297,18 +283,18 @@ public class YiDunAntiFraudUtil {
                             //建议动作，2：建议删除，1：建议审核，0：建议通过
                             int action = jsonObject.getIntValue("action");
-                            String labelMsg="";
-                            if(action!=0){
+                            String labelMsg = "";
+                            if (action != 0) {
                                 JSONArray labels = jsonObject.getJSONArray("labels");
-                                if(labels!=null&&labels.size()>0){
+                                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){
+                                        if (level != 0) {
                                             Integer label = jsonObjectone.getInteger("label");
                                             String name = ImageLabels.getName(label);
-                                            labelMsg=labelMsg+name;
+                                            labelMsg = labelMsg + name;
                                         }
                                     }
                                 }
@@ -316,12 +302,12 @@ public class YiDunAntiFraudUtil {
                             }
                             if (status == 0 && action == 1) {
-                                strBuffer.append("(第" + (i + 1) + "张图片嫌疑"+labelMsg+")");
+                                strBuffer.append("(第" + (i + 1) + "张图片嫌疑" + labelMsg + ")");
                                 count++;
                                 continue;
                             }
                             if (status == 0 && action == 2) {
-                                strBuffer.append("(第" + (i + 1) + "张图片违规"+labelMsg+")");
+                                strBuffer.append("(第" + (i + 1) + "张图片违规" + labelMsg + ")");
                                 count++;
                                 continue;
                             }
@@ -483,4 +469,29 @@ public class YiDunAntiFraudUtil {
         return null;
     }
+
+    /**
+     * 做签名验证
+     *
+     * @param secretId
+     * @param businessId
+     * @param signature
+     * @param callbackData
+     * @return
+     */
+    public static boolean verifySignature(String secretId, String businessId, String signature, String callbackData) {
+        if (org.apache.commons.lang3.StringUtils.isEmpty(secretId) || org.apache.commons.lang3.StringUtils.isEmpty(signature)) {
+            // 签名参数为空，直接返回失败
+            return false;
+        }
+        Map<String, String> params = new HashMap<>(3);
+        params.put("secretId", secretId);
+        params.put("businessId", businessId);
+        params.put("callbackData", callbackData);
+
+        // SECRETKEY:产品私有密钥 SECRETID:产品密钥ID BUSINESSID:业务ID,开通服务时，易盾会提供相关密钥信息
+        String serverSignature = genSignature(TEXTANDIMAGESECRETKEY, params);
+        return signature.equals(serverSignature) && TEXTANDIMAGESECRETID.equals(secretId) && AUDIOONDEMANDBUSINESSID.equals(businessId);
+    }
+
 }