Java實戰之實現文件資料上傳並生成縮略圖

/**
 * @Author: Promsing(張有博)
 * @Date: 2021/9/18 - 0:25
 * @Description: 圖片的處理類
 * @version： 1.0
 */
@Slf4j
@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;
 
    }
}

/**
 * @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;
 
    }
 
}

/**
 * @Author: Promsing(張有博)
 * @Date: 2021/9/18 - 0:29
 * @Description: PDF的處理類
 * @version： 1.0
 */
@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;
 
    }
}

 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;
    }