前言
MongoDB 是一种流行的 NoSQL 数据库，它的文档模型非常适合存储复杂的数据结构。但是，MongoDB 的文档大小有限制，如果要存储大文件，就需要使用 GridFS。GridFS 是 MongoDB 提供的一种用于存储和检索大文件的机制，它允许将大文件分成小块进行存储，并提供了方便的接口进行管理和检索。
本文将介绍如何使用 MongoDB 的 GridFS 存储和检索大文件，包括如何上传文件、下载文件和删除文件等操作。
GridFS 的基本概念
在使用 GridFS 之前，我们需要了解一些基本概念：
- 文件块（Chunk）：将大文件切分成固定大小的块，每个块默认大小为 255KB，最后一个块的大小可能小于 255KB。
- 文件元数据（Metadata）：存储文件的元数据信息，如文件名、文件类型、文件大小等。
- 文件对象（File Object）：将文件块和文件元数据组合在一起，形成一个文件对象。
GridFS 的上传操作
在使用 GridFS 上传文件时，需要将文件切分成块，并将块存储到 MongoDB 中。同时，还需要将文件的元数据信息存储到 MongoDB 中。
以下是 GridFS 上传文件的示例代码：
// javascriptcn.com 代码示例
const { MongoClient } = require('mongodb');
const fs = require('fs');
const uri = 'mongodb://localhost:27017';
const client = new MongoClient(uri, { useUnifiedTopology: true });
async function uploadFileToGridFS(filename, filepath) {
try {
await client.connect();
const db = client.db('mydatabase');
const bucket = new db.GridFSBucket();
const readStream = fs.createReadStream(filepath);
const uploadStream = bucket.openUploadStream(filename);
readStream.pipe(uploadStream);
return new Promise((resolve, reject) => {
uploadStream.on('finish', resolve);
uploadStream.on('error', reject);
});
} finally {
await client.close();
}
}
const filename = 'test.txt';
const filepath = './test.txt';
uploadFileToGridFS(filename, filepath)
.then(() => console.log('File uploaded successfully'))
.catch(console.error);在上面的代码中，我们首先创建了一个 MongoClient 实例，并连接到 MongoDB。然后，我们创建了一个 GridFSBucket 对象，用于将文件块存储到 MongoDB 中。
接着，我们使用 Node.js 内置的 fs 模块创建一个读取文件的可读流（readStream），并使用 GridFSBucket 的 openUploadStream 方法创建一个可写流（uploadStream）。我们将读取文件的可读流（readStream）通过管道（pipe）的方式传输给可写流（uploadStream），从而实现将文件块存储到 MongoDB 中。
最后，我们通过 Promise 对象的方式监听可写流的 finish 和 error 事件，以便在文件上传完成或发生错误时进行处理。
GridFS 的下载操作
在使用 GridFS 下载文件时，需要将文件块从 MongoDB 中取出，并将块组合成完整的文件。同时，还需要从 MongoDB 中获取文件的元数据信息。
以下是 GridFS 下载文件的示例代码：
// javascriptcn.com 代码示例
const { MongoClient } = require('mongodb');
const fs = require('fs');
const uri = 'mongodb://localhost:27017';
const client = new MongoClient(uri, { useUnifiedTopology: true });
async function downloadFileFromGridFS(filename, filepath) {
try {
await client.connect();
const db = client.db('mydatabase');
const bucket = new db.GridFSBucket();
const downloadStream = bucket.openDownloadStreamByName(filename);
const writeStream = fs.createWriteStream(filepath);
downloadStream.pipe(writeStream);
return new Promise((resolve, reject) => {
writeStream.on('finish', resolve);
writeStream.on('error', reject);
});
} finally {
await client.close();
}
}
const filename = 'test.txt';
const filepath = './test-download.txt';
downloadFileFromGridFS(filename, filepath)
.then(() => console.log('File downloaded successfully'))
.catch(console.error);在上面的代码中，我们首先创建了一个 MongoClient 实例，并连接到 MongoDB。然后，我们创建了一个 GridFSBucket 对象，用于从 MongoDB 中获取文件块。
接着，我们使用 GridFSBucket 的 openDownloadStreamByName 方法创建一个可读流（downloadStream），并使用 Node.js 内置的 fs 模块创建一个写入文件的可写流（writeStream）。我们将可读流（downloadStream）通过管道（pipe）的方式传输给可写流（writeStream），从而实现将文件块组合成完整的文件。
最后，我们通过 Promise 对象的方式监听可写流的 finish 和 error 事件，以便在文件下载完成或发生错误时进行处理。
GridFS 的删除操作
在使用 GridFS 删除文件时，需要同时删除文件块和文件元数据信息。
以下是 GridFS 删除文件的示例代码：
// javascriptcn.com 代码示例
const { MongoClient } = require('mongodb');
const uri = 'mongodb://localhost:27017';
const client = new MongoClient(uri, { useUnifiedTopology: true });
async function deleteFileFromGridFS(filename) {
try {
await client.connect();
const db = client.db('mydatabase');
const bucket = new db.GridFSBucket();
await bucket.delete(filename);
} finally {
await client.close();
}
}
const filename = 'test.txt';
deleteFileFromGridFS(filename)
.then(() => console.log('File deleted successfully'))
.catch(console.error);在上面的代码中，我们首先创建了一个 MongoClient 实例，并连接到 MongoDB。然后，我们创建了一个 GridFSBucket 对象，用于删除文件块和文件元数据信息。
接着，我们使用 GridFSBucket 的 delete 方法删除指定文件名的文件块和文件元数据信息。
最后，我们通过 Promise 对象的方式监听操作完成或发生错误时进行处理。
总结
本文介绍了如何使用 MongoDB 的 GridFS 存储和检索大文件。我们通过示例代码演示了如何上传文件、下载文件和删除文件等操作。希望本文能够帮助读者更好地理解 GridFS 的基本概念和操作方法，从而更好地应用 MongoDB 存储大文件。
来源：JavaScript中文网 ，转载请注明来源 本文地址：https://www.javascriptcn.com/post/653cb5377d4982a6eb6bd38f