Node.js MongoDB入库GridFS|极客笔记

Node.js MongoDB入库GridFS

在Node.js开发中，我们经常需要和数据库打交道。MongoDB是一个常用的NoSQL数据库，而在MongoDB中存储大文件通常会使用GridFS来处理。本文将详细介绍如何在Node.js中使用MongoDB的GridFS来将文件存储到数据库中。

GridFS简介

GridFS是MongoDB用于存储和检索大文件的规范。在GridFS中，一个大文件会被分成多个小文件存储在MongoDB的文件集合和块集合中。这样可以方便地存储和检索大文件，并且支持断点续传和分块下载。

安装MongoDB和相关依赖

首先，我们需要安装MongoDB并启动数据库服务。可以参考MongoDB官方文档进行安装步骤。

接下来，我们需要在Node.js项目中安装MongoDB的驱动程序mongodb和mongoose。

npm install mongodb mongoose

连接MongoDB数据库

在Node.js中使用MongoDB，首先需要连接数据库。我们可以使用mongoose来进行数据库连接。

const mongoose = require('mongoose');

mongoose.connect('mongodb://localhost:27017/mydatabase', { useNewUrlParser: true, useUnifiedTopology: true })
  .then(() => {
    console.log('MongoDB connected');
  })
  .catch(err => {
    console.error(err);
  });

其中，mongodb://localhost:27017/mydatabase是连接的数据库地址和数据库名称。

使用GridFS存储文件

接下来，我们来实现将文件存储到MongoDB中的GridFS中。我们可以使用gridfs-stream模块来实现这一功能。

npm install gridfs-stream

const mongoose = require('mongoose');
const grid = require('gridfs-stream');

let gfs;

mongoose.connection.once('open', () => {
  gfs = grid(mongoose.connection.db, mongoose.mongo);
});

const storage = require('multer-gridfs-storage')({
  url: 'mongodb://localhost:27017/mydatabase',
  file: (req, file) => {
    return {
      filename: ` ${Date.now()}-$ {file.originalname}`
    };
  }
});

const upload = multer({ storage });

app.post('/upload', upload.single('file'), async (req, res) => {
  try {
    res.send(req.file);
  } catch (error) {
    res.status(400).send(error);
  }
});

在上面的代码中，我们使用multer模块来处理文件上传，并利用gridfs-stream和multer-gridfs-storage模块将文件存储到GridFS中。

运行示例代码

接下来，我们可以编写一个简单的上传文件的页面来测试我们的代码。

<!DOCTYPE html>
<html lang="en">

<head>
  <meta charset="UTF-8">
  <meta http-equiv="X-UA-Compatible" content="IE=edge">
  <meta name="viewport" content="width=device-width, initial-scale=1.0">
  <title>File Upload</title>
</head>

<body>
  <h1>Upload File</h1>
  <form action="/upload" method="POST" enctype="multipart/form-data">
    <input type="file" name="file">
    <button type="submit">Submit</button>
  </form>
</body>

</html>

启动Node.js应用程序，并在浏览器中打开上传文件的页面，选择一个文件并提交表单，文件将被上传到MongoDB的GridFS中。