-
Notifications
You must be signed in to change notification settings - Fork 0
/
mongo-queue.js
75 lines (61 loc) · 1.88 KB
/
mongo-queue.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
var mongoose = require('mongoose');
var eventSchema = new mongoose.Schema({
when: Date,
data: {},
}, {
toObject: { virtuals: true }
})
eventSchema.index({when: 1});
module.exports = function MongoQueue(options) {
options = options || {};
var connection = options.connection || require('mongoose');;
var modelName = options.modelName || 'LongtermEvent';
var Event = connection.model('LongtermEvent', eventSchema, options.collection);
return {
peek: function(callback) {
Event.findOne({}).sort('when').limit(1).lean().exec(respond(callback ));
},
enqueue: function(when, data, callback) {
new Event({
when: when,
data: data
}).save(respond(callback));
},
update: function(id, data, callback) {
if (!isValid(id)) return callback(null, null);
Event.findByIdAndUpdate(id, {data: data}, {new: true}, respond(callback))
},
remove: function(id, callback) {
if (!isValid(id)) return callback(null, 0);
remove({_id: id}, callback);
},
find: function(id, callback) {
if (!isValid(id)) return callback(null, null);
Event.findById(id, respond(callback)).lean();
},
count: function(callback) {
Event.count({}, callback);
},
clear: function(callback) {
remove({}, callback);
}
}
function remove(what, callback) {
return Event.remove(what, function(err, data) {
var n = data && data.result && data.result.n ? data.result.n : 0;
callback(err, n);
})
}
}
function isValid(id) {
return typeof id === 'string' && /^[0-9a-fA-F]{24}$/.test(id);
}
function respond(callback) {
return function(err, event) {
if (err) console.error(err)
if (err) return callback(err);
if (event && typeof event.toObject === 'function') return callback(err, event.toObject());
if (event) event.id = String(event._id);
callback(err, event);
}
}