From 085e6aa80d2bbdfcbdd94b8ec7ac4c6078d0fc4c Mon Sep 17 00:00:00 2001
From: Fred Boniface <fred@fjla.uk>
Date: Wed, 31 May 2023 20:30:16 +0100
Subject: [PATCH] Add option to retreive _id from DB

Signed-off-by: Fred Boniface <fred@fjla.uk>
---
 src/services/dbAccess.services.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/services/dbAccess.services.js b/src/services/dbAccess.services.js
index 0818324..af9e060 100644
--- a/src/services/dbAccess.services.js
+++ b/src/services/dbAccess.services.js
@@ -13,12 +13,14 @@ const { MongoClient } = require('mongodb');
 const client = new MongoClient(uri);
 const db = client.db(dbName);
 
-async function query(collection, query){
+async function query(collection, query, returnId = false){
   await client.connect();
   log.out(`dbAccess.query: Connecting to collection: '${collection}'`, 'info');
   var qcoll = db.collection(collection);
   var qcursor = qcoll.find(query);
-  qcursor.project({_id: 0});
+  if (!returnId) {
+    qcursor.project({_id: 0});
+  }
   log.out(`dbAccess.query: Running Query: ${JSON.stringify(query)}`, 'info');
   increment(collection);
   return (await qcursor.toArray());