Spring将InputStream转成MultipartFile

╰+攻爆jí腚メ 2022-12-05 09:53 1461阅读 0赞

# 一、最近项目用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;
        }
    }