-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdb.js
61 lines (51 loc) · 1.39 KB
/
db.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
var express = require('express');
const path = require('path');
const crypto = require('crypto');
const dotenv = require('dotenv');
const multer = require('multer');
const GridFsStorage = require('multer-gridfs-storage');
const Grid = require('gridfs-stream');
const mongoose = require('mongoose');
dotenv.config();
// Set up mongoose connection
const mongoDB = process.env.DB_URL;
const conn = mongoose.connect(mongoDB, {
useNewUrlParser: true,
useUnifiedTopology: true,
useCreateIndex: true,
});
mongoose.Promise = global.Promise;
const dbInstance = mongoose.connection;
/*let gfs;
dbInstance.once('open', () => {
gfs = Grid(dbInstance.db, mongoose.mongo);
gfs.collection('fileUploads');
});*/
const gfs = { grid: undefined };
dbInstance.once('open', () => {
gfs.grid = Grid(dbInstance.db, mongoose.mongo);
gfs.grid.collection('fileUploads');
});
const storage = new GridFsStorage({
//url: mongoDB,
db: dbInstance,
file: (req, file) => {
return new Promise((resolve, reject) => {
crypto.randomBytes(32, (err, buff) => {
if (err) return reject(err);
const filename = buff.toString('hex') + path.extname(file.originalname);
const fileInfo = {
filename: filename,
bucketName: 'fileUploads',
};
resolve(fileInfo);
});
});
},
});
var upload = multer({ storage });
module.exports = {
dbInstance,
upload,
gfs,
};