-
Notifications
You must be signed in to change notification settings - Fork 158
/
Copy pathindex.js
83 lines (77 loc) · 2.22 KB
/
index.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
82
83
var mongoose = require('mongoose'),
extend = require('extend');
var counterSchema = new mongoose.Schema({
model: {
type: String,
require: true
},
field: {
type: String,
require: true
},
c: {
type: Number,
default: 0
}
});
counterSchema.index({ field: 1, model: 1 }, { unique: true, required: true, index: -1 });
var Counter = mongoose.connection.model('MAIC-Counter', counterSchema);
module.exports = exports = function (schema, options) {
var settings = {
model: null,
field: '_id',
startAt: 0,
incrementBy: 1,
incrementOnUpdate: false
},
fields = {},
ready = false;
switch (typeof(options)) {
case 'string':
settings.model = options;
break;
case 'object':
extend(settings, options);
break;
}
fields[settings.field] = {
type: Number,
unique: true,
require: true
};
schema.add(fields);
Counter.findOne(
{ model: settings.model, field: settings.field },
function (err, res) {
if (!res) {
var counter = new Counter({ model: settings.model, field: settings.field, c: settings.startAt });
counter.save(function () {
ready = true;
});
}
else
ready = true;
}
);
schema.pre('save', function (next) {
var doc = this;
(function save() {
if (ready) {
Counter.collection.findAndModify(
{ model: settings.model, field: settings.field },
null,
{ $inc: { c: settings.incrementBy } },
{ new: true, upsert: true },
function (err, res) {
if (err) return next(err);
if (typeof(doc[settings.field]) !== 'number' || settings.incrementOnUpdate)
doc[settings.field] = res.c - 1;
next();
}
);
}
else
setTimeout(save, 5);
})();
});
};