chore: remove asyncLock

closes #928

server/models/Collection.js

@@ -3,7 +3,6 @@ import { find, remove } from 'lodash';
 import slug from 'slug';
 import randomstring from 'randomstring';
 import { DataTypes, sequelize } from '../sequelize';
-import { asyncLock } from '../redis';
 import Document from './Document';
 import CollectionUser from './CollectionUser';
 import { welcomeMessage } from '../utils/onboarding';
@@ -143,53 +142,65 @@ Collection.prototype.addDocumentToStructure = async function(
 ) {
   if (!this.documentStructure) return;
 
-  let unlock;
+  let transaction;
 
-  // documentStructure can only be updated by one request at a time
-  if (options.save !== false) {
-    unlock = await asyncLock(`collection-${this.id}`);
-  }
+  try {
+    // documentStructure can only be updated by one request at a time
+    if (options.save !== false) {
+      transaction = await sequelize.transaction();
+    }
 
-  // If moving existing document with children, use existing structure
-  const documentJson = {
-    ...document.toJSON(),
-    ...options.documentJson,
-  };
-
-  if (!document.parentDocumentId) {
-    // Note: Index is supported on DB level but it's being ignored
-    // by the API presentation until we build product support for it.
-    this.documentStructure.splice(
-      index !== undefined ? index : this.documentStructure.length,
-      0,
-      documentJson
-    );
-  } else {
-    // Recursively place document
-    const placeDocument = documentList => {
-      return documentList.map(childDocument => {
-        if (document.parentDocumentId === childDocument.id) {
-          childDocument.children.splice(
-            index !== undefined ? index : childDocument.children.length,
-            0,
-            documentJson
-          );
-        } else {
-          childDocument.children = placeDocument(childDocument.children);
-        }
-
-        return childDocument;
-      });
+    // If moving existing document with children, use existing structure
+    const documentJson = {
+      ...document.toJSON(),
+      ...options.documentJson,
     };
-    this.documentStructure = placeDocument(this.documentStructure);
-  }
 
-  // Sequelize doesn't seem to set the value with splice on JSONB field
-  this.documentStructure = this.documentStructure;
+    if (!document.parentDocumentId) {
+      // Note: Index is supported on DB level but it's being ignored
+      // by the API presentation until we build product support for it.
+      this.documentStructure.splice(
+        index !== undefined ? index : this.documentStructure.length,
+        0,
+        documentJson
+      );
+    } else {
+      // Recursively place document
+      const placeDocument = documentList => {
+        return documentList.map(childDocument => {
+          if (document.parentDocumentId === childDocument.id) {
+            childDocument.children.splice(
+              index !== undefined ? index : childDocument.children.length,
+              0,
+              documentJson
+            );
+          } else {
+            childDocument.children = placeDocument(childDocument.children);
+          }
 
-  if (options.save !== false) {
-    await this.save(options);
-    if (unlock) unlock();
+          return childDocument;
+        });
+      };
+      this.documentStructure = placeDocument(this.documentStructure);
+    }
+
+    // Sequelize doesn't seem to set the value with splice on JSONB field
+    this.documentStructure = this.documentStructure;
+
+    if (options.save !== false) {
+      await this.save({
+        ...options,
+        transaction,
+      });
+      if (transaction) {
+        await transaction.commit();
+      }
+    }
+  } catch (err) {
+    if (transaction) {
+      await transaction.rollback();
+    }
+    throw err;
   }
 
   return this;
@@ -203,28 +214,39 @@ Collection.prototype.updateDocument = async function(
 ) {
   if (!this.documentStructure) return;
 
-  // documentStructure can only be updated by one request at the time
-  const unlock = await asyncLock(`collection-${this.id}`);
+  let transaction;
 
-  const { id } = updatedDocument;
+  try {
+    // documentStructure can only be updated by one request at the time
+    transaction = await sequelize.transaction();
 
-  const updateChildren = documents => {
-    return documents.map(document => {
-      if (document.id === id) {
-        document = {
-          ...updatedDocument.toJSON(),
-          children: document.children,
-        };
-      } else {
-        document.children = updateChildren(document.children);
-      }
-      return document;
-    });
-  };
+    const { id } = updatedDocument;
+
+    const updateChildren = documents => {
+      return documents.map(document => {
+        if (document.id === id) {
+          document = {
+            ...updatedDocument.toJSON(),
+            children: document.children,
+          };
+        } else {
+          document.children = updateChildren(document.children);
+        }
+        return document;
+      });
+    };
+
+    this.documentStructure = updateChildren(this.documentStructure);
+    await this.save({ transaction });
+    await transaction.commit();
+
+  } catch (err) {
+    if (transaction) {
+      await transaction.rollback();
+    }
+    throw err;
+  }
 
-  this.documentStructure = updateChildren(this.documentStructure);
-  await this.save();
-  unlock();
   return this;
 };
 
@@ -239,37 +261,48 @@ Collection.prototype.removeDocumentInStructure = async function(
 ) {
   if (!this.documentStructure) return;
   let returnValue;
-  let unlock;
+  let transaction;
 
-  // documentStructure can only be updated by one request at the time
-  unlock = await asyncLock(`collection-${this.id}`);
+  try {
+    // documentStructure can only be updated by one request at the time
+    transaction = await sequelize.transaction();
 
-  const removeFromChildren = async (children, id) => {
-    children = await Promise.all(
-      children.map(async childDocument => {
-        return {
-          ...childDocument,
-          children: await removeFromChildren(childDocument.children, id),
-        };
-      })
+    const removeFromChildren = async (children, id) => {
+      children = await Promise.all(
+        children.map(async childDocument => {
+          return {
+            ...childDocument,
+            children: await removeFromChildren(childDocument.children, id),
+          };
+        })
+      );
+
+      const match = find(children, { id });
+      if (match) {
+        if (!returnValue) returnValue = match;
+        remove(children, { id });
+      }
+
+      return children;
+    };
+
+    this.documentStructure = await removeFromChildren(
+      this.documentStructure,
+      document.id
     );
 
-    const match = find(children, { id });
-    if (match) {
-      if (!returnValue) returnValue = match;
-      remove(children, { id });
+    await this.save({
+      ...options,
+      transaction,
+    });
+    await transaction.commit();
+
+  } catch (err) {
+    if (transaction) {
+      await transaction.rollback();
     }
-
-    return children;
-  };
-
-  this.documentStructure = await removeFromChildren(
-    this.documentStructure,
-    document.id
-  );
-
-  await this.save(options);
-  if (unlock) await unlock();
+    throw err;
+  }
 
   return returnValue;
 };