一、最近项目用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)方法即可
public class FileUtil {
public static MultipartFile toMultipartFile(String path) {
FileItem fileItem = createFileItem(path);
return new CommonsMultipartFile(fileItem);
}
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;
}
}