Only use transactions on replica sets

gchq · Jun 29, 2024 · 1c9cde8 · 1c9cde8
1 parent 419071a
commit 1c9cde8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 10 deletions.
diff --git a/backend/src/services/model.ts b/backend/src/services/model.ts
@@ -11,6 +11,7 @@ import ModelCardRevisionModel, { ModelCardRevisionDoc } from '../models/ModelCar
 import { UserInterface } from '../models/User.js'
 import { GetModelCardVersionOptions, GetModelCardVersionOptionsKeys, GetModelFiltersKeys } from '../types/enums.js'
 import { isValidatorResultError } from '../types/ValidatorResultError.js'
+import { isReplicaSet } from '../utils/database.js'
 import { toEntity } from '../utils/entity.js'
 import { BadReq, Forbidden, InternalError, NotFound } from '../utils/error.js'
 import { convertStringToId } from '../utils/id.js'
@@ -257,16 +258,21 @@ export async function _setModelCard(
 
   const revision = new ModelCardRevisionModel({ ...newDocument, modelId, createdBy: user.dn })
 
-  const message = 'Unable to save model card revision'
-  await mongoose.connection
-    .transaction(async function executeUpdate(session) {
-      await revision.save({ session })
-      await ModelModel.updateOne({ id: modelId }, { $set: { card: newDocument } }, { session: session })
-    })
-    .catch((error) => {
-      log.error('Error when updating model card/revision. Transaction rolled back.', error)
-      throw InternalError(message, { modelId })
-    })
+  if (await isReplicaSet()) {
+    await mongoose.connection
+      .transaction(async function executeUpdate(session) {
+        await revision.save({ session })
+        await ModelModel.updateOne({ id: modelId }, { $set: { card: newDocument } }, { session: session })
+      })
+      .catch((error) => {
+        const message = 'Unable to save model card revision'
+        log.error('Error when updating model card/revision. Transaction rolled back.', error)
+        throw InternalError(message, { modelId })
+      })
+  } else {
+    await revision.save()
+    await ModelModel.updateOne({ id: modelId }, { $set: { card: newDocument } })
+  }
 
   return revision
 }

diff --git a/backend/src/utils/database.ts b/backend/src/utils/database.ts
@@ -31,6 +31,11 @@ export async function connectToMongoose() {
   }
 }
 
+export async function isReplicaSet(): Promise<boolean> {
+  const options = mongoose.connection.getClient().options
+  return Object.prototype.hasOwnProperty.call(options, 'replicaSet') && options.replicaSet.length > 0
+}
+
 export async function disconnectFromMongoose() {
   await mongoose.disconnect()
   log.info({ log: false }, 'Disconnected from Mongoose')