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
110
|
import * as mongodb from 'mongodb';
export class Database {
public static Instance = new Database()
private MongoClient = mongodb.MongoClient;
private url = 'mongodb://localhost:27017/Dash';
private db?: mongodb.Db;
constructor() {
this.MongoClient.connect(this.url, (err, client) => {
this.db = client.db()
})
}
private currentWrites: { [_id: string]: Promise<void> } = {};
public update(id: string, value: any, callback: () => void) {
if (this.db) {
let collection = this.db.collection('documents');
const prom = this.currentWrites[id];
const run = (promise: Promise<void>, resolve?: () => void) => {
collection.updateOne({ _id: id }, { $set: value }, {
upsert: true
}, (err, res) => {
if (err) {
console.log(err.message);
console.log(err.errmsg);
}
// if (res) {
// console.log(JSON.stringify(res.result));
// }
if (this.currentWrites[id] === promise) {
delete this.currentWrites[id]
}
if (resolve) {
resolve();
}
callback();
});
}
if (prom) {
const newProm: Promise<void> = prom.then(() => run(newProm));
this.currentWrites[id] = newProm;
} else {
const newProm: Promise<void> = new Promise<void>(res => run(newProm, res))
this.currentWrites[id] = newProm;
}
}
}
public delete(id: string) {
if (this.db) {
let collection = this.db.collection('documents');
collection.remove({ _id: id });
}
}
public deleteAll(collectionName: string = 'documents'): Promise<any> {
return new Promise(res => {
if (this.db) {
let collection = this.db.collection(collectionName);
collection.deleteMany({}, res);
}
})
}
public insert(kvpairs: any) {
if (this.db) {
let collection = this.db.collection('documents');
collection.insertOne(kvpairs, (err: any, res: any) => {
if (err) {
// console.log(err)
return
}
});
}
}
public getDocument(id: string, fn: (res: any) => void) {
var result: JSON;
if (this.db) {
let collection = this.db.collection('documents');
collection.findOne({ _id: id }, (err: any, res: any) => {
result = res
if (!result) {
fn(undefined)
}
fn(result)
})
};
}
public getDocuments(ids: string[], fn: (res: any) => void) {
if (this.db) {
let collection = this.db.collection('documents');
let cursor = collection.find({ _id: { "$in": ids } })
cursor.toArray((err, docs) => {
if (err) {
console.log(err.message);
console.log(err.errmsg);
}
fn(docs);
})
};
}
public print() {
console.log("db says hi!")
}
}
|
