Quarkus中Resteasy文件上传下载

时间:2022-05-23 kl 人气:0

前言

做java web开发，文件的上传和下载是一个非常常见的需求场景。在spring生态下的文件上传下载非常简单，记得五年前，我刚写博的时候也写过spring的上传和下载《SpringMvc实现文件上传与下载》，以及Struts2的文件上传和下载《java Struts2实现文件上传下载》。今天来看看Quarkus框架中是如何实现的，因为Quarkus的JAX-RS采用Resteasy的实现，故Quarkus中的文件上传和下载，也就是Resteasy框架的上传和下载了。

Resteasy文档：https://docs.jboss.org/resteasy/docs/4.3.1.Final/

文件上传

首先引入依赖

<dependency>
    <groupId>org.jboss.resteasy</groupId>
    <artifactId>resteasy-multipart-provider</artifactId>
</dependency>

这个依赖主要用于支持multipart/form-data的请求，接下来需要定义一个接收multipart/form-data表单数据的实体类，如下：

public class UploadEntity{
        @FormParam("file")
        private InputPart file;
        @FormParam("fileName")
        private String fileName;
        public InputPart getFile() {
            return file;
        }
        public void setFile(InputPart file) {
            this.file = file;
        }
        public String getFileName() {
            return fileName;
        }
        public void setFileName(String fileName) {
            this.fileName = fileName;
        }
    }

以上代码用来映射表单中的属性值，除了文件流，只支持字符串类型的属性。其中file字段用来接收上传的文件，除了可以用InputPart类型来接收，上传文件还可以使用InputStream、byte[]、File等接收，这里使用InputPart来接收是为了方便接下来从Header参数里获取文件名，如：

@POST
    @Path("/upload")
    @Consumes(MediaType.MULTIPART_FORM_DATA)
    public Response upload(@MultipartForm UploadEntity uploadEntity) throws IOException {
        InputStream i = uploadEntity.getFile().getBody(InputStream.class, null);
        String[] contentDisposition = uploadEntity.getFile().getHeaders().getFirst("Content-Disposition").split(";");
        String fileName = "";
        for (String filename : contentDisposition) {
            if ((filename.trim().startsWith("filename"))) {
                String[] name = filename.split("=");
                fileName = name[1].trim().replaceAll("\"", "");
            }
        }
        logger.info("上传文件名:{}", fileName);
        logger.info("表单提交的文件名:{}", uploadEntity.getFileName());
        logger.info("上传文件内容:{}", IOUtils.toString(i, StandardCharsets.UTF_8));
        return Response.ok("ok").build();
    }

InputPart实体承载了文件流，以及请求体信息，可以通过getBody来获取文件流，Header中的Content-Disposition参数获取真实的文件名称。除了这种方式获取真实文件名称，如代码所示，还可以通过表单参数，接收前端表单传过来的文件名称。

文件下载

@GET
    @Path("/download")
    @Produces(MediaType.APPLICATION_OCTET_STREAM)
    public Response download(@QueryParam("fileName") String fileName) {
        String filePath = "E:\\工作相关\\" + fileName;
        File file = FileUtils.getFile(filePath);
        //对输出的文件名进行编码，防止下载的中文文件名乱码
        String encodFileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8);
        return Response.ok(file)
                .header("content-disposition", "attachment; filename=\"" + encodFileName + "\"")
                .header("Content-Length", file.length())
                .build();
    }

相比于文件上传，文件下载的代码就更简单了，不过要注意，如果下载的文件名是中文的，最好将文件名使用URLEncoder的UTF-8编码下，防止中文名称乱码。

加载全部内容