Springboot如何实现视频上传及压缩功能

王林转载: 2023-05-10 17:16:13 1194浏览

一、定义视频上传请求接口

public AjaxResult videoUploadFile(MultipartFile file){
  try {
    if(null == file || file.isEmpty()){
      return AjaxResult.error("文件为空");
    }
    String ossFilePrefix = StringUtils.genUUID();
    String fileName = ossFilePrefix +"-"+ file.getOriginalFilename();
    String fileurl = AliOssUtils.videoUploadFile(file,fileName);
    AjaxResult ajax = AjaxResult.success();
    ajax.put("fileName", "after_"+fileName);
    ajax.put("url", fileurl);
    return ajax;
  } catch (Exception e) {
    return AjaxResult.error(e.getMessage());
  }
}

二、视频暂存至本地文件夹

public static final String uploadVideo(String baseDir, MultipartFile file, String fileName) throws FileSizeLimitExceededException, IOException {
    File desc = getAbsoluteFile(baseDir, fileName);
    file.transferTo(desc);
    String pathFileName = getPathFileName(baseDir, fileName);
    return pathFileName;
}

三、开始压缩视频

public static  boolean toCompressFile(String convertFile,String targetFile){
    try{
        /**将视频压缩为 每秒15帧 平均码率600k 画面的宽与高 为1280*720*/
        String cutCmd="ffmpeg -i " + convertFile + " -r 15 -b:v 600k  -s 1280x720 "+ targetFile;
        log.info("cutCmd: " + cutCmd);
        runCmd(cutCmd);
        log.info("文件："+convertFile+" 视屏压缩完成");
    }catch(Exception e){
        e.printStackTrace();
        log.info("压缩文件出现异常："+e.getMessage());
        return false;
    }
    return true;
}

四、上传至阿里云并获取压缩后的视频路径

private static String getFileUrl(String path) throws IOException {
    File file = new File(path);
    FileInputStream fileInputStream = new FileInputStream(file);
    MultipartFile multipartFile1 = new MockMultipartFile(file.getName(), file.getName(), ContentType.APPLICATION_OCTET_STREAM.toString(), fileInputStream);
    uploadFile(multipartFile1, file.getName());
    String url = getUrl(file.getName());
    return url;
}

五、核心调用

public static String videoUploadFile(MultipartFile multipartFile, String fileName) throws IOException {
    //存放路径
    String filePath = FileUploadUtils.uploadVideo(getDefaultBaseDir(), multipartFile, fileName);
    String convertFile = filePath.replace("/profile", getDefaultBaseDir()).replaceAll("//", "//m.sbmmt.com/m/");
    //字符串第一个字符最后出现的下标
    int lastIndex = convertFile.lastIndexOf("//m.sbmmt.com/m/");
    StringBuilder sb = new StringBuilder(convertFile);
    String convertFile1 = sb.insert(lastIndex + 1, "after_").toString();
    boolean flag = toCompressFile(convertFile, convertFile1);
    if (!flag) {
        throw new CustomException("文件压缩出现异常");
    }
    //读取压缩后的文件并上传至阿里云
    String url = getFileUrl(convertFile1);
    //删除本地暂存文件
    FileUtils.deleteFile(convertFile);
    log.info("文件：" + convertFile + " 删除成功");
    FileUtils.deleteFile(convertFile1);
    log.info("文件：" + convertFile1 + " 删除成功");
    return url;
}