资讯模块数据——视频功能修复

wisenergy-common/src/main/java/cn/wisenergy/common/utils/FrameGrabberKit.java

-//package cn.wisenergy.common.utils;
-//
-//
-//import java.awt.Image;
-//import java.awt.image.BufferedImage;
-//import java.io.File;
-//import javax.imageio.ImageIO;
-//
-//import org.bytedeco.javacpp.opencv_core;
-//import org.bytedeco.javacv.FFmpegFrameGrabber;
-//import org.bytedeco.javacv.Frame;
-//
-//
-///**
-// * 截取视频图片
-// * @ClassName:  FrameGrabberKit
-// */
-//public class FrameGrabberKit {
-//
-//    /**
-//     * 获取视频图片
-//     * @param videofile  源视频文件路径
-//     * @param framefile  截取帧的图片存放路径
-//     * @param imgfile 视频封面图保存路径
-//     * @return
-//     */
-//    public static String getVedioImg(String videofile, String framefile,String imgfile){
-//        String ImgUrl="";
-//        //截取封面图
-//        try {
-//            fetchFrame(videofile, framefile);
-//        } catch (Exception e) {
-//            e.printStackTrace();
-//        }
-//        // 完整的ImgUrl
-//        ImgUrl = imgfile;//视频封面图保存路径
-//        return ImgUrl;
-//    }
-//    /**
-//     * 获取指定视频的帧并保存为图片至指定目录
-//     * @param videofile  源视频文件路径
-//     * @param framefile  截取帧的图片存放路径 例：F:\hfkjrecorder\target\4.jpg
-//     * @throws Exception
-//     */
-//    public static void fetchFrame(String videofile, String framefile) throws Exception {
-//        long start = System.currentTimeMillis();
-//        File targetFile = new File(framefile);
-//        FFmpegFrameGrabber ff = new FFmpegFrameGrabber(videofile);
-//        ff.start();
-//        int lenght = ff.getLengthInFrames();
-//        int i = 0;
-//        int interceptionFrames = 30;//截取第几帧
-//        //默认截取第50帧，如果第50帧大于视频总帧数的8成直接取长度lenght * 0.3
-//        if(interceptionFrames >= lenght * 0.8) {
-//            interceptionFrames = (int)(lenght * 0.3);
-//        }
-//        Frame f = null;
-//        while (i < lenght) {
-//            // 过滤 前  interceptionFrames 帧，避免出现全黑的图片，依自己情况而定
-//            f = ff.grabFrame();
-//            if ((i > interceptionFrames) && (f.image != null)) {
-//                break;
-//            }
-//            i++;
-//        }
-//        opencv_core.IplImage img = f.image;
-//        int owidth = img.width();
-//        int oheight = img.height();
-//        // 对截取的帧进行等比例缩放 宽350、高160
-////        if(owidth > oheight) {//宽大于高
-////
-////        }else {//高大于宽
-////
-////        }
-//        int width = 800;
-//        int height = (int) (((double) width / owidth) * oheight);
-//        /**
-//         width - 所创建图像的宽度
-//         height - 所创建图像的高度
-//         imageType - 所创建图像的类型
-//         TYPE_3BYTE_BGR - 表示一个具有 8 位 RGB 颜色分量的图像，对应于 Windows 风格的 BGR 颜色模型，具有用 3 字节存储的 Blue、Green 和 Red 三种颜色。
-//         */
-//        BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_3BYTE_BGR);
-//        //此方法返回 Graphics2D，但此处是出于向后兼容性的考虑。
-//        bi.getGraphics().drawImage(f.image.getBufferedImage().getScaledInstance(width, height, Image.SCALE_SMOOTH),
-//                0, 0, null);
-//        ImageIO.write(bi, "jpg", targetFile);
-//        ff.flush();
-//        ff.stop();
-//        System.out.println(System.currentTimeMillis() - start);
-//    }
-//}
-//
+package cn.wisenergy.common.utils;
+
+
+import java.awt.Image;
+import java.awt.image.BufferedImage;
+import java.io.File;
+import javax.imageio.ImageIO;
+
+
+import org.bytedeco.javacv.FFmpegFrameGrabber;
+import org.bytedeco.javacv.Frame;
+import org.bytedeco.javacv.Java2DFrameConverter;
+
+
+/**
+ * 截取视频图片
+ *
+ * @ClassName: FrameGrabberKit
+ */
+public class FrameGrabberKit {
+
+    /**
+     * 获取视频图片
+     *
+     * @param videofile 源视频文件路径
+     * @param framefile 截取帧的图片存放路径
+     * @param imgfile   视频封面图保存路径
+     * @return
+     */
+    public static String getVedioImg(String videofile, String framefile, String imgfile) {
+        String ImgUrl = "";
+        //截取封面图
+        try {
+            fetchFrame(videofile, framefile);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        // 完整的ImgUrl
+        ImgUrl = imgfile;//视频封面图保存路径
+        return ImgUrl;
+    }
+
+    /**
+     * 获取指定视频的帧并保存为图片至指定目录
+     *
+     * @param videofile 源视频文件路径
+     * @param framefile 截取帧的图片存放路径 例：F:/hfkjrecorder/target/4.jpg
+     * @throws Exception
+     */
+    public static void fetchFrame(String videofile, String framefile) throws Exception {
+        FFmpegFrameGrabber ff = new FFmpegFrameGrabber(videofile);
+        ff.start();
+        int lenght = ff.getLengthInFrames();
+        int i = 0;
+        int interceptionFrames = 30;//截取第几帧
+        //默认截取第50帧，如果第50帧大于视频总帧数的8成直接取长度lenght * 0.3
+        if (interceptionFrames >= lenght * 0.8) {
+            interceptionFrames = (int) (lenght * 0.3);
+        }
+        Frame f = null;
+        while (i < lenght) {
+            // 过滤 前  interceptionFrames 帧，避免出现全黑的图片，依自己情况而定
+            f = ff.grabFrame();
+            if ((i > interceptionFrames) && (f.image != null)) {
+                break;
+            }
+            i++;
+        }
+
+        int owidth = f.imageWidth;
+        int oheight = f.imageHeight;
+        // 对截取的帧进行等比例缩放
+        int width = 800;
+        int height = (int) (((double) width / owidth) * oheight);
+        Java2DFrameConverter converter = new Java2DFrameConverter();
+        BufferedImage fecthedImage = converter.getBufferedImage(f);
+        BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_3BYTE_BGR);
+        bi.getGraphics().drawImage(fecthedImage.getScaledInstance(width, height, Image.SCALE_SMOOTH),
+                0, 0, null);
+        File targetFile = new File(framefile);
+        ImageIO.write(bi, "jpg", targetFile);
+        //ff.flush();
+        ff.stop();
+    }
+
+}

wisenergy-service/pom.xml

@@ -100,6 +100,16 @@
             <artifactId>xxl-job-core</artifactId>
             <version>1.9.2</version>
         </dependency>
+        <!--start:视频处理依赖-->
+        <dependency>
+            <groupId>org.bytedeco</groupId>
+            <artifactId>javacv</artifactId>
+        </dependency>
+        <!--end:视频处理依赖-->
+        <dependency>
+            <groupId>org.bytedeco.javacpp-presets</groupId>
+            <artifactId>ffmpeg-platform</artifactId>
+        </dependency>
     </dependencies>
 
     <!-- MAVEN构建 -->

wisenergy-service/src/main/java/cn/wisenergy/service/app/impl/UploadServiceImpl.java

@@ -51,25 +51,22 @@ public class UploadServiceImpl implements UploadService {
      * 视频文件上传
      */
     @Override
-        public Map<String, Object> uploadVideo(MultipartFile file,HttpServletRequest request,String userId,String zxField,String zxAddress) throws Exception {
+    public Map<String, Object> uploadVideo(MultipartFile file,HttpServletRequest request,String userId,String zxField, String zxAddress) throws Exception {
         Map<String, Object> resultMap=new HashMap<String, Object>();
 
         String basePath = request.getScheme() + "://" + request.getServerName()
                 + ":" + request.getServerPort()+"/mimi/upload/video/";
 
-            Long time = System.currentTimeMillis();
-//.getOriginalFilename()
-            String zxName = file.getName();//文件原始名称
-            String suffixName = "." + FilenameUtils.getExtension(zxName);
-            String newzxName = time+suffixName; //文件新名称
-            //设置文件存储路径，可以存放在你想要指定的路径里面
-            String rootPath="/opt/upload/video/"; //上传视频存放位置
+        Long time = new Date().getTime();
 
-            String filePath = rootPath+newzxName;
+        String fileName = file.getOriginalFilename();//文件原始名称
+        String suffixName = fileName.substring(fileName.lastIndexOf("."));//从最后一个.开始截取。截取fileName的后缀名
+        String newFileName = time+suffixName; //文件新名称
+        //设置文件存储路径，可以存放在你想要指定的路径里面
+        String rootPath="D:/mimi/"+File.separator+"upload/video/"; //上传视频存放位置
 
+        String filePath = rootPath+newFileName;
         File newFile = new File(filePath);
-
-
         //判断目标文件所在目录是否存在
         if(!newFile.getParentFile().exists()){
             //如果目标文件所在的目录不存在，则创建父目录
@@ -79,24 +76,19 @@ public class UploadServiceImpl implements UploadService {
         //将内存中的数据写入磁盘
         file.transferTo(newFile);
         //视频上传保存url
-            String zxUrl = basePath + newzxName;
+        String videoUrl = basePath + newFileName;
 
         //视频封面图处理
         String newImgName = time+".jpg";
         String framefile = rootPath + newImgName;
         String imgUrlSave = basePath+newImgName;//图片最终位置路径
-
-
         //视频截取封面图
-          //  String imgUrl=FrameGrabberKit.getVedioImg(zxUrl, framefile, imgUrlSave);
+        String imgUrl=FrameGrabberKit.getVedioImg(videoUrl, framefile, imgUrlSave);
 
-
-            resultMap.put("zxUrl", zxUrl);
-            //resultMap.put("imgUrl", imgUrl);
-            resultMap.put("returnCode", 0);
-            System.out.println("上传的文件名为："+zxName+",后缀名为："+newzxName);
-            String imgUrl= " ";
-            shopZxMapper.zxadd(zxUrl,userId,zxName,zxField,imgUrl,zxAddress);
+        resultMap.put("videoUrl", videoUrl);
+        resultMap.put("imgUrl", imgUrl);
+        resultMap.put("returnCode", 200);
+        //System.out.println("上传的文件名为："+fileName+",后缀名为："+newFileName);
         return resultMap;
     }
 
@@ -133,7 +125,7 @@ public class UploadServiceImpl implements UploadService {
 
             resultMap.put("imgUrl", imgUrl);
             resultMap.put("returnCode", 0);
-            String zxUrl="";
+            String zxUrl=null;
             shopZxMapper.zxadd(zxUrl,userId,zxName,zxField,imgUrl,zxAddress);
             return resultMap;
         }