-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
109 lines (98 loc) · 3.1 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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
'use strict'
const mongodb = require('mongodb');
let MongoClient = mongodb.MongoClient;
let connectPromise;
function connect(url, options) {
if(!options) {
options = {};
}
return new Promise((resolve, reject)=> {
MongoClient.connect(url, options, function(err, db){
if(err) reject(err);
else resolve(db);
});
});
}
function getConnectPromise(url, options) {
connectPromise = connect(url, options);
return connectPromise;
}
function close() {
connectPromise.then(db=> db.close());
}
function findList(collectionName, query) {
return connectPromise
.then(db=>{
return new Promise((resolve, reject) => {
db.collection(collectionName).find(query).toArray(function(err, docs) {
if(err) reject(err);
else resolve(docs);
});
});
});
}
function insertOne(collectionName, data) {
return connectPromise
.then(db=>{
return new Promise((resolve, reject) => {
db.collection(collectionName).insertOne(data, function(err, res) {
if(err) reject(err);
else resolve(res);
});
});
});
}
function insertMany(collectionName, data) {
return connectPromise
.then(db=>{
return new Promise((resolve, reject) => {
if(data && data.length>=1){
db.collection(collectionName).insertMany(data, function(err, res) {
if(err) reject(err);
else {
if(res && res.insertedCount==data.length){
resolve(res);
}else{
//this case not been tested, mongo seems can insert different type to same field
reject(new Error("Only "+res.insertedCount+" inserted from " +data.length));
}
}
});
}else{
reject(new Error('input is not an array'));
}
});
});
}
function findOne(collectionName, query) {
return connectPromise
.then(db=>{
return new Promise((resolve, reject) => {
db.collection(collectionName).findOne(query, function(err, doc) {
if(err) reject(err);
else resolve(doc);
});
});
});
}
function deleteMany(collectionName, filter, options) {
if(!options) {
options = {};
}
return connectPromise
.then(db=> {
return new Promise((resolve, reject) => {
db.collection(collectionName).deleteMany(filter, options, function(err, result) {
if(err) reject(err);
else resolve(result);
});
});
});
}
exports.getConnectPromise = getConnectPromise;
exports.close = close;
exports.insertOne = insertOne;
exports.insertMany = insertMany;
exports.findOne = findOne;
exports.findList = findList;
exports.deleteMany = deleteMany;