feat: monitor MongoDB client

Author: masontikhonov

infra/mongo.js

@@ -1,50 +1,54 @@
+// @ts-check
 const { MongoClient, ObjectId } = require('mongodb');
+const { mongodb, logs } = require('@codefresh-io/cf-telemetry');
 const { getDbNameFromUri } = require('./helper');
 
 class Mongo {
     constructor() {
         this.db = undefined;
         this.ObjectId = ObjectId;
+        this.logger = new logs.Logger('codefresh:infra:mongo');
     }
 
     /**
      * starts the connection to mongo
+     * @returns {Promise<void>}
      */
     async init(config) {
-        const clientSettings = { ...config.mongo.options };
-        const logger = require('cf-logs').Logger('codefresh:infra:mongo'); // eslint-disable-line
-        this.logger = logger;
-
-        const { uri } = config.mongo;
-        const dbName = config.mongo.dbName || getDbNameFromUri(uri);
-        const client = new MongoClient(uri, clientSettings);
-        logger.info(`Mongo db name ${dbName}`);
-
         try {
-            await client.connect();
-            logger.info('Mongo driver connected');
+            const clientSettings = { ...config.mongo.options };
+            const { uri } = config.mongo;
+            const dbName = config.mongo.dbName || getDbNameFromUri(uri);
+            this.logger.info(`Mongo db name ${dbName}`);
+            const client = new MongoClient(uri, clientSettings);
+            mongodb.monitorMongoDBClient(client);
+
+            this.client = await client.connect();
+            this.logger.info('Mongo driver connected');
+            this.db = this.client.db(dbName);
+            this.logger.info('Mongo db initialized');
         } catch (error) {
-            logger.error('Error connecting to MongoDB:', error);
+            this.logger.error(error, 'Error connecting to MongoDB');
             throw error;
         }
-
-        this.client = client;
-        this.db = this.client.db(dbName);
-        logger.info('Mongo db initialized');
     }
 
 
     /**
      * stops the connection to mongo
+     * @returns {Promise<void>}
      */
     async stop() {
-        if (!this.db) {
-            return;
-        }
+        if (!this.client) return;
         await this.client.close();
     }
 
+    /**
+     * @param {string} collectionName
+     * @returns {import('mongodb').Collection}
+     */
     collection(collectionName) {
+        // @ts-ignore
         return this.db.collection(collectionName);
     }
 }

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codefresh-io/service-base",
-  "version": "7.0.0",
+  "version": "7.1.0",
   "main": "index.js",
   "description": "",
   "bin": {
@@ -35,6 +35,7 @@
     "@codefresh-io/authenticated-entity": "^2.17.1",
     "@codefresh-io/cf-monitor": "^12.0.0",
     "@codefresh-io/cf-openapi": "^0.7.20",
+    "@codefresh-io/cf-telemetry": "^1.0.2",
     "@codefresh-io/eventbus": "^2.0.0",
     "@codefresh-io/http-infra": "^1.8.15",
     "@codefresh-io/internal-service-config": "^1.0.3",
@@ -52,7 +53,7 @@
     "js-yaml": "^3.13.1",
     "lodash": "4.17.21",
     "method-override": "^3.0.0",
-    "mongodb": "6.3.0",
+    "mongodb": "^6.12.0",
     "morgan": "^1.9.1",
     "node-uuid": "^1.4.8",
     "proxyquire": "^1.8.0",
@@ -71,5 +72,6 @@
     "jest": "^29.7.0",
     "mocha": "^8.2.1",
     "mongodb-memory-server": "^9.1.6"
-  }
+  },
+  "packageManager": "[email protected]+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
 }