易盾音频检测结果回调

徐近程
1 parent 83f731b0
Showing 5 changed files with 172 additions and 15 deletions
src/main/java/com/cnlive/shenhe/api/AudioControllerApi.java
src/main/java/com/cnlive/shenhe/job/AudioJob.java
src/main/java/com/cnlive/shenhe/serviceImpl/AudioServiceImpl.java
src/main/java/com/cnlive/shenhe/utils/YiDunAntiFraudUtil.java
src/main/resources/application.properties
 package com.cnlive.shenhe.api;
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.cnlive.shenhe.bean.ErrorEnum;
 import com.cnlive.shenhe.bean.ResponseBean;
 import com.cnlive.shenhe.entity.ShyAudio;
+import com.cnlive.shenhe.entity.ShyContent;
+import com.cnlive.shenhe.entity.ShyYidun;
+import com.cnlive.shenhe.enums.TextLabels;
 import com.cnlive.shenhe.serviceImpl.AudioServiceImpl;
+import com.cnlive.shenhe.serviceImpl.YiDunServiceImpl;
+import com.cnlive.shenhe.utils.AuditPass;
+import com.cnlive.shenhe.utils.CommonConst;
 import com.cnlive.shenhe.utils.YiDunAntiFraudUtil;
 import com.google.common.collect.EvictingQueue;
 import com.google.common.collect.Maps;
@@ -36,6 +43,10 @@ public class AudioControllerApi {
     @Autowired
     AudioServiceImpl audioService;
+    @Autowired
+    YiDunServiceImpl yiDunServiceImpl;
+    @Autowired
+    AuditPass Pass;
     /**
      * 音频入库
@@ -74,7 +85,7 @@ public class AudioControllerApi {
                         queue.add(shyAudio.getAudioId());
                         shyAudio.setCreateTime(new Date());
                         shyAudio.setState(0);
-                        //目前直接设置为人工审
+                        //获取审核类型
                         audioService.getShenheType(shyAudio);
                         shyAudio.setReceive(0);
                         audioService.importAudioMsg(shyAudio);
@@ -101,14 +112,87 @@ public class AudioControllerApi {
      */
     @PostMapping("/callbackAudio")
     @ResponseBody
-    public void callbackYiDunAudio(String secretId,String businessId,String signature,String callbackData) {
-        logger.info("易盾音频检测结果回调信息secretId:{},businessId:{},signature:{},callbackData:{}",secretId,businessId,signature,callbackData);
+    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);
-        logger.info("result:{}",flag);
+        logger.info("易盾音频检测校验签名结果result:{}", flag);
+        if (flag) {
+            try {
+                JSONObject jsonObject = JSONObject.parseObject("callbackData");
+                JSONArray antispamArray = jsonObject.getJSONArray("antispam");
+                if (antispamArray != null && antispamArray.size() > 0) {
+                    for (Object obj : antispamArray) {
+                        ShyYidun newYiDunMsg = new ShyYidun();
+                        ShyAudio newshyAudio = new ShyAudio();
+                        try {
+                            JSONObject objectAntispam = (JSONObject) obj;
+                            //0：通过，1：嫌疑，2：不通过
+                            int action = objectAntispam.getIntValue("action");
+                            //2：检测中，3：检测完成，4：检测失败
+                            int asrStatus = objectAntispam.getIntValue("asrStatus");
+                            String taskId = objectAntispam.getString("taskId");
+                            ShyYidun yiDunMsg = yiDunServiceImpl.getYiDunMsg(taskId);
+                            if(yiDunMsg==null){
+                                continue;
+                            }
+                            newYiDunMsg.setId(yiDunMsg.getId());
+                            newYiDunMsg.setDuration(-1);
+                            newYiDunMsg.setUpdate_time(new Date());
+                            yiDunMsg.setResult_type(String.valueOf(action));
+                            ShyAudio shyAudio = audioService.selectByPrimaryKey(Integer.parseInt(yiDunMsg.getContent_id()));
+                            newshyAudio.setId(shyAudio.getId());
+                            if(action==0&&asrStatus==3){
+                                yiDunMsg.setDesc_msg("易盾点播音频审核通过了");
+                                newshyAudio.setMsg("易盾点播音频审核通过了");
+                                newshyAudio.setUploader("机审通过");
+                                newshyAudio.setState(1);
+                                newshyAudio.setShenhe_type(CommonConst.AUDIT_TYPE_MACHANE);
+                                newshyAudio.setReceive(CommonConst.RECEIVE_AUDIT);
+                                Pass.audioPass(shyAudio.getAuditor_id(), 3, shyAudio.getAttr_tags(), "易盾点播音频审核通过了", shyAudio.getCallback());
+                            }else{
+                                String labelMsg = "";
+                                if (action != 0&&asrStatus==3) {
+                                    JSONArray labels = objectAntispam.getJSONArray("labels");
+                                    labelMsg = "(";
+                                    if (labels != null && labels.size() > 0) {
+                                        for (Object labelObj : labels) {
+                                            JSONObject jsonObjectone = (JSONObject) labelObj;
+                                            Integer label = jsonObjectone.getInteger("label");
+                                            String name = TextLabels.getName(label);
+                                            labelMsg = labelMsg + name;
+                                        }
+                                    }
+                                }
+                                if(action==1&&asrStatus==3){
+                                    yiDunMsg.setDesc_msg("易盾文本审核文本嫌疑" + labelMsg + ")");
+                                    newshyAudio.setMsg("易盾文本审核文本违规" + labelMsg + ")");
+                                }
+                                if(action==2&&asrStatus==3){
+                                    yiDunMsg.setDesc_msg("易盾文本审核文本违规" + labelMsg + ")");
+                                    newshyAudio.setMsg("易盾文本审核文本违规" + labelMsg + ")");
+                                }
+                                newshyAudio.setState(0);
+                                newshyAudio.setShenhe_type(CommonConst.AUDIT_TYPE_USER);
+                                newshyAudio.setReceive(CommonConst.RECEIVE_BEFORE);
+                            }
+                        } catch (NumberFormatException e) {
+                            e.printStackTrace();
+                            //审核类型改为人工审核
+                            newshyAudio.setShenhe_type(CommonConst.AUDIT_TYPE_USER);
+                            //拒绝原因
+                            newshyAudio.setMsg("易盾点播音频审核其它问题需排查");
+                            newYiDunMsg.setDesc_msg("易盾点播音频审核其它问题需排查");
+                        }
+                        yiDunServiceImpl.updateYiDunMsg(newYiDunMsg);
+                        audioService.updateAudioMsg(newshyAudio);
+                    }
+                }
+            } catch (NumberFormatException e) {
+                e.printStackTrace();
+            }
+        }
     }
-
-
 }
 \ No newline at end of file
@@ -2,7 +2,9 @@ package com.cnlive.shenhe.job;
 import com.cnlive.shenhe.entity.ShyAudio;
 import com.cnlive.shenhe.entity.ShyVideos;
+import com.cnlive.shenhe.entity.ShyYidun;
 import com.cnlive.shenhe.serviceImpl.AudioServiceImpl;
+import com.cnlive.shenhe.serviceImpl.YiDunServiceImpl;
 import com.cnlive.shenhe.utils.CommonConst;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -23,12 +25,14 @@ public class AudioJob {
     @Autowired
     AudioServiceImpl audioService;
+    @Autowired
+    YiDunServiceImpl yiDunServiceImpl;
     /**
      * 白名单sp音频通过
      */
     @Scheduled(fixedRate = 60 * 1000)
-    public void updateVideos() {
+    public void updateAudio() {
         ShyAudio shyAudio = new ShyAudio();
         // 审核类型为免审
         shyAudio.setShenhe_type(CommonConst.AUDIT_TYPE_FREE);
@@ -44,4 +48,29 @@ public class AudioJob {
             }
         }
     }
+
+
+    /**
+     * 定时向易盾发送点播音频进行检测
+     */
+    @Scheduled(cron = "0 0 0/1 * * ?")//每小时执行一次
+    public void audioFilter() {
+        ShyAudio shyAudio = new ShyAudio();
+        //审核类型为机审
+        shyAudio.setShenhe_type(CommonConst.AUDIT_TYPE_MACHANE);
+        //待审核状态
+        shyAudio.setState(CommonConst.AUDIT_INTT);
+        List<ShyAudio> shyAudioList = audioService.findshyAudio(shyAudio);
+        if (shyAudioList != null && shyAudioList.size() > 0) {
+            for (ShyAudio audio : shyAudioList) {
+                ShyYidun shyYidun = new ShyYidun();
+                shyYidun.setContent_id(audio.getId().toString());
+                ShyYidun shyYidun1 = yiDunServiceImpl.getidunMsgByContentId(shyYidun);
+                if (shyYidun1 == null) {
+                    logger.info("定时向易盾发送点播音频检测视频的audioId：" + audio.getAudioId());
+                    audioService.audioFilter(shyAudio);
+                }
+            }
+        }
+    }
 }
@@ -3,22 +3,24 @@ package com.cnlive.shenhe.serviceImpl;
 import com.alibaba.fastjson.JSONObject;
 import com.cnlive.shenhe.bean.ErrorEnum;
 import com.cnlive.shenhe.bean.ResponseBean;
-import com.cnlive.shenhe.entity.ShyAudio;
-import com.cnlive.shenhe.entity.ShyUsers;
-import com.cnlive.shenhe.entity.ShyVideos;
+import com.cnlive.shenhe.entity.*;
 import com.cnlive.shenhe.mapper.ShyAudioMapper;
 import com.cnlive.shenhe.mapper.ShyUsersMapper;
 import com.cnlive.shenhe.utils.AuditPass;
 import com.cnlive.shenhe.utils.CommonConst;
 import com.cnlive.shenhe.utils.CommonUtils;
+import com.cnlive.shenhe.utils.YiDunAntiFraudUtil;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
 import org.apache.ibatis.session.RowBounds;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import tk.mybatis.mapper.entity.Example;
+
+import java.util.Date;
 import java.util.List;
 /**
@@ -44,6 +46,10 @@ public class AudioServiceImpl {
     @Autowired
     SitesServiceImpl sitesService;
+    @Autowired
+    YiDunServiceImpl yiDunServiceImpl;
+    @Value("${yiDun_audio_callback}")
+    String yiDun_audio_callback;
     public ShyAudio getAudioMsg(String audioId) {
         return shyAudioMapper.getAudioMsg(audioId);
@@ -337,7 +343,7 @@ public class AudioServiceImpl {
         logger.info("sites" + sites.toString());
         //初始化一个审核状态
         //审核类型 1-免审 2-机审 3-人工审
-        Integer shenheType = 3;
+        Integer shenheType = 2;
         //如果当前sp是免审sp
         if (sites.contains(shyAudio.getSpId())) {
             //免审
@@ -345,4 +351,24 @@ public class AudioServiceImpl {
         }
         shyAudio.setShenhe_type(shenheType);
     }
+    /**
+     * 易盾点播音频过滤
+     *
+     * @param shyAudio
+     * @return
+     */
+    public void audioFilter(ShyAudio shyAudio) {
+        //调用易盾点播音频检测接口
+        JSONObject jsonObject = YiDunAntiFraudUtil.submitAudioOnDemandUrl(shyAudio.getAudioUrl(),yiDun_audio_callback);
+        if (jsonObject != null) {
+            JSONObject result = jsonObject.getJSONObject("result");
+            String taskId = result.getString("taskId");
+            ShyYidun shyYidun = new ShyYidun();
+            shyYidun.setId(taskId);
+            shyYidun.setUrl(shyAudio.getAudioUrl());
+            shyYidun.setContent_id(shyAudio.getId().toString());
+            shyYidun.setCreate_time(new Date());
+            yiDunServiceImpl.addYiDunMsg(shyYidun);
+        }
+    }
 }
@@ -113,8 +113,25 @@ public class YiDunAntiFraudUtil {
         //getURLFilter("http://3g.huabian.com/api/hz0227/show/440549.html");
         //智能点播音频审核
-        submitAudioOnDemandUrl("http://vod2017.qn.cnlive.com/60/vod/2020/0519/60_e74190fc4365479eaa206c5de9dbbe2d/ff8080817105357401722b713464073c_128.mp4");
+        //submitAudioOnDemandUrl("http://vod2017.qn.cnlive.com/60/vod/2020/0519/60_e74190fc4365479eaa206c5de9dbbe2d/ff8080817105357401722b713464073c_128.mp4","http://test.shen.cnlive.com/api/audio/callbackAudio");
+        //submitAudioOnDemandUrl("http://tjzg.vods2.cnlive.com/995/vod/2020/0628/995_9e3355af962e46aaa555dc2041c892d4/ff808081710535740172f901bf891c7e_128.mp4","http://test.shen.cnlive.com/api/audio/callbackAudio");
+        //submitAudioOnDemandUrl("http://tjzg.vods2.cnlive.com/995/vod/2020/0628/995_949677ad3e1640fbabfb014252c8e4bd/ff808081710535740172f904769c1c92_128.mp4","http://test.shen.cnlive.com/api/audio/callbackAudio");
+        //submitAudioOnDemandUrl("http://tjzg.vods2.cnlive.com/995/vod/2020/0628/995_c5fb9f1474a44906836c4f5f02070fb7/ff808081710535740172f90820321cb2_128.mp4","http://test.shen.cnlive.com/api/audio/callbackAudio");
+        //submitAudioOnDemandUrl("http://tjzg.vods2.cnlive.com/995/vod/2020/0628/995_2694acb0a2b644e3841c9c615b1bdeaf/ff808081710535740172f908280c1cbe_128.mp4","http://test.shen.cnlive.com/api/audio/callbackAudio");
+
+        /*for (int i = 0; i < 3; i++) {
+            ShyYidun shyYidun = new ShyYidun();
+            if(i==1){
+                shyYidun.setDesc_msg("aaa");
+                System.out.println(shyYidun.getDesc_msg());
+            }
+            if(i==2){
+                System.out.println(shyYidun.getDesc_msg());
+            }
+
+
+        }*/
     }
@@ -124,7 +141,7 @@ public class YiDunAntiFraudUtil {
      * @param url
      * @return
      */
-    public static JSONObject submitAudioOnDemandUrl(String url) {
+    public static JSONObject submitAudioOnDemandUrl(String url,String callbackUrl) {
         try {
             Map<String, String> params = new HashMap<>();
             // 1.设置公共参数
@@ -136,7 +153,7 @@ public class YiDunAntiFraudUtil {
             // 2.设置私有参数
             params.put("url", url);
-            params.put("callbackUrl", "http://test.shen.cnlive.com/api/audio/callbackAudio");
+            params.put("callbackUrl", callbackUrl);
             // 3.生成签名信息
             String signature = YiDunAntiFraudUtil.genSignature(TEXTANDIMAGESECRETKEY, params);
@@ -103,4 +103,5 @@ ks_domain=http://test.wideo.cnlive.com
 #ks_domain=http://sxzg.ys1.cnliveimg.com
 #\u56fe\u7247\u5b58\u653e\u6839\u8def\u5f84
-desdir=shen
 \ No newline at end of file
+desdir=shen
+yiDun_audio_callback=http://test.shen.cnlive.com/api/audio/callbackAudio
 \ No newline at end of file