MongoDb下载文件转为MultipartFile格式

最新推荐文章于 2024-09-26 16:46:11 发布

black____king

最新推荐文章于 2024-09-26 16:46:11 发布

阅读量87

点赞数

文章标签： mongodb 数据库 java

本文链接：https://blog.csdn.net/black____king/article/details/131052856

版权

该代码示例展示了如何从MongoDB的GridFS存储中下载文件，并将其转换为SpringMVC的MultipartFile对象。通过GridFSBucket打开下载流，读取文件内容到字节数组，然后用这些数据创建MultipartFile实例。

摘要由CSDN通过智能技术生成

公司做了一个从mongodb下载文件存储服务器上,需要将mongodb下载的文件格式转为MultipartFile .记录一下

直接放代码

import org.springframework.web.multipart.MultipartFile;
        import com.mongodb.client.gridfs.GridFSBucket;
        import com.mongodb.client.gridfs.GridFSDownloadStream;

// 将 GridFS 文件转为 MultipartFile
public MultipartFile convertToMultipartFile(GridFSBucket gridFSBucket, String fileId) throws IOException {
        // 从 GridFS 中获取下载流
        GridFSDownloadStream downloadStream = gridFSBucket.openDownloadStream(fileId);

        // 读取流中的数据到字节数组
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] buffer = new byte[1024];
        int len;
        while ((len = downloadStream.read(buffer)) != -1) {
        baos.write(buffer, 0, len);
        }
        byte[] fileBytes = baos.toByteArray();

        // 创建 MultipartFile 对象
        MultipartFile multipartFile = new MockMultipartFile(downloadStream.getGridFSFile().getFilename(),
        downloadStream.getGridFSFile().getFilename(),
        downloadStream.getGridFSFile().getContentType(),
        fileBytes);
        return multipartFile;
        }