Spring将InputStream转成MultipartFile

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

二、转换方法

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

1.1.导入依赖

<dependency>
   <groupId>org.springframework</groupId>
   <artifactId>spring-test</artifactId>
   <version>RELEASE</version>
</dependency>

1.2.代码实现

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

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

2.1.导入依赖

 <dependency>
     <groupId>commons-fileupload</groupId>
     <artifactId>commons-fileupload</artifactId>
     <version>1.3.1</version>
 </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;
    }
}