spring webflux 系列之 上传文件到七牛云篇(不用本地创建文件直接内存上传)
spring webflux上传单文件与多文件 至 七牛云
下载文件篇
本次我介绍springboot 下 webflux 上传文件到七牛云;
网上基本上全部教程都是创建本地文件 才去上传七牛云 又要删除本地文件;
所以我直接放在内存里面 上传
代码如下:
//写法一:单文件 Mono<FilePart>
@PostMapping("/upload")
public Mono<Response> uploadFile(@RequestPart(value = "file", required = false) FilePart file) throws IOException, DataException {
if (file == null) {
throw new CustomException(HttpCodeEnum.filenoempty);
}
//获取文件名并检验后缀
String fileName = file.filename();
String type = fileName.indexOf(".") != -1 ? fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length()) : null;
if (!"GIF".equals(type.toUpperCase()) &&
!"PNG".equals(type.toUpperCase()) &&
!"JPG".equals(type.toUpperCase()) &&
!"JPEG".equals(type.toUpperCase()) &&
!"MP4".equals(type.toUpperCase()) &&
!"MP3".equals(type.toUpperCase())) {
throw new CustomException(HttpCodeEnum.filesuffixnosupport);
}
Flux<DataBuffer> bufferFlux = file.content();
// Mono<InputStream> mono = bufferFlux.map(dataBuffer->dataBuffer.asInputStream()).reduce(SequenceInputStream::new);
Mono<Response> mono = bufferFlux.map(dataBuffer->dataBuffer.asInputStream()).reduce(SequenceInputStream::new).flatMap(
inputStream -> {
String fileNames = "jiumi" + "/" + JodaTimeUtils.format(new Date(), "yyyyMMddHHmmss") + RandomUtils.get3Random() + "." + type;
//七牛云上传并获取返回链接
String url = null;
try {
url = QiNiuUtils.uploadFile((ImageUtils.readInputStream(inputStream)), fileNames, "pingan");
} catch (QiniuException e) {
e.printStackTrace();
}
//新增删除临时文件 通常位于tmp目录下
file.delete().subscribe();
return Mono.justOrEmpty(new Response().add("url", url).success());
}
);
return mono;
}
/**
* 写法二:多文件 Flux<FilePart>
* spring webflux 实现可能有点麻烦 毕竟最大文件max 是9MB 推荐使用上面的单文件上传
*/
@PostMapping("/uploads")
public Mono<Response> uploadFiles(@RequestPart(value = "files", required = false) Flux<FilePart> files) throws IOException, DataException {
files.map(filePart -> {
String fileName = filePart.filename();
String type = fileName.indexOf(".") != -1 ? fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length()) : null;
if (!"GIF".equals(type.toUpperCase()) &&
!"PNG".equals(type.toUpperCase()) &&
!"JPG".equals(type.toUpperCase()) &&
!"JPEG".equals(type.toUpperCase()) &&
!"MP4".equals(type.toUpperCase()) &
!"MP3".equals(type.toUpperCase())) {
throw new CustomException(HttpCodeEnum.filesuffixnosupport);
}
return filePart;
});
Mono<Response> mono=files.flatMap(filePart -> {
String fileName = filePart.filename();
String type = fileName.indexOf(".") != -1 ? fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length()) : null;
return filePart.content().map(dataBuffer->dataBuffer.asInputStream()).reduce(SequenceInputStream::new).flatMap(
inputStream -> {
String fileNames = "jiumi" + "/" + JodaTimeUtils.format(new Date(), "yyyyMMddHHmmss") + RandomUtils.get3Random() + "." + type;
//七牛云上传并获取返回链接
String url = null;
try {
url = QiNiuUtils.uploadFile((ImageUtils.readInputStream(inputStream)), fileNames, "pingan");
} catch (QiniuException e) {
e.printStackTrace();
}
//新增删除临时文件 通常位于tmp目录下
filePart.delete().subscribe();
return Mono.just(url);
});
}).collect(Collectors.toList()).flatMap(
strings -> {
return Mono.just(new Response().add("list",strings).success());
}
);
return mono;
}
Response是我自定义返回体 整个项目是返回json的
这里可以改成你喜欢的返回方式
可能还会涉及到spring webflux 上传文件大小的限制
我的配置如下:
@Override
public void configureHttpMessageCodecs(ServerCodecConfigurer configurer) {
ObjectMapper mapper = mapper();
configurer.defaultCodecs().jackson2JsonEncoder(new Jackson2JsonEncoder(mapper));
configurer.defaultCodecs().jackson2JsonDecoder(new Jackson2JsonDecoder(mapper));
//硬盘的
DefaultPartHttpMessageReader partReader = new DefaultPartHttpMessageReader();
partReader.setMaxParts(1);//这里最大文件数量
partReader.setMaxDiskUsagePerPart(9L*1000L*1024L);//这里就是最大文件大小 但是除了文件流 他也包括了请求数据在里面 9MB
// partReader.setMaxHeadersSize(921600);
partReader.setEnableLoggingRequestDetails(true);
MultipartHttpMessageReader multipartReader = new MultipartHttpMessageReader(partReader);
multipartReader.setEnableLoggingRequestDetails(true);
configurer.defaultCodecs().multipartReader(multipartReader);
}
后续我会继续更新spring webflux 系列
并整合成项目开源 github开源地址