PDF/Excel文件预览功能完整实现-java版本

新需求

最近接到一个新的需求，说是之前直接下载的PDF文件或者是Excel文件，现在不能直接下载，需要实现在线预览功能。

前端人员拿到这个需求后，去看了一下以前的代码，以前调用的下载接口和PDF文件预览接口都是直接将文件成二进制

流的形式，然后响应到前端。有的接口即使是动态生成PDF文件或者是Excel文件都是同样的操作，也是将动态生成的对

象的二进制流写入到响应对象中。前端人员遇到的问题是，如果直接将二进制流返回给前端，前端调用的时候，会直接

载而不能做其他处理。然后找我协商，看看后端人员有没有好的解决办法，我去百度里面查找，找到一个靠谱的方法，

大致意思就是将文件转为Base64格式的字符串数据，然后返回给前端，前端在去做响应的处理。

解决方案

确定方案后，然后我继续去百度里面查找如何将文件转为Base64格式的数据，找到一个可行的方法，网址如下:

https://blog.csdn.net/dbeautifullife/article/details/77650713

使用里面的方法写成了一个工具类，然后在根据业务的需要，添加了动态生成PDF文件和动态生成Excel

文件的方法，很好的解决了文件转换为Base64的方式。

import lombok.extern.slf4j.Slf4j;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.context.annotation.Scope;
import sun.misc.BASE64Encoder;

import java.io.*;

/**
 * @Description: 文件转 base64 工具类
 * @ClassName: ToBase64Utils
 * @author: dengchao
 * @date: 2020/11/24 9:27
 */

@Slf4j
@Scope(value = "singleton")//单例
public class ToBase64Utils {

    //私有化构造方法
    private ToBase64Utils(){}

    /* @Description: ava 自带的转换base64对象
     * @author: dengchao
     * @date: 2020-10-24 10:10
     */
    private final static BASE64Encoder encoder = new BASE64Encoder();

    /* @Description: 读取实体文件转 Base64
     * @author: dengchao
     * @date: 2020/11/24 9:29
     * @param: file 文件对象
     * @return: String 字符串
     */
    public static String PDFToBase64(File file) {
        if(file == null || !file.exists()){
            throw new BaseException(MessageConstant.A000001, "传入的file对象为null或者文件不存在");
        }

        if(file.isDirectory()){
            throw new BaseException(MessageConstant.A000001, "传入的file对象是一个路径，不是具体的文件");
        }

        ToBase64Utils.log.error("PDFToBase64-开始读取文件!");

        try ( FileInputStream fin = new FileInputStream(file);
              BufferedInputStream bin = new BufferedInputStream(fin);
              ByteArrayOutputStream baos = new ByteArrayOutputStream();
              BufferedOutputStream bout = new BufferedOutputStream(baos);
                ){
            byte[] buffer = new byte[1024];
            int len = bin.read(buffer);
            while(len != -1){
                bout.write(buffer, 0, len);
                len = bin.read(buffer);
            }
            //刷新此输出流并强制写出所有缓冲的输出字节
            bout.flush();
            return ToBase64Utils.byteArrayToBase64(baos);

        } catch (FileNotFoundException e) {
            ToBase64Utils.log.error("PDFToBase64-文件未找到!");
            e.printStackTrace();
        } catch (IOException e) {
            ToBase64Utils.log.error("PDFToBase64-读取文件异常!");
            e.printStackTrace();
        }

        return "";
    }

    /* @Description: 动态生成 Excel对象 转 Base64
     * @author: dengchao
     * @date: 2020/11/24 9:37
     * @param: No such property: code for class: Script1
     * @return:
     */
    public static String workToBase64(Workbook workbook){
        if(workbook == null){
            throw new BaseException(MessageConstant.A000001, "传入Excel文件对象错误");
        }

        String excelBase64 = "";
        try(ByteArrayOutputStream bos = new ByteArrayOutputStream()){
            workbook.write(bos);
            bos.flush();
            excelBase64 = ToBase64Utils.byteArrayToBase64(bos);
        }catch(Exception e){
            excelBase64 = "";
            ToBase64Utils.log.error("EXCEL文件转Base64错误!", e);
        }

        return excelBase64;
    }

    /* @Description: 字节输出流转 Base64
     * @author: dengchao
     * @date: 2020/11/24 9:48
     * @param: No such property: code for class: Script1
     * @return:
     */
    public static String byteArrayToBase64(ByteArrayOutputStream bos){
        if(bos == null){
            throw new BaseException(MessageConstant.A000001, "传入的字节输入流对象错误");
        }else{
            byte[] byteArray = bos.toByteArray();
            String resultString = ToBase64Utils.encoder.encodeBuffer(byteArray).trim();
            return resultString;
        }
    }
}
核心原理就是将二进制的输出流转换为Base64字符串即可。

实际操作

工具类写好之后，立马和前端人员进行调试，测试通过，能够正常返回Base64的字符串，并且前端人员拿到数据后，可以进一步做其他操作。

前端人员的代码能看懂一大半，使用的VUE组件。大致意思就是引用VUE组件生成一个excel文件对象，然后在动态的拼接数据，最后显示

在页面中。

const workbook = await XLSX.read(res.responseBody, { type: 'base64', cellDates: true })
workbook.SheetNames.map(async item => {
  if (item) {
    const title = item
    let html = await XLSX.utils.sheet_to_html(workbook.Sheets[item]).match(/<table>.*<\/table>/)[0]
    const totalTr = html.match(/<\/tr>/g).length
    const totalEmptyTd = html.match(/<td><\/td><\/tr>/g).length
    // 如果每行结尾都是空 td，则过滤掉
    if (totalTr === totalEmptyTd) {
      html = html.replace(/<td><\/td><\/tr>/g, '</tr>')
    }
    this.xlsxHtml.push({title, html})
  }
})

最终完成这个功能，PDF也是同样的处理方式。对于一般人来说，这完全能够满足需求，实现在线预览的功能，而且效果也很好。
可是对于专业人士来说，这种操作方式还是存在问题的，只要拿到文件的Base64的字符串数据，就可以拿到整个文件的所有数
据。所以对于安全要求比较严格的业务来说，这个是不达标的。不过对于我们来说，已经满足要求了。