如何在NodeJS中的MongoDB中保存和检索pdf文件

时间:2019-01-04 00:43:40

标签: node.js mongodb pdfmake

我遇到了一个问题,我已将使用Express在Node.js服务器后端中创建的pdf小文件(〜128KB)保存到Mongodb中的文档中。我没有使用Mongo GridFS,因为文件将始终在16MB的限制之下。该集合具有以下架构:

const mongoose = require('mongoose');
const Schema = mongoose.Schema;

const ReportsSchema = new Schema({
  ReportFileData: Buffer,
  Cert_objid: { type: Schema.Types.ObjectId, ref: 'Certs' },
  Report_Type: String,
  Note: String,
  Dau_objID: { type: Schema.Types.ObjectId, ref: 'Dau' },
  Pau_objID: { type: Schema.Types.ObjectId, ref: 'Pau' }
});

module.exports = Reports = mongoose.model('Reports', ReportsSchema);

我使用以下代码创建pdf文件并保存。我没有包括docDefinition,因为如果将其直接保存到服务器中,它将生成正确的文件。

const PdfPrinter = require('pdfmake/src/printer');
const path = require('path');
const moment = require('moment');

const Reports = require('../../models/Reports');  // Mongoose Schema

const createFAReport = data => {
    docDefinition...

createPdfBinary(docDefinition, binary => {
  const rpt = new Reports({
    ReportFileData: binary,
    Cert_objid: data._id,
    Report_Type: 'Water Use Report',
    Note: 'Testing 123'
  });

  rpt.save(err => {
    if (err) throw err;
    });
});

const createPdfBinary = (pdfDoc, callback) => {
  const fonts = {
    Roboto: {
    normal: path.join(__dirname, '../../', '/fonts/Roboto-Regular.ttf'),
    bold: path.join(__dirname, '../../', '/fonts/Roboto-Medium.ttf'),
    italics: path.join(__dirname, '../../', '/fonts/Roboto-Italic.ttf'),
    bolditalics: path.join(__dirname, '../../', '/fonts/Roboto-MediumItalic.ttf')
    }
  };

  const printer = new PdfPrinter(fonts);
  const doc = printer.createPdfKitDocument(pdfDoc);
  const chunks = [];
  let result;

  doc.on('data', function(chunk) {
    chunks.push(chunk);
  });
  doc.on('end', function() {
    result = Buffer.concat(chunks);
    callback('data:application/pdf;base64,' + result.toString('base64'));
  });
  doc.end();
};

然后从MongoDB中检索编码的文档并将其写入本地文件以进行测试,我使用了以下代码(请注意,汇总是为了获取一些检索正确报告所需的关联字段):

router.get('/getReport', passport.authenticate('jwt', { session: false }), (req, res) => {
  Certs.aggregate([
    {
      $match: {
        Cert_ID: '1578'
      }
    },
    {
      $lookup: {
        from: 'reports',
        localField: '_id',
        foreignField: 'Cert_objid',
        as: 'rpt'
      }
    },
    {
      $unwind: {
        path: '$rpt',
        includeArrayIndex: '<<string>>',
        preserveNullAndEmptyArrays: false
      }
    }
  ]).then(result => {
    result.map(rslt => {
      console.log(rslt.Cert_ID);
      res.json({ msg: 'Got the report.' });

      const fullfilePath = path.join(__dirname, '../../', '/public/pdffiles/', `1578.pdf`
      );

      fs.writeFile(fullfilePath, rslt.rpt.ReportFileData, 'base64', () => {
        console.log('File Saved.');
       });
     });
   });
 });

一切似乎都可以正常工作,但当我打开文件时,出现错误,表明文件已损坏。我想知道将其保存到“ base64”是一个问题还是MongoDB的数据类型是一个问题。数据类型是缓冲区,所以您会检索它作为缓冲区吗?任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:0)

我不建议直接将pdf或图像写入数据库。以下是有关Storing Images in DB - Yea or Nay?

的一些信息

通常,您可以保存文件名并将文件存储在文件系统中,您自己的或更具扩展性的选项将类似于S3。

这是一个可以帮助您https://www.npmjs.com/package/formidable的模块,如果您打算自己动手,您仍然可以从中获得一些启发。