fastdfs上传文件资料(PDF,视频,图片,FileCaseUtil,FileUploadUtil)并生成缩略图

javacv里面有类型转换器的使用,对视频图片的处理工具等等

一、UML图

在这里插入图片描述

二、代码

pom依赖


<!-- fastdfs-->
        <dependency>
            <groupId>com.github.tobato</groupId>
            <artifactId>fastdfs-client</artifactId>
            <version>1.27.2</version>
        </dependency>

<!--JavaCV的依赖-->
        <!-- 视频操作类-->
        <dependency>
            <groupId>org.bytedeco</groupId>
            <artifactId>javacv</artifactId>
            <version>1.4.3</version>
        </dependency>
        <dependency>
            <groupId>org.bytedeco.javacpp-presets</groupId>
            <artifactId>ffmpeg-platform</artifactId>
            <version>4.0.2-1.4.3</version>
        </dependency>
        <!--项目在本地window用FFmpeg-windows-x86_64,项目在Linux用FFmpeg-linux-x86_64-->
        <dependency>
            <groupId>com.github.hoary.ffmpeg</groupId>
            <artifactId>FFmpeg-windows-x86_64</artifactId>
            <version>2.0.1.1</version>
        </dependency>
        <dependency>
            <groupId>com.github.hoary.ffmpeg</groupId>
            <artifactId>FFmpeg-linux-x86_64</artifactId>
            <version>2.0.1.1</version>
        </dependency>

        <!-- 网络资源转File-->
        <dependency>
            <groupId>commons-io</groupId>
            <artifactId>commons-io</artifactId>
            <version>2.11.0</version>
        </dependency>
        


		 <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>

        <dependency>
            <groupId>org.apache.httpcomponents</groupId>
            <artifactId>httpcore</artifactId>
            <version>4.4.15</version>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
        </dependency>

<!--pdf转图片-->
        <dependency>
            <groupId>org.apache.pdfbox</groupId>
            <artifactId>fontbox</artifactId>
            <version>2.0.9</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.apache.pdfbox/pdfbox -->
        <dependency>
            <groupId>org.apache.pdfbox</groupId>
            <artifactId>pdfbox</artifactId>
            <version>2.0.9</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/commons-logging/commons-logging -->
        <dependency>
            <groupId>commons-logging</groupId>
            <artifactId>commons-logging</artifactId>
            <version>1.2</version>
        </dependency>

工具类

FileCaseUtil

package com.zykj.healthcode.utils;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.apache.http.entity.ContentType;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.Java2DFrameConverter;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * 文件类型转换
 * 
 * @author lc
 * @version 1.0
 * @date 2022/3/29 16:36
 */
@Component
@Slf4j
public class FileCaseUtil {
 	public static final ThreadLocal<String> threadLocal = new ThreadLocal<>();
 
    /**
     * 读取网络中的图片- 网络资源转File
     * 需要引入依赖commons-io
     *
     * @param url https://www.kziyue.com/wp-content/uploads/2019/06/5bca-hxyuaph9825616.jpg
     * @return
     */
    public File URLToFile(String url){
        File file1 = new File("test.png");
        try {

            URL url1 = new URL(url);
            FileUtils.copyURLToFile(url1,file1);

        } catch (IOException e) {
            e.printStackTrace();
        }
        File absoluteFile = file1.getAbsoluteFile();
        return file1;
    }

    /**
     * 文件上传 - 网络资源转MultipartFile
     * 需要引入依赖spring-web
     *
     * @param urlStr url地址
     * @return multipartFile
     */
    public MultipartFile fileUpload(String urlStr) throws Exception {
        try {
            //把地址转换成URL对象
            URL url = new URL(urlStr);
            //创建http链接
            HttpURLConnection conn = (HttpURLConnection)url.openConnection();
            //设置超时间为3秒
            conn.setConnectTimeout(3*1000);
            //防止屏蔽程序抓取而返回403错误
            conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36)");
            //得到输入流
            InputStream inputStream = conn.getInputStream();
            //截取链接中的文件名
            String fileName= urlStr.substring(urlStr.lastIndexOf("/")+1);
            MultipartFile multipartFile = new MockMultipartFile(fileName,fileName, ContentType.APPLICATION_OCTET_STREAM.toString(), inputStream);

            return multipartFile;
        } catch (Exception e) {
            e.printStackTrace();
        }
        throw new Exception();

    }

    /**
     * 文件类型转换 - File转MultipartFile
     *
     * @param filePath 文件file
     * @return MultipartFile
     */
    public static MultipartFile caseFileToMultipartFile(File filePath) {
        MultipartFile multipartFile = null;
        try {
            log.info("开始进行文件转换");
            FileInputStream fileInputStream = new FileInputStream(filePath);
            multipartFile = new MockMultipartFile(filePath.getName(), filePath.getName(),
                    ContentType.APPLICATION_OCTET_STREAM.toString(), fileInputStream);
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
        return multipartFile;
    }

    /**
     * 将文件转为字节数组 - File转字节数组
     *
     * @param file
     * @param size 1024
     * @return
     */
    public static byte[] BufferStreamForByte(File file, int size) {
        byte[] content = null;
        try {
            BufferedInputStream bis = null;
            ByteArrayOutputStream out = null;
            try {
                FileInputStream input = new FileInputStream(file);
                bis = new BufferedInputStream(input, size);
                byte[] bytes = new byte[1024];
                int len;
                out = new ByteArrayOutputStream();
                while ((len = bis.read(bytes)) > 0) {
                    out.write(bytes, 0, len);
                }

                bis.close();
                content = out.toByteArray();
            } finally {
                if (bis != null) {
                    bis.close();
                }
                if (out != null) {
                    out.close();
                }
            }
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return content;

    }


    /**
     * Frame转BufferedImage
     * 需要引入依赖javacv
     *
     * @param frame
     * @return
     */
    public static BufferedImage FrameToBufferedImage(Frame frame) {
        //创建BufferedImage对象
        Java2DFrameConverter converter = new Java2DFrameConverter();
        BufferedImage bufferedImage = converter.getBufferedImage(frame);
        return bufferedImage;
    }

    /**
     * BufferedImage转MultipartFile
     *
     * @param image
     * @return
     */
    public static MultipartFile fileCase(BufferedImage image){
        //得到BufferedImage对象
        // BufferedImage bufferedImage = JoinTwoImage.testEncode(200, 200, url);
        MultipartFile multipartFile= null;
        try {
            //创建一个ByteArrayOutputStream
            ByteArrayOutputStream os = new ByteArrayOutputStream();
            //把BufferedImage写入ByteArrayOutputStream
            ImageIO.write(image, "jpg", os);
            //ByteArrayOutputStream转成InputStream
            InputStream input = new ByteArrayInputStream(os.toByteArray());
            //InputStream转成MultipartFile
            multipartFile =new MockMultipartFile("file", "file.jpg", "text/plain", input);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return multipartFile;

    }



}


FileUploadUtil

package com.zykj.healthcode.utils;

import com.github.tobato.fastdfs.domain.fdfs.StorePath;
import com.github.tobato.fastdfs.service.FastFileStorageClient;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.Java2DFrameConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.URL;

/**
 * @author lc
 * @version 1.0
 * @date 2022/3/29 16:05
 */
@Slf4j
@Component
public class FileUploadUtil {
    private static final ThreadLocal<String> threadLocal = new ThreadLocal<>();

    private static String group="http://192.168.y.y:8888/group1/";

    @Autowired
    private static FastFileStorageClient fastFileStorageClient;

    public static File URLToFile(String url){
        log.info("读取FastDFS上的ppt");
        File file1 = new File("test.pdf");
        try {

            URL url1 = new URL(url);
            FileUtils.copyURLToFile(url1,file1);

        } catch (IOException e) {
            e.printStackTrace();
        }
        File absoluteFile = file1.getAbsoluteFile();
        threadLocal.set(absoluteFile.toString());
        log.info("ppt已经存储到本地"+absoluteFile.toString());
        return file1;
    }

    public static BufferedImage FrameToBufferedImage(Frame frame) {
        //创建BufferedImage对象
        Java2DFrameConverter converter = new Java2DFrameConverter();
        BufferedImage bufferedImage = converter.getBufferedImage(frame);
        return bufferedImage;
    }

    public static MultipartFile fileCase(BufferedImage image){
        //得到BufferedImage对象
        // BufferedImage bufferedImage = JoinTwoImage.testEncode(200, 200, url);
        MultipartFile multipartFile= null;
        try {
            //创建一个ByteArrayOutputStream
            ByteArrayOutputStream os = new ByteArrayOutputStream();
            //把BufferedImage写入ByteArrayOutputStream
            ImageIO.write(image, "jpg", os);
            //ByteArrayOutputStream转成InputStream
            InputStream input = new ByteArrayInputStream(os.toByteArray());
            //InputStream转成MultipartFile
            multipartFile =new MockMultipartFile("file", "file.jpg", "text/plain", input);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return multipartFile;

    }

    public static String fileLoad(MultipartFile mf) throws IOException {
        //上传视频并返回视频地址
        StorePath storePath = fastFileStorageClient.uploadFile(group,mf.getInputStream(), mf.getSize(), mf.getOriginalFilename().substring(mf.getOriginalFilename().lastIndexOf(".")+1));
        //将视频地址和项目id存到项目资料表中
        String videoUrl = "http://d-godone.yyy.tech/goDone/" + storePath.getPath();
        return videoUrl;
    }
}

抽象类

package com.zykj.healthcode.controller;

/**
 * @author lc
 * @version 1.0
 * @date 2022/3/29 16:19
 */
public abstract class TypeProcessing {
    public abstract  String fileUpload(String videoFileName);
}

2.1 上传图片生成缩略图 – FastDFS

package com.zykj.healthcode.controller;

import com.github.tobato.fastdfs.domain.fdfs.StorePath;
import com.github.tobato.fastdfs.domain.fdfs.ThumbImageConfig;
import com.github.tobato.fastdfs.service.FastFileStorageClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;

/**
 * 图片缩略图
 *
 * @author lc
 * @version 1.0
 * @date 2022/3/29 15:31
 */
@Component
public class ImageProcessing extends TypeProcessing{

    @Autowired
    private FastFileStorageClient fastFileStorageClient;

    @Autowired
    private ThumbImageConfig thumbImageConfig;

    private  String group="http://192.168.y.y:8888/group1/";

    /**
     *
     * @param fileName 本地文件的位置
     * @return
     */
    @Override
    public String fileUpload(String fileName) {

        File file = new File(fileName);


        // 上传并且生成缩略图
        StorePath storePath = null;
        try {
            storePath = this.fastFileStorageClient.uploadImageAndCrtThumbImage(
                    new FileInputStream(file), file.length(), "png", null);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        //String path = thumbImageConfig.getThumbImagePath(storePath.getPath());

        String imageUrl=group+thumbImageConfig.getThumbImagePath(storePath.getPath());

        return imageUrl;

    }



}

2.2 上传视频生成缩略图 – javacv

package com.zykj.healthcode.controller;

import com.zykj.healthcode.utils.FileCaseUtil;
import com.zykj.healthcode.utils.FileUploadUtil;
import lombok.extern.slf4j.Slf4j;
import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.bytedeco.javacv.Frame;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;


import java.awt.image.BufferedImage;

/**
 * @Author: Promsing(张有博)
 * @Date: 2021/9/18 - 0:37
 * @Description: 视频的处理类
 * @version: 1.0
 */
@Component
@Slf4j
public class VideoProcessing extends TypeProcessing {

    @Override
    public String fileUpload(String videoFileName) {
        //最后获取到的视频的图片的路径
        String videPicture="";
        //Frame对象
        Frame frame = null;
        //标识
        int flag = 0;
        try {
             /*
            获取视频文件
            */
            // FFmpegFrameGrabber fFmpegFrameGrabber = new FFmpegFrameGrabber(videoPath + "/" + videoFileName);
            FFmpegFrameGrabber fFmpegFrameGrabber = new FFmpegFrameGrabber( videoFileName);
            fFmpegFrameGrabber.start();

            //获取视频总帧数
            int ftp = fFmpegFrameGrabber.getLengthInFrames();
            log.info("时长 " + ftp / fFmpegFrameGrabber.getFrameRate() / 60);


            while (flag <= ftp) {
                frame = fFmpegFrameGrabber.grabImage();

                /*
                对视频的第五帧进行处理
                 */
                if (frame != null && flag==5) {

//                    //文件转换
                    BufferedImage bufferedImage =FileCaseUtil.FrameToBufferedImage(frame);

                    MultipartFile multipartFile = FileCaseUtil. fileCase(bufferedImage);
                    log.info("开始文件上传:");
                    //文件上传
                    String fileLoad = FileUploadUtil.fileLoad(multipartFile);

                    videPicture=fileLoad;
                    log.info("文件上传成功{}",fileLoad);
                    break;

                }
                flag++;
            }
            fFmpegFrameGrabber.stop();
            fFmpegFrameGrabber.close();
        } catch (Exception E) {
            E.printStackTrace();
        }
        return videPicture;

    }

}

2.3 上传PDF生成缩略图 – pdfbox

package com.zykj.healthcode.controller;

import org.apache.pdfbox.rendering.ImageType;
import com.zykj.healthcode.utils.FileCaseUtil;
import com.zykj.healthcode.utils.FileUploadUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.encryption.InvalidPasswordException;
import org.apache.pdfbox.rendering.PDFRenderer;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

/**
 * @author lc
 * @version 1.0
 * @date 2022/3/29 16:47
 */
@Slf4j
@Component
public class PDFProcessing extends TypeProcessing{

    @Override
    public String fileUpload(String fileName) {

        String pdfUrl="";
        try
        {
            //将读取URL生成File
            File file = FileCaseUtil.URLToFile(fileName);


            // 打开来源 使用pdfbox中的方法
            PDDocument pdfDocument = PDDocument.load(file);
            PDFRenderer pdfRenderer = new PDFRenderer(pdfDocument);

            // 提取的页码
            int pageNumber = 0;
            // 以300 dpi 读取存入 BufferedImage 对象
            int dpi = 300;
            BufferedImage buffImage = pdfRenderer.renderImageWithDPI(pageNumber, dpi, ImageType.RGB);
            // 将 BufferedImage 写入到 png
            // ImageIOUtil.writeImage(buffImage, "c:/temp/xx.png", dpi);

            //这里使用文件BufferedImage上传

            // 文件上传
            MultipartFile multipartFile =FileCaseUtil. fileCase(buffImage);
            log.info("PDF开始上传:");
            pdfUrl= FileUploadUtil. fileLoad(multipartFile);
            log.info("PDF上传成功:{}",pdfUrl);

//            //文件储存对象
//            File outPut = new File("C:\\Users\\Administrator\\Videos\\"+ UUID.randomUUID().toString()+".jpg");
//            // ImageIO.write(FrameToBufferedImage(frame), "jpg", outPut);
//            ImageIO.write(buffImage, "jpg",outPut);

            // 关闭文档
            pdfDocument.close();

            //注意关闭文档与删除文档的顺序
            //删除临时的file
            String s = FileCaseUtil.threadLocal.get();
            System.out.println(s);

            File f=new File(s);
            boolean delete = f.delete();
            System.out.println("文件的删除状态"+delete);
        }
        catch (InvalidPasswordException e)
        {
            e.printStackTrace();
        }
        catch (IOException e)
        {
            e.printStackTrace();
        }

        return pdfUrl;

    }
}

三、编码过程中遇到的问题

1.文件截取比较慢,如何解决?

方法:使用多线程

2.通过new File()无法获取网络文件?

File file = new File(“http://d-godone.dmsd.tech/goDone/M00/00/0A/wKg8O2D2mnqEMg7wAAAAALbl5Ys275.pdf”);

方法:使用工具类FileUtils.copyURLToFile(url,file)

3.文件上传需要MultipartFile类型,但是截取出来的类型为Frame和BufferedImage,怎么弄?

方法:查阅资料封装出,文件转换的方法。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

LC超人在良家

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值