簡體   English   中英

如何使用 GridFS 使用 Node.js 和 Mongoose 存儲圖像

[英]How to use GridFS to store images using Node.js and Mongoose

我是 Node.js 的新手。 誰能給我一個例子,說明如何使用 GridFS 存儲和檢索二進制數據,例如圖像,使用 Node.js 和 Mongoose? 我需要直接訪問 GridFS 嗎?

我對這里評分最高的答案不滿意,所以我提供了一個新答案:我最終使用了可以通過 npm 安裝的節點模塊“gridfs-stream” (那里有很棒的文檔!)。 有了它,再加上貓鼬,它可能看起來像這樣:

var fs = require('fs');
var mongoose = require("mongoose");
var Grid = require('gridfs-stream');
var GridFS = Grid(mongoose.connection.db, mongoose.mongo);

function putFile(path, name, callback) {
    var writestream = GridFS.createWriteStream({
        filename: name
    });
    writestream.on('close', function (file) {
      callback(null, file);
    });
    fs.createReadStream(path).pipe(writestream);
}

請注意,路徑是本地系統上文件的路徑。

至於我的文件讀取功能,就我而言,我只需要將文件流式傳輸到瀏覽器(使用 express):

try {
    var readstream = GridFS.createReadStream({_id: id});
    readstream.pipe(res);
} catch (err) {
    log.error(err);
    return next(errors.create(404, "File not found."));
}

到目前為止的答案都很好,但是,我相信在這里記錄如何使用官方的mongodb nodejs 驅動程序而不是依賴於諸如“gridfs-stream”之類的進一步抽象將是有益的。

之前的一個答案確實使用了官方的 mongodb 驅動程序,但是他們使用了 Gridstore API; 此后已被棄用,請參見此處 我的示例將使用新的GridFSBucket API

這個問題非常廣泛,因此我的答案將是整個 nodejs 程序。 這將包括設置 express 服務器、mongodb 驅動程序、定義路由以及處理 GET 和 POST 路由。

使用的 Npm 包

  • express(nodejs Web 應用程序框架,用於簡化此代碼段)
  • multer(用於處理多部分/表單數據請求)
  • mongodb(官方 mongodb nodejs 驅動)

GET photo 路由以 Mongo ObjectID 作為參數來檢索圖像。

我配置 multer 將上傳的文件保存在內存中。 這意味着照片文件不會隨時寫入文件系統,而是直接從內存流式傳輸到 GridFS。


/**
 * NPM Module dependencies.
 */
const express = require('express');
const photoRoute = express.Router();

const multer = require('multer');
var storage = multer.memoryStorage()
var upload = multer({ storage: storage, limits: { fields: 1, fileSize: 6000000, files: 1, parts: 2 }});

const mongodb = require('mongodb');
const MongoClient = require('mongodb').MongoClient;
const ObjectID = require('mongodb').ObjectID;
let db;

/**
 * NodeJS Module dependencies.
 */
const { Readable } = require('stream');

/**
 * Create Express server && Routes configuration.
 */
const app = express();
app.use('/photos', photoRoute);

/**
 * Connect Mongo Driver to MongoDB.
 */
MongoClient.connect('mongodb://localhost/photoDB', (err, database) => {
  if (err) {
    console.log('MongoDB Connection Error. Please make sure that MongoDB is running.');
    process.exit(1);
  }
  db = database;
});

/**
 * GET photo by ID Route
 */
photoRoute.get('/:photoID', (req, res) => {
  try {
    var photoID = new ObjectID(req.params.photoID);
  } catch(err) {
    return res.status(400).json({ message: "Invalid PhotoID in URL parameter. Must be a single String of 12 bytes or a string of 24 hex characters" }); 
  }

  let bucket = new mongodb.GridFSBucket(db, {
    bucketName: 'photos'
  });

  let downloadStream = bucket.openDownloadStream(photoID);

  downloadStream.on('data', (chunk) => {
    res.write(chunk);
  });

  downloadStream.on('error', () => {
    res.sendStatus(404);
  });

  downloadStream.on('end', () => {
    res.end();
  });
});

/**
 * POST photo Route
 */
photoRoute.post('/', (req, res) => {
  upload.single('photo')(req, res, (err) => {
    if (err) {
      return res.status(400).json({ message: "Upload Request Validation Failed" });
    } else if(!req.body.name) {
      return res.status(400).json({ message: "No photo name in request body" });
    }
    
    let photoName = req.body.name;
    
    // Covert buffer to Readable Stream
    const readablePhotoStream = new Readable();
    readablePhotoStream.push(req.file.buffer);
    readablePhotoStream.push(null);

    let bucket = new mongodb.GridFSBucket(db, {
      bucketName: 'photos'
    });

    let uploadStream = bucket.openUploadStream(photoName);
    let id = uploadStream.id;
    readablePhotoStream.pipe(uploadStream);

    uploadStream.on('error', () => {
      return res.status(500).json({ message: "Error uploading file" });
    });

    uploadStream.on('finish', () => {
      return res.status(201).json({ message: "File uploaded successfully, stored under Mongo ObjectID: " + id });
    });
  });
});

app.listen(3005, () => {
  console.log("App listening on port 3005!");
});

我寫了一篇關於這個主題的博客文章; 是對我的回答的詳細說明。 在這里可用

進一步閱讀/靈感:

我建議看看這個問題: MongoDB GridFS Saving Files with Node.JS 的問題

從答案中復制的示例(歸功於 christkv):

// You can use an object id as well as filename now
var gs = new mongodb.GridStore(this.db, filename, "w", {
  "chunk_size": 1024*4,
  metadata: {
    hashpath:gridfs_name,
    hash:hash,
    name: name
  }
});

gs.open(function(err,store) {
  // Write data and automatically close on finished write
  gs.writeBuffer(data, true, function(err,chunk) {
    // Each file has an md5 in the file structure
    cb(err,hash,chunk);
  });
});

看起來 writeBuffer 已經被棄用了。

/Users/kmandrup/private/repos/node-mongodb-native/HISTORY:
   82  * Fixed dereference method on Db class to correctly dereference Db reference objects. 
   83  * Moved connect object onto Db class(Db.connect) as well as keeping backward compatibility.
   84: * Removed writeBuffer method from gridstore, write handles switching automatically now.
   85  * Changed readBuffer to read on Gridstore, Gridstore now only supports Binary Buffers no Strings anymore.

刪除文件上傳庫,如果它給出了一些與多部分標題相關的錯誤,而不是從標題中刪除內容類型

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM