webflux 下载excel

重生之我是一名程序员

于 2024-05-24 09:14:00 发布

阅读量34

点赞数

分类专栏： webflux 文章标签： webflux

原文链接：https://editor.csdn.net/md?not_checkout=1&spm=1011.2415.3001.6217&articleId=139162653

版权

webflux 专栏收录该内容

11 篇文章 1 订阅

订阅专栏

webflux 下载excel

方案1

public class DownloadUtil {

    /**
     * 下载 excel
     * @param data 数据列表
     * @param clz 类型
     * @param filename 文件名称
     * @param response
     * @return
     */
    public static <T> Mono<Void> downloadExcel(List<T> data, Class<T> clz,String filename,ServerHttpResponse response) {
        HttpHeaders headers = response.getHeaders();
        headers.set(HttpHeaders.CONTENT_DISPOSITION,"attachment;filename=".concat(filename).concat(""));
        //vnd.ms-excel 可以替换为 octet-stream
        MediaType application = new MediaType("application", "vnd.ms-excel", Charset.forName("UTF-8"));
        headers.setContentType(application);
        DefaultDataBuffer dataBuffer = new DefaultDataBufferFactory().allocateBuffer();
        OutputStream outputStream = dataBuffer.asOutputStream();
        EasyExcel.write(outputStream, clz).sheet("sheet1").doWrite(data);
        Flux<DataBuffer> dataBufferFlux = Flux.create((FluxSink<DataBuffer> emitter) -> {
            emitter.next(dataBuffer);
            emitter.complete();
        });
        return response.writeWith(dataBufferFlux);
    }
}


@GetMapping(value = "/download")
public Mono<Void> downloadDept(ServerHttpResponse response) {
    List<ExportDto> result = departmentService.downloadDept(merchantCode);
    return DownloadUtil.downloadExcel(result,ExportDto.class,"download.xlsx", response);
}

方案2

@GetMapping(value = "/downloadFile")
    public Mono<Void> downloadTest(ServerHttpResponse response) {
        File file = new File("test.xlsx");
        //这里为了模拟流 而去读取本地文件
        FileInputStream in =null;
        try {
            in = new FileInputStream(file);
            Flux<DataBuffer> dataBufferFlux = DataBufferUtils.readByteChannel(in::getChannel,new DefaultDataBufferFactory(),4096);
            ZeroCopyHttpOutputMessage zeroCopyHttpOutputMessage = (ZeroCopyHttpOutputMessage)response;
            HttpHeaders headers = zeroCopyHttpOutputMessage.getHeaders();
            headers.set(HttpHeaders.CONTENT_DISPOSITION,"attachment;filename=test.xlsx");
            //vnd.ms-excel 可以替换为 octet-stream
            MediaType application = new MediaType("application", "vnd.ms-excel", Charset.forName("UTF-8"));
            headers.setContentType(application);
            return zeroCopyHttpOutputMessage.writeWith(dataBufferFlux);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return  Mono.empty();
    }