Mark either top-level or children modified

shashankbrgowda · Oct 31, 2023 · 2dd944a · 2dd944a
1 parent 41232ca
commit 2dd944a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/packages/apollo-shared/src/Changes/DiscontinuousLocationEndChange.ts b/packages/apollo-shared/src/Changes/DiscontinuousLocationEndChange.ts
@@ -117,7 +117,11 @@ export class DiscontinuousLocationEndChange extends FeatureChange {
       }
 
       try {
-        topLevelFeature.markModified('children')
+        if (topLevelFeature._id.equals(feature._id)) {
+          topLevelFeature.markModified('discontinuousLocations')
+        } else {
+          topLevelFeature.markModified('children')
+        }
         await topLevelFeature.save()
       } catch (error) {
         logger.debug?.(`*** FAILED: ${error}`)

diff --git a/packages/apollo-shared/src/Changes/DiscontinuousLocationStartChange.ts b/packages/apollo-shared/src/Changes/DiscontinuousLocationStartChange.ts
@@ -117,7 +117,11 @@ export class DiscontinuousLocationStartChange extends FeatureChange {
       }
 
       try {
-        topLevelFeature.markModified('children')
+        if (topLevelFeature._id.equals(feature._id)) {
+          topLevelFeature.markModified('discontinuousLocations')
+        } else {
+          topLevelFeature.markModified('children')
+        }
         await topLevelFeature.save()
       } catch (error) {
         logger.debug?.(`*** FAILED: ${error}`)