在mlab中上传图像

时间:2018-08-13 15:32:42

标签: node.js mongodb mongoose multer

当前,我将由节点其余服务器上载的图像存储在本地目录“ / uploads”中。这正在不断增加我的回购规模。 为避免这种情况,我想像服务一样将图像文件存储在mongoDB地图集或mlab中。

    const express = require("express");
    const router = express.Router();
    const mongoose = require("mongoose");
    const multer = require('multer');

    const storage = multer.diskStorage({
      destination: function(req, file, cb) {
        cb(null, './uploads/');
      },
       filename: function(req, file, cb) {
        cb(null, new Date().toISOString() + file.originalname);
      }
    });

    const fileFilter = (req, file, cb) => {
     // reject a file
     if (file.mimetype === 'image/jpeg' || file.mimetype === 'image/png')           
     {
      cb(null, true);
      } else {
        cb(null, false);
     }
    };

    const upload = multer({
      storage: storage,
      limits: {
        fileSize: 1024 * 1024 * 5
      },
      fileFilter: fileFilter
    });

请为此提供帮助。预先感谢。

1 个答案:

答案 0 :(得分:2)

您可以通过使用猫鼬模式和fs核心模块对图像进行编码并从/uploads取消链接文件来实现此目的。

我将首先创建一个Mongoose模式,以设置要存储的与上载文件有关的所有信息的模型。

在此示例中,我将使用base64编码。

uploadModel.js

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

mongoose.set('useCreateIndex', true);

let uploadSchema = new Schema({
    name: {
      type: String,
    },
    mimetype: {
      type: String,
    },
    size: {
      type: Number,
    },
    base64: {
      type: String,
    }
})

module.exports = mongoose.model('upload',uploadSchema);

设置模型后,创建一个函数以对base64进行编码,并module.exports也对其进行编码。

要对文件进行编码,请使用fs.readFileSync(path_to_file, encode_type)。将文件编码并保存到变量后,可以使用fs.unlink(path_to_file)/uploads文件夹中删除文件。

uploadModel.js

module.exports.base64_encode = function(file) {
  return new Promise((resolve, reject) => {
    if(file == undefined){
      reject('no file found');
    } else {
      let encodedData = fs.readFileSync(file, 'base64');
      fs.unlink(file);
      resolve(encodedData.toString('base64'));
    }
  })
} 

现在,您的路线文件中需要您的模型。

route.js

const Upload = require('path_to_uploadModel');

router.post('/path_to_upload', upload.single('form_name_of_file'), (req, res) => {
  let img = req.file;

  let model = new Upload({
    name: img.originalname,
    size: img.size,
    mimetype: img.mimetype,
  })


  Upload.base64_encode(img.path)
    .then((base64) => {
      model['base64'] = base64;
      model.save((err)=> {
        if(err) throw err;
      });
    }
})

希望这会有所帮助