添加抽帧部分信息

周伟鹏
1 parent b002079b
Showing 6 changed files with 278 additions and 20 deletions
src/main/java/com/cnlive/shenhe/api/AvsampleCallBackControllerApi.java
src/main/java/com/cnlive/shenhe/api/VideosControllerApi.java
src/main/java/com/cnlive/shenhe/entity/ShyVideos.java
src/main/java/com/cnlive/shenhe/utils/CommonConst.java
src/main/resources/application.properties
src/main/resources/mapper/ShyVideosMapper.xml
+package com.cnlive.shenhe.api;
+
+import com.alibaba.fastjson.JSONObject;
+import com.cnlive.shenhe.entity.ShyVideos;
+import com.cnlive.shenhe.serviceImpl.VideosServiceImpl;
+import com.cnlive.shenhe.utils.CommonConst;
+import com.cnlive.shenhe.utils.CommonUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+
+import java.util.List;
+
+/**
+ * @Auther: 周伟鹏
+ * @Date: 2018/12/21 13:43
+ * @Description:抽帧回调接口
+ */
+@Controller
+@RequestMapping("/api/avsample")
+public class AvsampleCallBackControllerApi {
+
+    private static Logger logger = LoggerFactory.getLogger(AvsampleCallBackControllerApi.class);
+
+    @Autowired
+    VideosServiceImpl videosService;
+
+    /**
+     * 接收抽帧结果
+     * @param json
+     */
+    @PostMapping("/callback")
+    public void callback(@RequestBody JSONObject json) {
+        logger.info("抽帧回调请求。。。。json=="+json.toJSONString());
+        String taskId = json.getString("taskId");
+        if(CommonUtils.isNotEmpty(taskId)){
+            ShyVideos shyVideo = new ShyVideos();
+            shyVideo.setTask_id(taskId);
+            List<ShyVideos> shyVideosList = videosService.findshyVideo(shyVideo);
+            if(!shyVideosList.isEmpty()){
+                ShyVideos video = shyVideosList.get(0);
+                Integer code = json.getInteger("code");
+                if(code==3){
+                    video.setAvsample(CommonConst.AVSAMPLE_SUCCESS);
+                    video.setAvsample_imgs(json.getJSONArray("images").toJSONString());
+                }else{
+                    video.setAvsample(CommonConst.AVSAMPLE_FAIL);
+                }
+                video.setAvsample_msg(json.toJSONString());
+                videosService.updateshyVideos(video);
+            }
+
+        }
+    }
+}
 package com.cnlive.shenhe.api;
  
+import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.cnlive.shenhe.bean.ErrorEnum;
 import com.cnlive.shenhe.bean.ResponseBean;
 import com.cnlive.shenhe.entity.ShyVideos;
 import com.cnlive.shenhe.serviceImpl.UsersServiceImpl;
 import com.cnlive.shenhe.serviceImpl.VideosServiceImpl;
-import com.cnlive.shenhe.utils.AuditPass;
-import com.cnlive.shenhe.utils.CommonConst;
-import com.cnlive.shenhe.utils.CommonUtils;
+import com.cnlive.shenhe.utils.*;
+import com.sun.org.apache.bcel.internal.generic.NEW;
 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.Controller;
 import org.springframework.web.bind.annotation.PostMapping;
 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.List;
+import java.util.Map;
  
 /**
  * @Auther: chenhao
@@ -35,6 +39,25 @@ public class VideosControllerApi {
     @Autowired
     UsersServiceImpl usersService;
  
+    @Value("${avsmple_appkey}")
+    String appkey;
+
+    @Value("${avsmple_url}")
+    String url;
+
+    @Value("${avsmple_callback}")
+    String callback;
+
+    @Value("${qn_bucket}")
+    String qn_bucket;
+
+    @Value("${ks_bucket}")
+    String ks_bucket;
+
+    @Value("${desdir}")
+    String desdir;
+
+
     @PostMapping("/import")
     @ResponseBody
     public ResponseBean impotVideo(String param) {
@@ -64,8 +87,10 @@ public class VideosControllerApi {
             shyVideos.setVideo_upload_time(json.getDate(" uploadTime"));
             shyVideos.setVideo_priority(json.getString("priority"));
             shyVideos.setCallback(json.getString("callback"));
-            shyVideos.setSpid(Integer.parseInt(json.getString("spId")));
+            shyVideos.setSpid(json.getInteger("spId"));
             shyVideos.setState(CommonConst.AUDIT_INTT);
+            shyVideos1.setBucket(json.getString("bucketName"));
+            shyVideos1.setPlat(json.getInteger("bucketType"));
             boolean YorN = videosService.updateshyVideos(shyVideos);
             if (YorN == true) {
                 return new ResponseBean(ErrorEnum.SUCCESS);
@@ -88,8 +113,11 @@ public class VideosControllerApi {
         shyVideos1.setVideo_upload_time(json.getDate("uploadTime"));
         shyVideos1.setVideo_priority(json.getString("priority"));
         shyVideos1.setCallback(json.getString("callback"));
-        shyVideos1.setSpid(Integer.parseInt(json.getString("spId")));
+        shyVideos1.setSpid(json.getInteger("spId"));
         shyVideos1.setState(CommonConst.AUDIT_INTT);
+        shyVideos1.setBucket(json.getString("bucketName"));
+        shyVideos1.setPlat(json.getInteger("bucketType"));
+
         boolean result = videosService.impotVideo(shyVideos1);
         if (result == true) {
             return new ResponseBean(ErrorEnum.SUCCESS);
@@ -97,4 +125,65 @@ public class VideosControllerApi {
             return new ResponseBean(ErrorEnum.ERROR);
         }
     }
+
+    /**
+     * 申请抽帧
+     * @param video
+     * @return
+     */
+    public ShyVideos avsample(ShyVideos video) {
+        try {
+            JSONObject params = new JSONObject();
+            String video_url = video.getVideo_url();
+            Integer plat = video.getPlat();
+            String bucket = video.getBucket();
+            Integer spid = video.getSpid();
+
+            JSONObject attach = new JSONObject();
+            attach.put("num",24);
+            if (CommonConst.KS_PLAT == plat) {
+                //金山
+                params.put("plat", 0);
+                params.put("dstBucket",ks_bucket);
+            } else if (CommonConst.QINIU_PLAT == plat) {
+                //七牛
+                params.put("plat", 1);
+                params.put("dstBucket",qn_bucket);
+            }
+            String path = video_url.replace("//", "");
+            path = "/" + bucket + path.substring(path.indexOf("/"), path.length());
+            String dstDir = desdir+"/"+spid+"/"+CommonUtils.getCurrentDate("yyyy/MM/dd");
+            String formats = video_url.substring(video_url.indexOf(".")+1);
+            params.put("path",path);
+            params.put("dstDir",dstDir);
+            params.put("type","avsample");
+            params.put("formats",formats.toLowerCase());
+            params.put("spId",spid);
+            params.put("cbUrl",callback);
+            params.put("attach",attach);
+
+            HashMap<String, String> dataMap = new HashMap<>();
+            dataMap.put("timestamp", new Date().getTime() / 1000 + "");
+            dataMap.put("params",params.toJSONString());
+            String httpUrl = OpenUtil.buildURL(url, dataMap, appkey);
+            logger.info("抽帧请求httpUrl===" + httpUrl);
+            Map<String, String> resultMap = HttpUtil.init().post(httpUrl);
+            JSONObject result = JSON.parseObject(resultMap.get("result"));
+            Integer code = result.getInteger("code");
+            if(code==0){
+                video.setAvsample(CommonConst.AVSAMPLE_ING);
+            }else{
+                video.setAvsample(CommonConst.AVSAMPLE_FAIL);
+            }
+            video.setTask_id(result.getString("taskId"));
+            video.setAvsample_msg(result.toJSONString());
+        } catch (Exception e) {
+            logger.error("请求抽帧失败,video_id:{}", video.getVideo_id(), e);
+            video.setAvsample(CommonConst.AVSAMPLE_FAIL);
+        }
+        return video;
+    }
+
+
+
 }
@@ -96,9 +96,9 @@ public class ShyVideos {
     private String receive_user_id;
  
     /**
-     * 是否抽帧
+     * 抽帧状态：0:未抽帧；1:抽帧中；2:抽帧成功；3:抽帧失败
      */
-    private Boolean avsample;
+    private Integer avsample;
  
     /**
      * 抽帧图片地址
@@ -111,11 +111,34 @@ public class ShyVideos {
     private String avsample_msg;
  
     /**
+     * 抽帧请求标记
+     */
+    private String task_id;
+
+    /**
+     * 视频存储所在的bucket
+     */
+    private String bucket;
+
+    /**
+     * 视频存放位置1:金山；2:七牛
+     */
+    private Integer plat;
+
+    /**
      * 审核员
      */
     @Transient
     private String auditor;
  
+    public String getAuditor() {
+        return auditor;
+    }
+
+    public void setAuditor(String auditor) {
+        this.auditor = auditor;
+    }
+
     /**
      * @return id
      */
@@ -471,20 +494,20 @@ public class ShyVideos {
     }
  
     /**
-     * 获取是否抽帧
+     * 获取抽帧状态：0:未抽帧；1:抽帧中；2:抽帧成功；3:抽帧失败
      *
-     * @return avsample - 是否抽帧
+     * @return avsample - 抽帧状态：0:未抽帧；1:抽帧中；2:抽帧成功；3:抽帧失败
      */
-    public Boolean getAvsample() {
+    public Integer getAvsample() {
         return avsample;
     }
  
     /**
-     * 设置是否抽帧
+     * 设置抽帧状态：0:未抽帧；1:抽帧中；2:抽帧成功；3:抽帧失败
      *
-     * @param avsample 是否抽帧
+     * @param avsample 抽帧状态：0:未抽帧；1:抽帧中；2:抽帧成功；3:抽帧失败
      */
-    public void setAvsample(Boolean avsample) {
+    public void setAvsample(Integer avsample) {
         this.avsample = avsample;
     }
  
@@ -524,11 +547,57 @@ public class ShyVideos {
         this.avsample_msg = avsample_msg;
     }
  
-    public String getAuditor() {
-        return auditor;
+    /**
+     * 获取抽帧请求标记
+     *
+     * @return task_id - 抽帧请求标记
+     */
+    public String getTask_id() {
+        return task_id;
     }
  
-    public void setAuditor(String auditor) {
-        this.auditor = auditor;
+    /**
+     * 设置抽帧请求标记
+     *
+     * @param task_id 抽帧请求标记
+     */
+    public void setTask_id(String task_id) {
+        this.task_id = task_id;
+    }
+
+    /**
+     * 获取视频存储所在的bucket
+     *
+     * @return bucket - 视频存储所在的bucket
+     */
+    public String getBucket() {
+        return bucket;
+    }
+
+    /**
+     * 设置视频存储所在的bucket
+     *
+     * @param bucket 视频存储所在的bucket
+     */
+    public void setBucket(String bucket) {
+        this.bucket = bucket;
+    }
+
+    /**
+     * 获取视频存放位置1:金山；2:七牛
+     *
+     * @return plat - 视频存放位置1:金山；2:七牛
+     */
+    public Integer getPlat() {
+        return plat;
+    }
+
+    /**
+     * 设置视频存放位置1:金山；2:七牛
+     *
+     * @param plat 视频存放位置1:金山；2:七牛
+     */
+    public void setPlat(Integer plat) {
+        this.plat = plat;
     }
 }
 \ No newline at end of file
@@ -55,4 +55,31 @@ public class CommonConst {
  
     public static JSONObject BUSINESS = JSONObject.parseObject("{\"1\":\"点播业务\",\"2\":\"评论业务\"}");
  
+    /**
+     * 抽帧状态：未抽帧
+     */
+    public static Integer AVSAMPLE_INTIT = 0;
+    /**
+     * 抽帧状态：抽帧中
+     */
+    public static Integer AVSAMPLE_ING = 1;
+    /**
+     * 抽帧状态：抽帧完成
+     */
+    public static Integer AVSAMPLE_SUCCESS = 2;
+    /**
+     * 抽帧状态：抽帧失败
+     */
+    public static Integer AVSAMPLE_FAIL = 3;
+
+    /**
+     * 金山标记
+     */
+    public static Integer KS_PLAT = 1;
+    /**
+     * 七牛标记
+     */
+    public static Integer QINIU_PLAT = 2;
+
+
 }
@@ -36,5 +36,16 @@ url=http://mam.innerapi.cnlive.com/v1/inner/ShenHeInfo/shenheNotify
 platformKey =AUDIT
 platformValue=10011438
  
-#评论回调地址
-commenturl=http://comment.cnlive.com/services/commentForHd/auditCallBack
 \ No newline at end of file
+#抽帧常量
+avsmple_appkey=6962df5e02d846f490d64f3a68ac89fc
+avsmple_url=http://transcode.cnlive.com/api/request
+avsmple_callback=http://test.shen.cnlive.com/api/avsample/callback
+
+#抽帧存放bucket和域名
+qn_bucket=cnlivetest
+qn_domain=http://test.qnvod.cnlive.com
+ks_bucket=cnlive-test
+ks_domain=http://test.wideo.cnlive.com
+
+#图片存放根路径
+desdir=shen
@@ -26,8 +26,11 @@
     <result column="updated_at" jdbcType="TIMESTAMP" property="updated_at" />
     <result column="receive" jdbcType="BIT" property="receive" />
     <result column="receive_user_id" jdbcType="VARCHAR" property="receive_user_id" />
-    <result column="avsample" jdbcType="BIT" property="avsample" />
+    <result column="avsample" jdbcType="INTEGER" property="avsample" />
     <result column="avsample_imgs" jdbcType="VARCHAR" property="avsample_imgs" />
     <result column="avsample_msg" jdbcType="VARCHAR" property="avsample_msg" />
+    <result column="task_id" jdbcType="VARCHAR" property="task_id" />
+    <result column="bucket" jdbcType="VARCHAR" property="bucket" />
+    <result column="plat" jdbcType="INTEGER" property="plat" />
   </resultMap>
 </mapper>
 \ No newline at end of file