Spring Boot使用MongoDB GridFS進行文件的操作

1. GridFS簡介

GridFS 用於存儲和恢復那些超過16M（BSON文件限制）的文件(如：圖片、音頻、視頻等)，但是它是存儲在MonoDB的集合中。

GridFS 會將文件對象分割成多個的chunk(文件片段)，一般為256k/個，每個chunk將作為MongoDB的一個文檔(document)被存儲在chunks集合中。

GridFS 用兩個集合來存儲一個文件：fs.files與fs.chunks。

每個文件的實際內容被存在chunks(二進制數據)中，和文件有關的meta數據(filename、content_type、還有用戶自定義的屬性)將會被存在files集合中。

以下是簡單的 fs.files 集合文檔：

{
    "_id" : ObjectId("5de752d43f189c3260063ce9"),    // 文件id
    "filename" : "index_banner.ftl",                 // 文件名
    "length" : NumberLong(1908),                     // 文件長度
    "chunkSize" : 261120,                            // chunk的大小
    "uploadDate" : ISODate("2019-12-04T14:31:48.487+08:00"),  // 上傳時間
    "md5" : "a8d14fbafebfca33c8abdac2b83840cc",        // 文件的md5值
    "contentType" : "text/plain"     // 文件的MIME類型
}

以下是簡單的 fs.chunks 集合文檔：

{
   "_id" : ObjectId("5de752d43f189c3260063cea"),        // chunk的id
    "files_id" : ObjectId("5de752d43f189c3260063ce9"),  // 文件id，對應fs.files中對象id
    "n" : 0,                                    // 文件的第幾個chunk塊，如果文件大於chunksize的話，會被分割成多個chunk塊
    "data" : "Mongo Binary Data"                // 文件的二進制數據
}

2.代碼示例

代碼基於spring boot，主要實現GridFS的基本操作。

（1）pom.xml中引入  

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>

（2）application.yml配置

spring:
  data:
    mongodb:
      uri: mongodb://localhost:27017
      database: note_test

 （3）基本操作

package com.maybesuch.controller;

import com.mongodb.client.gridfs.model.GridFSFile;
import org.apache.commons.io.IOUtils;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.gridfs.GridFsResource;
import org.springframework.data.mongodb.gridfs.GridFsTemplate;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

/**
 * @author : maybesuch
 * @version : 1.0
 * @Description : GridFS文件基本操作
 * @Date : 2020/1/8 10:16
 * @Copyright : Copyright (c) 2020 All Rights Reserved
 **/
@RestController
@RequestMapping("/gridfs")
public class GridFSController {

    @Autowired
    private GridFsTemplate gridFsTemplate;

    /**
     * 上傳文件
     *
     * @param multipartFile 文件
     * @return 上傳成功文件id
     */
    @PostMapping("/upload")
    public String uploadFile(@RequestParam(value = "file") MultipartFile multipartFile) {

        // 設置meta數據值
        Map<String, String> metaData = new HashMap<>();
        metaData.put("tags", "test");
        // ...
        try (
                InputStream inputStream = multipartFile.getInputStream();
        ) {
            // 獲取文件的源名稱
            String fileName = multipartFile.getOriginalFilename();
            // 進行文件存儲
            ObjectId objectId = gridFsTemplate.store(inputStream, fileName, metaData);
            // 返回文件的id
            return objectId.toHexString();

        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 獲取文件信息
     * @param fileId 文件id
     */
    @GetMapping("/get/{fileId}")
    public void getFile(@PathVariable("fileId") String fileId) {

        //根據id查詢文件
        GridFSFile gridFSFile = gridFsTemplate.findOne(Query.query(Criteria.where("_id").is(fileId)));

        if (gridFSFile == null) {
            throw new RuntimeException("No file with id: " + fileId);
        }
        //獲取流對象
        GridFsResource resource = gridFsTemplate.getResource(gridFSFile);

        /*可根據實際需求進行數據的獲取*/
        try {
            //獲取流中的數據
            String content = IOUtils.toString(resource.getInputStream(), "UTF-8");
            //獲取byte[]信息
            byte[] bytes = IOUtils.toByteArray(resource.getInputStream());

        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 刪除文件
     * @param fileId 文件id
     */
    @DeleteMapping("/delete")
    public void deleteFile(@RequestParam(value = "fileId") String fileId) {
        // 根據文件id刪除fs.files和fs.chunks中的記錄
        gridFsTemplate.delete(Query.query(Criteria.where("_id").is(fileId)));

    }

}