-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathuploader.js
81 lines (53 loc) · 1.64 KB
/
uploader.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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
module.exports = function(router){
var multer = require('multer');
var processor = require('./processor');
var mongoose = require('mongoose');
var imgcontroller = require('./imageController');
var Img = mongoose.model('Image');
var async = require('async');
var q = async.queue(function(task, callback){
processor.runDreamer(task.upload_name, task.dream_name, 0, callback);
}, 2);
Img.find({ IsProcessed : false }).sort({ timestamp : 1 }).exec(function(err, images){
for(var i = 0; i < images.length; i++){
enqueue(images[i]);
}
});
q.drain = function() {
console.log('Queue drained!');
}
//Configuring multer
var storage = multer.diskStorage({
destination: function(req, file, cb){
cb(null, './upload');
},
filename: function(req, file, cb){
cb(null, file.originalname + '-' + Date.now());
},
});
var fileFilter = function(req, file, cb){
if(file.mimetype !== 'image/jpeg'){
return cb(new Error('Only jpegs are allowed!'));
}
cb(null, true);
}
var upload = multer({ storage: storage, fileFilter: fileFilter });
router.post('/api/image/process', upload.single('dream'), function(req, res, next){
var fname = req.file.filename;
var i = Img({ upload_name : fname, dream_name : 'dd_' + fname });
i.save(function(err, fb){
if(err) console.log(err);
enqueue(fb);
});
res.redirect('/');
});
function enqueue(item){
q.push(item, function(err){
if(err) console.log(err);
Img.update({ _id : item._id }, { IsProcessed : true }, function(err, item){
if(err) console.log(err);
console.log('Processed image: ' + item.upload_name);
});
});
}
}