二进制流转MultipartFile的MVC的上传文件对象

最新推荐文章于 2025-02-19 11:00:14 发布

原创最新推荐文章于 2025-02-19 11:00:14 发布 · 3k 阅读

1 ·

CC 4.0 BY-SA版权

java 同时被 2 个专栏收录

17 篇文章

订阅专栏

Maven

4 篇文章

订阅专栏

本文介绍了一个关于在Spring框架中使用MockMultipartFile进行文件上传测试的常见问题。在本地测试环境下运行正常，但在生产环境或跳过测试打包时会遇到类未找到的错误。文章提供了一个自定义的MockMultipartFile实现，用于替代测试包中的版本，确保代码在不同环境下都能正确运行。

网上一般都是使用测试的maven包中的MockMultipartFile这个类，就是下边这个maven包：

   <dependency>

    <groupId>org.springframework</groupId>

    <artifactId>spring-test</artifactId>

    <version>RELEASE</version>

</dependency>

但是，使用这个包，有一个大坑，就是你在本地测试的时候还好，但是一旦发布到生产环境，或者你跳过测试打包的话。就会报错说找不到这个类。所以提醒大家，测试包中的类不要写到生产代码中去。

那如何去解决这个问题呢。看下边这个例子。完美解决。

工具类：MockMultipartFile

package com.didichuxing.ep.bpm.common.utils;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;

import org.springframework.util.Assert;
import org.springframework.util.FileCopyUtils;
import org.springframework.web.multipart.MultipartFile;

/**
* 根据二进制流和数组
* 创建MultipartFile对象的工具类
*/
public class MockMultipartFile implements MultipartFile {
    private final String name;
    private String originalFilename;
    private String contentType;
    private final byte[] content;

    public MockMultipartFile(String name, byte[] content) {
        this(name, "", (String) null, (byte[]) content);
    }

    public MockMultipartFile(String name, InputStream contentStream) throws IOException {
        this(name, "", (String) null, (byte[]) FileCopyUtils.copyToByteArray(contentStream));
    }

    public MockMultipartFile(String name, String originalFilename, String contentType, byte[] content) {
        Assert.hasLength(name, "Name must not be null");
        this.name = name;
        this.originalFilename = originalFilename != null ? originalFilename : "";
        this.contentType = contentType;
        this.content = content != null ? content : new byte[0];
    }

    public MockMultipartFile(String name, String originalFilename, String contentType, InputStream contentStream) throws IOException {
        this(name, originalFilename, contentType, FileCopyUtils.copyToByteArray(contentStream));
    }

    public String getName() {
        return this.name;
    }

    public String getOriginalFilename() {
        return this.originalFilename;
    }

    public String getContentType() {
        return this.contentType;
    }

    public boolean isEmpty() {
        return this.content.length == 0;
    }

    public long getSize() {
        return (long) this.content.length;
    }

    public byte[] getBytes() throws IOException {
        return this.content;
    }

    public InputStream getInputStream() throws IOException {
        return new ByteArrayInputStream(this.content);
    }

    public void transferTo(File dest) throws IOException, IllegalStateException {
        FileCopyUtils.copy(this.content, dest);
    }
}

转换示例：

/**
* 更新上传的流程模型XMl的key
*
* @param mFile 导入的MultipartFile对象
* @param key   流程模型的key
* @return newXmlString新的xmlString
*/
public MultipartFile updateFile(MultipartFile mFile, String key) {
    if (StringUtils.isBlank(key) || Objects.isNull(mFile)) {
        return null;
    }
    SAXReader saxReader = new SAXReader();
    org.dom4j.Document doc = null;
    XMLWriter writer = null;
    MultipartFile file = null;
    try {
        doc = saxReader.read(mFile.getInputStream());
        //查询xml根节点下的process标签
        Element process = doc.getRootElement().element("process");
        //得到process标签下的属性id对象
        Attribute processIdAttr = process.attribute("id");
        //修改属性值
        processIdAttr.setValue(key);

        Element bpmnDiagram = doc.getRootElement().element("BPMNDiagram");
        Attribute bpmnDiagramIdAttr2 = bpmnDiagram.attribute("id");
        bpmnDiagramIdAttr2.setValue("BPMNDiagram_" + key);

        //查询xml中BPMNDiagram节点下的BPMNPlane标签
        Element bpmnPlane = bpmnDiagram.element("BPMNPlane");
        Attribute bpmPlaneBpmnElementAttr = bpmnPlane.attribute("bpmnElement");
        Attribute bpmPlaneIdAttr = bpmnPlane.attribute("id");
        bpmPlaneBpmnElementAttr.setValue(key);
        bpmPlaneIdAttr.setValue("BPMNPlane_" + key);

        //格式化xml输出格式
        OutputFormat format = OutputFormat.createCompactFormat();
        format.setEncoding("UTF-8");
        format.setIndent(true);//设置是否缩进
        format.setIndent(" ");//设置空格缩进
        format.setNewlines(true);//设置是否换行

        //将修改后的二进制数据流，通过ByteArrayOutputStream转成MultipartFile返回
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        writer = new XMLWriter(baos, format);
        writer.write(doc);
        ByteArrayInputStream in = new ByteArrayInputStream(baos.toByteArray());
        file = new MockMultipartFile(mFile.getName(), mFile.getOriginalFilename(), ContentType.APPLICATION_OCTET_STREAM.toString(), in);

    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        try {
            writer.flush();
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    return file;
}