-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmdb.ts
78 lines (63 loc) · 2.55 KB
/
mdb.ts
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
import {MongoClient, MongoClientOptions} from 'mongodb';
import { Collection, Document } from 'mongodb';
async function connectToMongoDB(url: string, options?: MongoClientOptions) {
const client = await MongoClient.connect(url, options);
return client;
}
async function queryDocuments(collection: Collection<Document>, query: object): Promise<Document[]> {
const cursor = collection.find(query);
const documents = await cursor.toArray();
return documents;
}
async function insertDocument(collection: Collection<Document>, document: Document): Promise<void> {
await collection.insertOne(document);
}
async function insertDocuments(collection: Collection<Document>, documents: Document[]): Promise<void> {
await collection.insertMany(documents);
}
async function updateDocument(collection: Collection<Document>, query: object, document: Document): Promise<void> {
await collection.updateOne(query, document);
}
async function updateDocuments(collection: Collection<Document>, query: object, document: Document): Promise<void> {
await collection.updateMany(query, document);
}
async function deleteDocument(collection: Collection<Document>, query: object): Promise<void> {
await collection.deleteOne(query);
}
async function deleteDocuments(collection: Collection<Document>, query: object): Promise<void> {
await collection.deleteMany(query);
}
async function countDocuments(collection: Collection<Document>, query: object): Promise<number> {
const count = await collection.countDocuments(query);
return count;
}
async function createCollection(client: MongoClient, collectionName: string): Promise<Collection<Document>> {
const collection = await client.db().createCollection(collectionName);
return collection;
}
async function dropCollection(client: MongoClient, collectionName: string): Promise<void> {
await client.db().dropCollection(collectionName);
}
async function listCollections(client: MongoClient): Promise<string[]> {
const collections = await client.db().listCollections().toArray();
return collections.map(collection => collection.name);
}
async function listDatabases(client: MongoClient): Promise<string[]> {
const databases = await client.db().admin().listDatabases();
return databases.databases.map(database => database.name);
}
export {
connectToMongoDB,
queryDocuments,
insertDocument,
insertDocuments,
updateDocument,
updateDocuments,
deleteDocument,
deleteDocuments,
countDocuments,
createCollection,
dropCollection,
listCollections,
listDatabases
}