jaredwray#1084 Implement disconnect for mongodb adapter for a gracefu…

…l shutdown.

packages/mongo/src/index.ts

@@ -75,7 +75,9 @@ class KeyvMongo extends EventEmitter implements KeyvStoreAdapter {
 					await store.createIndex({'metadata.expiresAt': 1});
 					await store.createIndex({'metadata.lastAccessed': 1});
 
-					resolve({bucket, store, db: database});
+					resolve({
+						bucket, store, db: database, mongoClient: client,
+					});
 				} else {
 					let collection = 'keyv';
 					if (this.opts.collection) {
@@ -87,7 +89,7 @@ class KeyvMongo extends EventEmitter implements KeyvStoreAdapter {
 					await store.createIndex({key: 1}, {unique: true, background: true});
 					await store.createIndex({expiresAt: 1}, {expireAfterSeconds: 0, background: true});
 
-					resolve({store});
+					resolve({store, mongoClient: client});
 				}
 			} catch (error) {
 				this.emit('error', error);
@@ -317,6 +319,11 @@ class KeyvMongo extends EventEmitter implements KeyvStoreAdapter {
 		const document = await client.store.count(filter);
 		return document !== 0;
 	}
+
+	async disconnect(): Promise<void> {
+		const client = await this.connect;
+		await client.mongoClient.close();
+	}
 }
 
 export default KeyvMongo;

packages/mongo/src/types.ts

@@ -1,4 +1,5 @@
 import {
+	type MongoClient,
 	type Collection, type Db, type GridFSBucket, type ReadPreference,
 } from 'mongodb';
 
@@ -22,6 +23,7 @@ export type KeyvMongoConnect = {
 	bucket?: GridFSBucket;
 	store: Collection;
 	db?: Db;
+	mongoClient: MongoClient;
 };
 
 export type PifyFunction = (...arguments_: any[]) => any;

packages/mongo/test/test.ts

@@ -15,6 +15,7 @@ test.afterAll(async () => {
 	await keyv.clear();
 	keyv = new KeyvMongo({collection: 'foo', useGridFS: true, ...options});
 	await keyv.clear();
+	await keyv.disconnect();
 });
 
 test.beforeEach(async () => {
@@ -205,3 +206,36 @@ test.it('iterator with namespace', async t => {
 	entry = await iterator.next();
 	t.expect(entry.value).toBeUndefined();
 });
+
+test.it('Close connection successfully on GridFS', async t => {
+	const keyv = new KeyvMongo({useGridFS: true, ...options});
+	t.expect(await keyv.get('foo')).toBeUndefined();
+	await keyv.disconnect();
+	try {
+		await keyv.get('foo');
+		t.expect.fail();
+	} catch {
+		t.expect(true).toBeTruthy();
+	}
+});
+
+test.it('Close connection successfully', async t => {
+	const keyv = new KeyvMongo({namespace: 'key1', ...options});
+	t.expect(await keyv.get('foo')).toBeUndefined();
+	await keyv.disconnect();
+	try {
+		await keyv.get('foo');
+		t.expect.fail();
+	} catch {
+		t.expect(true).toBeTruthy();
+	}
+});
+
+test.it('Close connection should fail', async t => {
+	const keyv = new KeyvMongo({namespace: 'key1', ...options});
+	try {
+		await keyv.disconnect();
+	} catch {
+		t.expect(true).toBeTruthy();
+	}
+});