fix for realtime multiple subscription

Author: lohanidamodar

templates/flutter/lib/src/realtime_mixin.dart.twig

@@ -2,7 +2,7 @@ import 'dart:async';
 import 'dart:convert';
 import 'package:flutter/foundation.dart';
 import 'package:web_socket_channel/web_socket_channel.dart';
-import 'package:web_socket_channel/status.dart';
+import 'package:web_socket_channel/status.dart' as status;
 import 'exception.dart';
 import 'realtime_subscription.dart';
 import 'client.dart';
@@ -15,15 +15,20 @@ typedef GetFallbackCookie = String? Function();
 
 mixin RealtimeMixin {
   late Client client;
-  final Map<String, List<StreamController<RealtimeMessage>>> _channels = {};
+  final Set<String> _channels = {};
   WebSocketChannel? _websok;
   String? _lastUrl;
   late WebSocketFactory getWebSocket;
   GetFallbackCookie? getFallbackCookie;
   int? get closeCode => _websok?.closeCode;
+  int _subscriptionsCounter = 0;
+  Map<int, RealtimeSubscription> _subscriptions = {};
+  bool _notifyDone = true;
+  StreamSubscription? _websocketSubscription;
 
   Future<dynamic> _closeConnection() async {
-    await _websok?.sink.close(normalClosure);
+    await _websocketSubscription?.cancel();
+    await _websok?.sink.close(status.normalClosure, 'Ending session');
     _lastUrl = null;
   }
 
@@ -36,14 +41,16 @@ mixin RealtimeMixin {
       if (_lastUrl == uri.toString() && _websok?.closeCode == null) {
         return;
       }
+      _notifyDone = false;
       await _closeConnection();
       _lastUrl = uri.toString();
       _websok = await getWebSocket(uri);
+      _notifyDone = true;
     }
     debugPrint('subscription: $_lastUrl');
 
     try {
-      _websok?.stream.listen((response) {
+      _websocketSubscription = _websok?.stream.listen((response) {
         final data = RealtimeResponse.fromJson(response);
         switch (data.type) {
           case 'error':
@@ -67,48 +74,45 @@ mixin RealtimeMixin {
             break;
           case 'event':
             final message = RealtimeMessage.fromMap(data.data);
-            for(var channel in message.channels) {
-              if (_channels[channel] != null) {
-                for( var stream in _channels[channel]!) {
-                  stream.sink.add(message);
+            for (var subscription in _subscriptions.values) {
+              for (var channel in message.channels) {
+                if (subscription.channels.contains(channel)) {
+                  subscription.controller.add(message);
                 }
               }
             }
             break;
         }
       }, onDone: () {
-        for (var list in _channels.values) {
-          for (var stream in list) {
-            stream.close();
-          }
+        if (!_notifyDone) return;
+        for (var subscription in _subscriptions.values) {
+          subscription.close();
         }
         _channels.clear();
         _closeConnection();
       }, onError: (err, stack) {
-        for (var list in _channels.values) {
-          for (var stream in list) {
-            stream.sink.addError(err, stack);
-          }
+        for (var subscription in _subscriptions.values) {
+          subscription.controller.addError(err, stack);
         }
         if (_websok?.closeCode != null && _websok?.closeCode != 1008) {
           debugPrint("Reconnecting in one second.");
           Future.delayed(Duration(seconds: 1), _createSocket);
         }
       });
     } catch (e) {
-      if (e is {{spec.title | caseUcfirst}}Exception) {
+      if (e is AppwriteException) {
         rethrow;
       }
       if (e is WebSocketChannelException) {
-        throw {{spec.title | caseUcfirst}}Exception(e.message);
+        throw AppwriteException(e.message);
       }
-      throw {{spec.title | caseUcfirst}}Exception(e.toString());
+      throw AppwriteException(e.toString());
     }
   }
 
   Uri _prepareUri() {
     if (client.endPointRealtime == null) {
-      throw {{spec.title | caseUcfirst}}Exception(
+      throw AppwriteException(
           "Please set endPointRealtime to connect to realtime server");
     }
     var uri = Uri.parse(client.endPointRealtime!);
@@ -118,43 +122,49 @@ mixin RealtimeMixin {
       port: uri.port,
       queryParameters: {
         "project": client.config['project'],
-        "channels[]": _channels.keys.toList(),
+        "channels[]": _channels.toList(),
       },
       path: uri.path + "/realtime",
     );
   }
 
   RealtimeSubscription subscribeTo(List<String> channels) {
     StreamController<RealtimeMessage> controller = StreamController.broadcast();
-    for(var channel in channels) {
-      if (!_channels.containsKey(channel)) {
-        _channels[channel] = [];
-      }
-      _channels[channel]!.add(controller);
-    }
+    _channels.addAll(channels);
     Future.delayed(Duration.zero, () => _createSocket());
+    int counter = _subscriptionsCounter++;
     RealtimeSubscription subscription = RealtimeSubscription(
-        stream: controller.stream,
+        controller: controller,
+        channels: channels,
         close: () async {
+          _subscriptions.remove(counter);
+          _subscriptionsCounter--;
           controller.close();
-          for(var channel in channels) {
-            _channels[channel]!.remove(controller);
-            if (_channels[channel]!.isEmpty) {
-              _channels.remove(channel);
-            }
-          }
-          if(_channels.isNotEmpty) {
+          _cleanup(channels);
+
+          if (_channels.isNotEmpty) {
             await Future.delayed(Duration.zero, () => _createSocket());
           } else {
             await _closeConnection();
           }
         });
+    _subscriptions[counter] = subscription;
     return subscription;
   }
 
+  void _cleanup(List<String> channels) {
+    for (var channel in channels) {
+      bool found = _subscriptions.values
+          .any((subscription) => subscription.channels.contains(channel));
+      if (!found) {
+        _channels.remove(channel);
+      }
+    }
+  }
+
   void handleError(RealtimeResponse response) {
     if (response.data['code'] == 1008) {
-      throw {{spec.title | caseUcfirst}}Exception(response.data["message"], response.data["code"]);
+      throw AppwriteException(response.data["message"], response.data["code"]);
     } else {
       debugPrint("Reconnecting in one second.");
       Future.delayed(const Duration(seconds: 1), () {

templates/flutter/lib/src/realtime_subscription.dart.twig

@@ -1,13 +1,22 @@
+import 'dart:async';
+
 import 'realtime_message.dart';
 
 /// Realtime Subscription
 class RealtimeSubscription {
   /// Stream of [RealtimeMessage]s
   final Stream<RealtimeMessage> stream;
 
+  final StreamController<RealtimeMessage> controller;
+
+  /// List of channels
+  List<String> channels;
+
   /// Closes the subscription
   final Future<void> Function() close;
 
   /// Initializes a [RealtimeSubscription]
-  RealtimeSubscription({required this.stream, required this.close});
+  RealtimeSubscription(
+      {required this.close, required this.channels, required this.controller})
+      : stream = controller.stream;
 }

templates/flutter/test/src/realtime_subscription_test.dart.twig

@@ -1,20 +1,20 @@
-import 'package:mockito/mockito.dart';
-import 'package:{{language.params.packageName}}/src/realtime_message.dart';
-import 'package:{{language.params.packageName}}/src/realtime_subscription.dart';
+import 'package:appwrite/src/realtime_message.dart';
+import 'package:appwrite/src/realtime_subscription.dart';
 import 'package:flutter_test/flutter_test.dart';
-
-class MockStream<T> extends Mock implements Stream<T> {}
-
-
+import 'dart:async';
 
 void main() {
   group('RealtimeSubscription', () {
-    final mockStream = MockStream<RealtimeMessage>();
+    final mockStream = StreamController<RealtimeMessage>.broadcast();
     final mockCloseFunction = () async {};
-    final subscription = RealtimeSubscription(stream: mockStream, close: mockCloseFunction);
+    final subscription = RealtimeSubscription(
+        controller: mockStream,
+        close: mockCloseFunction,
+        channels: ['documents']);
 
     test('should have the correct stream and close function', () {
-      expect(subscription.stream, equals(mockStream));
+      expect(subscription.controller, equals(mockStream));
+      expect(subscription.stream, equals(mockStream.stream));
       expect(subscription.close, equals(mockCloseFunction));
     });
   });