如何使用nodeJS下载保存在gridFS中的文件

时间:2015-10-09 10:00:37

标签: node.js mongodb gridfs

我需要从GridFS下载一份简历,下面是我写的代码,但这似乎没有给我一个物理文件供下载,这用于阅读内容。我该如何下载文件?

exports.getFileById = function(req, res){
var conn = mongoose.connection;
var gfs = Grid(conn.db, mongoose.mongo);
var id = req.params.ID;
gfs.exist({_id: id,root: 'resume'}, function (err, found) {
    if (err) return handleError(err);
    if (!found)
        return res.send('Error on the database looking for the file.');
    gfs.createReadStream({_id: id,root: 'resume'}).pipe(res);
});
};

2 个答案:

答案 0 :(得分:5)

希望这有帮助!

exports.getFileById = function(req, res){
var role = req.session.user.role;
var conn = mongoose.connection;
var gfs = Grid(conn.db, mongoose.mongo);
gfs.findOne({ _id: req.params.ID, root: 'resume' }, function (err, file) {
    if (err) {
        return res.status(400).send(err);
    }
    else if (!file) {
        return res.status(404).send('Error on the database looking for the file.');
    }

    res.set('Content-Type', file.contentType);
    res.set('Content-Disposition', 'attachment; filename="' + file.filename + '"');

    var readstream = gfs.createReadStream({
      _id: req.params.ID,
      root: 'resume'
    });

    readstream.on("error", function(err) { 
        res.end();
    });
    readstream.pipe(res);
  });
};

答案 1 :(得分:2)

我从接受的答案中得到了提示。但是我必须跳过一些障碍才能使它正常工作,希望这会有所帮助。

const mongodb = require('mongodb');
const mongoose = require('mongoose');
const Grid = require('gridfs-stream');
eval(`Grid.prototype.findOne = ${Grid.prototype.findOne.toString().replace('nextObject', 'next')}`);

const mongoURI = config.mongoURI;
const connection = mongoose.createConnection(mongoURI);

app.get('/download', async (req, res) => {
    var id = "<file_id_xyz>";
    gfs = Grid(connection.db, mongoose.mongo);

    gfs.collection("<name_of_collection>").findOne({ "_id": mongodb.ObjectId(id) }, (err, file) => {
        if (err) {
            // report the error
            console.log(err);
        } else {
            // detect the content type and set the appropriate response headers.
            let mimeType = file.contentType;
            if (!mimeType) {
                mimeType = mime.lookup(file.filename);
            }
            res.set({
                'Content-Type': mimeType,
                'Content-Disposition': 'attachment; filename=' + file.filename
            });

            const readStream = gfs.createReadStream({
                _id: id
            });
            readStream.on('error', err => {
                // report stream error
                console.log(err);
            });
            // the response will be the file itself.
            readStream.pipe(res);
        }
    });