Spring将InputStream转成MultipartFile

一、最近项目用google的puppeteer对网页进行截图,截图之后需要将图片保存到阿里云OSS上,由于puppeteer截图后只能拿到图片的Inputstream对象,直接上传文件流到阿里云OSS一直走不通(官方有提供API,不知道为什么走不通),无奈之下只能将InputStream转成MultipartFile,项目上传文件一直都是使用MultipartFile的,可以确定MultipartFile是可以上传到阿里云OSS,果然如此,下面记录一下如何将InputStream转成MultipartFile

二、转换方法

1.使用MockMultipartFile类进行转换(注意:该类只能用于测试环境,项目打包是不会打包对应的依赖包,部署项目将会报错)

1.1.导入依赖

<dependency>
   <groupId>org.springframeworkgroupId>
   <artifactId>spring-testartifactId>
   <version>RELEASEversion>
dependency>

1.2.代码实现

MultipartFile multipartFile = new MockMultipartFile("example.jpg","example.jpg","", inputStream);

2.将文件保存到本地,然后将本地文件转化成FileItem,再将FileItem转成MultipartFile

2.1.导入依赖

 <dependency>
     <groupId>commons-fileuploadgroupId>
     <artifactId>commons-fileuploadartifactId>
     <version>1.3.1version>
 dependency>

2.2.代码实现:直接调用toMultipartFile(path)方法即可

/**
 * @author zx
 * @date 2020/9/7
 */
public class FileUtil {

    /**
     * 将本地文件转成MultipartFile
     * @param path
     * @return
     */
    public static MultipartFile toMultipartFile(String path) {
        //先将本地文件转化成FileItem
        FileItem fileItem = createFileItem(path);
        return new CommonsMultipartFile(fileItem);
    }

    /**
     * 根据filePath将puppeteer截取到的图片转成FileItem
     * @param filePath 文件路径
     * @return
     */
    public static FileItem createFileItem(String filePath) {
        FileItemFactory factory = new DiskFileItemFactory(16, null);
        String fieldName = "example"; //文件名
        FileItem item = factory.createItem(fieldName, "image/jpg", true, UUIDUtils.genUUid() + ".jpeg");
        File newfile = new File(filePath);
        int bytesRead = 0;
        byte[] buffer = new byte[8192];
        try {
            FileInputStream fis = new FileInputStream(newfile);
            OutputStream os = item.getOutputStream();
            while ((bytesRead = fis.read(buffer, 0, 8192)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
            os.close();
            fis.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return item;
    }
}

你可能感兴趣的:(小记)