参数处理

yzg-util-file/src/main/java/com/yanzuoguang/cloud/file/YzgFileServiceImpl.java

 package com.yanzuoguang.cloud.file;
 
 import com.yanzuoguang.cloud.file.vo.*;
+import com.yanzuoguang.util.MediaHelper;
+import com.yanzuoguang.util.MediaParameter;
 import com.yanzuoguang.util.cache.MemoryCache;
 import com.yanzuoguang.util.exception.CodeException;
 import com.yanzuoguang.util.helper.DateHelper;
@@ -189,27 +191,24 @@ public class YzgFileServiceImpl implements YzgFileService {
         fileConfig.init(info, fromUrl);
 
         String serverFullPath = fileConfig.getServerFullPath(fromUrl);
+        try {
             switch (info.getType()) {
                 case FileHelper.FILE_TYPE_AUDIO: {
-                FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(serverFullPath);
-                info.setBitrate(Double.valueOf(grabber.getVideoBitrate()));
+                    MediaParameter parameter = MediaHelper.getGrabberParameter(serverFullPath);
+                    info.setBitrate(Double.valueOf(parameter.getAudioBitRate()));
                     break;
                 }
                 case FileHelper.FILE_TYPE_VIDEO: {
-                FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(serverFullPath);
-                info.setBitrate(Double.valueOf(grabber.getVideoBitrate()));
-                info.setWidth(Double.valueOf(grabber.getImageWidth()));
-                info.setHeight(Double.valueOf(grabber.getImageHeight()));
+                    MediaParameter parameter = MediaHelper.getGrabberParameter(serverFullPath);
+                    info.setBitrate(Double.valueOf(parameter.getVideoBitRate()));
+                    info.setWidth(Double.valueOf(parameter.getVideoWidth()));
+                    info.setHeight(Double.valueOf(parameter.getVideoHeight()));
                     break;
                 }
                 case FileHelper.FILE_TYPE_IMAGE: {
-                try {
                     BufferedImage sourceImg = ImageIO.read(new FileInputStream(serverFullPath));
                     info.setWidth(Double.valueOf(sourceImg.getWidth()));
                     info.setHeight(Double.valueOf(sourceImg.getHeight()));
-                } catch (IOException e) {
-                    throw new RuntimeException(e);
-                }
                     break;
                 }
                 case FileHelper.FILE_TYPE_NONE:
@@ -217,6 +216,9 @@ public class YzgFileServiceImpl implements YzgFileService {
                     break;
                 }
             }
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
         return info;
     }
 

yzg-util-image/src/main/java/com/yanzuoguang/util/MediaHelper.java

@@ -206,6 +206,25 @@ public class MediaHelper extends ImageHelper {
         }
     }
 
+    /**
+     * 获取视频缩略图
+     *
+     * @param filePath 视频路径
+     * @throws IOException
+     */
+    public static MediaParameter getGrabberParameter(String filePath) throws IOException {
+        FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(filePath);
+        MediaParameter parameter = new MediaParameter();
+        try {
+            // 打开视频
+            grabber.start();
+            parameter.init(grabber);
+        } finally {
+            grabber.close();
+        }
+        return parameter;
+    }
+
     /**
      * 旋转图片
      *