acquire mutex in proofreading and release after saving

Author: philippotto

frontend/javascripts/viewer/model/sagas/volume/proofread_saga.ts

@@ -1,9 +1,11 @@
 import {
+  acquireAnnotationMutex,
   type NeighborInfo,
   getAgglomeratesForSegmentsFromTracingstore,
   getEdgesForAgglomerateMinCut,
   getNeighborsForAgglomerateNode,
   getPositionForSegmentInAgglomerate,
+  releaseAnnotationMutex,
 } from "admin/rest_api";
 import { V3 } from "libs/mjs";
 import Toast from "libs/toast";
@@ -358,6 +360,7 @@ function* handleSkeletonProofreadingAction(action: Action): Saga<void> {
   const { agglomerateFileMag, getDataValue, activeMapping, volumeTracing } = preparation;
   const { tracingId: volumeTracingId } = volumeTracing;
 
+  const annotationId = yield* select((state) => state.annotation.annotationId);
   // Use untransformedPosition because agglomerate trees should not have
   // any transforms, anyway.
   if (yield* select((state) => areGeometriesTransformed(state))) {
@@ -441,6 +444,7 @@ function* handleSkeletonProofreadingAction(action: Action): Saga<void> {
 
   yield* put(pushSaveQueueTransaction(items));
   yield* call([Model, Model.ensureSavedState]);
+  yield* call(releaseAnnotationMutex, annotationId);
 
   if (action.type === "MIN_CUT_AGGLOMERATE_WITH_NODE_IDS" || action.type === "DELETE_EDGE") {
     if (sourceAgglomerateId !== targetAgglomerateId) {
@@ -704,6 +708,7 @@ function* handleProofreadMergeOrMinCut(action: Action) {
   const allowUpdate = yield* select((state) => state.annotation.restrictions.allowUpdate);
   if (!allowUpdate) return;
 
+  const annotationId = yield* select((state) => state.annotation.annotationId);
   const preparation = yield* call(prepareSplitOrMerge, false);
   if (!preparation) {
     return;
@@ -789,6 +794,7 @@ function* handleProofreadMergeOrMinCut(action: Action) {
 
   yield* put(pushSaveQueueTransaction(items));
   yield* call([Model, Model.ensureSavedState]);
+  yield* call(releaseAnnotationMutex, annotationId);
 
   if (action.type === "MIN_CUT_AGGLOMERATE") {
     console.log("start updating the mapping after a min-cut");
@@ -893,6 +899,7 @@ function* handleProofreadCutFromNeighbors(action: Action) {
   const allowUpdate = yield* select((state) => state.annotation.restrictions.allowUpdate);
   if (!allowUpdate) return;
 
+  const annotationId = yield* select((state) => state.annotation.annotationId);
   const preparation = yield* call(prepareSplitOrMerge, false);
   if (!preparation) {
     return;
@@ -947,6 +954,7 @@ function* handleProofreadCutFromNeighbors(action: Action) {
 
   yield* put(pushSaveQueueTransaction(items));
   yield* call([Model, Model.ensureSavedState]);
+  yield* call(releaseAnnotationMutex, annotationId);
 
   // Now that the changes are saved, we can split the mapping locally (because it requires
   // communication with the back-end).
@@ -1130,6 +1138,13 @@ function* prepareSplitOrMerge(isSkeletonProofreading: boolean): Saga<Preparation
     return null;
   }
 
+  const annotationId = yield* select((state) => state.annotation.annotationId);
+  const { canEdit } = yield* call(acquireAnnotationMutex, annotationId);
+  if (!canEdit) {
+    Toast.error("Could not acquire mutex. Somebody else is proofreading at the moment.");
+    return null;
+  }
+
   return {
     agglomerateFileMag,
     getDataValue,