fix(datastore): Clear subscriptions on Stop

aws-amplify · Aug 8, 2024 · 1dc7e6f · 1dc7e6f
1 parent ff0e634
commit 1dc7e6f
Show file tree

Hide file tree

Showing 6 changed files with 45 additions and 0 deletions.
diff --git a/...d/src/main/kotlin/com/amazonaws/amplify/amplify_datastore/pigeons/NativePluginBindings.kt b/...d/src/main/kotlin/com/amazonaws/amplify/amplify_datastore/pigeons/NativePluginBindings.kt
diff --git a/packages/amplify_datastore/ios/Classes/SwiftAmplifyDataStorePlugin.swift b/packages/amplify_datastore/ios/Classes/SwiftAmplifyDataStorePlugin.swift
@@ -612,6 +612,9 @@ public class SwiftAmplifyDataStorePlugin: NSObject, FlutterPlugin, NativeAmplify
                         error: error,
                         flutterResult: flutterResult)
                 case .success():
+                    DispatchQueue.main.async {
+                        self.nativeApiPlugin.onStop {}
+                    }
                     print("Successfully stopped datastore cloud syncing")
                     flutterResult(nil)
                 }

diff --git a/packages/amplify_datastore/ios/Classes/pigeons/NativePluginBindings.swift b/packages/amplify_datastore/ios/Classes/pigeons/NativePluginBindings.swift
diff --git a/packages/amplify_datastore/lib/amplify_datastore.dart b/packages/amplify_datastore/lib/amplify_datastore.dart
@@ -375,4 +375,14 @@ class NativeAmplifyApi
       _subscriptionsCache.remove(subscriptionId);
     }
   }
+
+
+  // Amplify.DataStore.Stop() callback
+  // Clean up subscriptions on stop.
+  @override
+  Future<void> onStop() async {
+    _subscriptionsCache.forEach((subId, _) async {
+      await unsubscribe(subId);
+    });
+  }
 }
diff --git a/packages/amplify_datastore/lib/src/native_plugin.g.dart b/packages/amplify_datastore/lib/src/native_plugin.g.dart
diff --git a/packages/amplify_datastore/pigeons/native_plugin.dart b/packages/amplify_datastore/pigeons/native_plugin.dart
@@ -41,6 +41,10 @@ abstract class NativeApiPlugin {
 
   @async
   void unsubscribe(String subscriptionId);
+
+
+  @async
+  void onStop();
 }
 
 /// Bridge for calling Amplify from Flutter into Native
-Original file line number
+Diff line change
@@ Expand Up / @@ -41,6 +41,10 @@ abstract class NativeApiPlugin { @@
       @async
       void unsubscribe(String subscriptionId);
+      @async
+      void onStop();
     }
     /// Bridge for calling Amplify from Flutter into Native
@@ Expand Down @@