Springboot集成MongoDB存储文件/读取文件

# 应用名称
spring.application.name=demo
#server.port=10086 不配置的话,默认8080

# springboot下mongoDB的配置参数。
spring.data.mongodb.host=localhost
spring.data.mongodb.port=27017
# 指定数据库库名
spring.data.mongodb.database=temp

package com.onepiece.mongo;

import org.bson.Document;
import org.bson.types.Binary;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;

import java.io.*;
import java.util.Base64;
import java.util.List;

/**
 * @author zaoyu
 * @description  用于演示Mongodb的文件存储（单个文件不大于16MB）
 */

@SpringBootTest
public class MongoSaveFiles {

    @Autowired
    private MongoTemplate mongoTemplate;

    // collection名
    private String IMAGE_COLLECTION = "image";
    // 源文件完整路径
    private String FILE_PATH = "D:\\temp\\onepiece.jpg";
    // 输出文件路径
    private String FILE_OUTPUT_PATH = "C:\\Users\\onepiece\\Desktop\\";
    // 限制16MB
    private Long FILE_SIZE_LIMIT = 16L * 1024L * 1024L;

    @Test
    public void saveFiles({
        byte[] fileContent = null;
        FileInputStream fis = null;
        try {
            File file = new File(FILE_PATH;
            long length = file.length(;
            // 校验文件大小，大于16MB返回。 这里的操作逻辑依据你自己业务需求调整即可。
            if (length >= FILE_SIZE_LIMIT {
                System.out.println("文件: " + file.getAbsolutePath( + "  超出单个文件16MB的限制。" ;
                return;
            }
            fileContent = new byte[(int file.length(];
            fis = new FileInputStream(file;
            // 读取整个文件
            fis.read(fileContent;
            // 把文件内容以二进制格式写入到mongodb
            Document document = new Document(;
            // fileName字段、content字段自定义。 
            document.append("fileName", file.getName(;
            document.append("content", new Binary(fileContent;
            Document insert = mongoTemplate.insert(document, IMAGE_COLLECTION;
            System.out.println("文件 " + file.getName( + " 已存入mongodb，对应ID是: " + insert.get("_id".toString(;
        } catch (IOException e {
            e.printStackTrace(;
        } finally {
            try {
                fis.close(;
            } catch (IOException e {
                e.printStackTrace(;
            }
        }
    }

    /**
     * 测试读取并写入到指定路径。
     */
    @Test
    public void readAndWriteFiles({
        // 这里也是，默认查所有，需要条件自行增加。 简单取1条验证。 
        List<Document> result = mongoTemplate.find(new Query(, Document.class, IMAGE_COLLECTION;
        Document document = result.get(0;
        // 取出存储的二进制数据，这里用binary.class处理。
        Binary content = document.get("content", Binary.class;
        String fileName = document.get("fileName", String.class;
        try {
            String newFilePath = FILE_OUTPUT_PATH + fileName;
            // 写入到指定路径
            FileOutputStream fos = new FileOutputStream(newFilePath;
            fos.write(content.getData(;
        } catch (IOException e {
            e.printStackTrace(;
        }
    }

  @Test
    public void testBase64({
        saveFileWithBase64(FILE_PATH;
        getFileWithBase64(;
    }

    public void saveFileWithBase64(String filePath{
        // 读取文件并编码为 Base64 格式
        File file = new File(filePath;
        byte[] fileContent = new byte[(int file.length(];
        try (FileInputStream inputStream = new FileInputStream(file {
            inputStream.read(fileContent;
        } catch (IOException e {
            e.printStackTrace(;
        }
        // 把读取到的流转成base64
        String encodedString = Base64.getEncoder(.encodeToString(fileContent;
        // 将 Base64 编码的文件内容存储到 MongoDB 文档中
        Document document = new Document(;
        document.put("fileName", file.getName(;
        document.put("base64Content", encodedString;
        Document insert = mongoTemplate.insert(document, IMAGE_COLLECTION;
        System.out.println("文件 " + file.getName( + " 已存入mongodb，对应ID是: " + insert.get("_id".toString(;
    }

    public void getFileWithBase64({
        Criteria criteria = Criteria.where("base64Content".exists(true;
        List<Document> result = mongoTemplate.find(new Query(criteria, Document.class, IMAGE_COLLECTION;
        Document document = result.get(0;
        String base64Content = document.get("base64Content", String.class;
        String fileName = document.get("fileName", String.class;
        byte[] decode = Base64.getDecoder(.decode(base64Content;
        try {
            String newFilePath = FILE_OUTPUT_PATH + fileName;
            FileOutputStream fos = new FileOutputStream(newFilePath;
            fos.write(decode;
            System.out.println("文件已读取并复制到指定路径，详情为：" + newFilePath;
        } catch (IOException e {
            e.printStackTrace(;
        }
    }

package com.onepiece.mongo; import com.mongodb.client.MongoDatabase; import com.mongodb.client.gridfs.GridFSBucket; import com.mongodb.client.gridfs.GridFSBuckets; import com.mongodb.client.gridfs.model.GridFSUploadOptions; import org.bson.Document; import org.bson.types.ObjectId; import org.junit.jupiter.api.Test; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.data.mongodb.core.MongoTemplate; import org.springframework.data.mongodb.core.query.Query; import org.springframework.util.FileCopyUtils; import java.io.*; import java.util.List; /** * @author zaoyu * @description：使用GridFS存储文件并做读取。 */ @SpringBootTest public class MongoGridFS { @Autowired private MongoTemplate mongoTemplate; // GridFS下的bucket，自行指定要把文件存储到哪个bucket。 private String BUCKET_NAME = "images"; // 源文件，即要被存储的文件的绝对路径 private String FILE_PATH = "D:\\temp\\onepiece.jpg"; // 存储文件后自动生成的存储文件信息的collection，一般是xx.files。 private String COLLECTION_NAME = "images.files"; // 用于演示接收输出文件的路径 private String FILE_OUTPUT_PATH = "C:\\Users\\onepiece\\Desktop\\"; @Test public void testGridFSSaveFiles( { saveToGridFS(; System.out.println("------------"; readFromGridFS(; } /** * 传入bucketName得到指定bucket操作对象。 * * @param bucketName * @return */ public GridFSBucket createGridFSBucket(String bucketName { MongoDatabase db = mongoTemplate.getDb(; return GridFSBuckets.create(db, bucketName; } /** * 储存文件到GridFS */ public void saveToGridFS( { // 先调用上面方法得到一个GridFSBucket的操作对象 GridFSBucket gridFSBucket = createGridFSBucket(BUCKET_NAME; File file = new File(FILE_PATH; FileInputStream inputStream = null; try { inputStream = new FileInputStream(file; } catch (FileNotFoundException e { e.printStackTrace(; } // 设置GridFS存储配置，这里是设置了每个chunk(块的大小为1024个字节，也可以设置大一点。 MetaData是对文件的说明，如果不需要可以不写。 也是以键值对存在，BSON格式。 GridFSUploadOptions options = new GridFSUploadOptions(.chunkSizeBytes(1024.metadata(new Document("user", "onepiece"; // 调用GridFSBucket中的uploadFromStream方法，把对应的文件流传递进去，然后就会以binary(二进制格式存储到GridFS中，并得到一个文件在xx.files中的主键ID，后面可以用这个ID来查找关联的二进制文件数据。 ObjectId objectId = gridFSBucket.uploadFromStream(file.getName(, inputStream, options; System.out.println(file.getName( + "已存入mongodb gridFS, 对应id是：" + objectId; } /** * 从GridFS中读取文件 */ public void readFromGridFS( { // 这里查找条件我先不写，默认查所有，取第一条做验证演示。 用Document类接收。 List<Document> files = mongoTemplate.find(new Query(, Document.class, COLLECTION_NAME; Document file = files.get(0; // 得到主键ID，作为等下要查询的文件ID值。 ObjectId fileId = file.getObjectId("_id"; String filename = file.getString("filename"; // 先调用上面方法得到一个GridFSBucket的操作对象 GridFSBucket gridFSBucket = createGridFSBucket(BUCKET_NAME; // 调用openDownloadStream方法得到文件IO流。 InputStream downloadStream = gridFSBucket.openDownloadStream(fileId; FileOutputStream fileOutputStream = null; try { fileOutputStream = new FileOutputStream(FILE_OUTPUT_PATH + filename; // 把IO流直接到指定路径的输出流对象实现输出。 FileCopyUtils.copy(downloadStream, fileOutputStream; } catch (IOException e { e.printStackTrace(; } finally { try { fileOutputStream.close(; } catch (IOException e { e.printStackTrace(; } } } }