hamed/minor_refactor

Author: hamed-rezaee

lib/state/connection/connection_cubit.dart

@@ -14,7 +14,7 @@ import 'package:flutter_deriv_api/services/connection/api_manager/enums.dart';
 
 part 'connection_state.dart';
 
-/// Bringing [ConnectionCubit] to flutter-deriv-api to simplify the usage of api.
+/// [ConnectionCubit] is a [Cubit] which manages connection status of websocket.
 class ConnectionCubit extends Cubit<ConnectionState> {
   /// Initializes [ConnectionCubit].
   ConnectionCubit(
@@ -23,9 +23,8 @@ class ConnectionCubit extends Cubit<ConnectionState> {
     this.enableDebug = false,
     this.printResponse = false,
   }) : super(const ConnectionInitialState()) {
-    APIInitializer().initialize(
-      api: api ?? BinaryAPI(key: _key, enableDebug: enableDebug),
-    );
+    APIInitializer()
+        .initialize(api: api ?? BinaryAPI(key: _key, enableDebug: enableDebug));
 
     _api = Injector()<BaseAPI>();
 
@@ -46,6 +45,9 @@ class ConnectionCubit extends Cubit<ConnectionState> {
   /// Default value is `false`.
   final bool printResponse;
 
+  /// Connection status of websocket stream subscription.
+  StreamSubscription<APIStatus>? _connectionStatusSubscription;
+
   static late ConnectionInformation _connectionInformation;
 
   /// Gets connection information of WebSocket (endpoint, brand, appId).
@@ -80,18 +82,35 @@ class ConnectionCubit extends Cubit<ConnectionState> {
   }
 
   void _handleConnectionStatus() {
-    _api!.connectionStatus.listen((APIStatus status) {
-      switch (status) {
-        case APIStatus.connecting:
-          emit(const ConnectionConnectingState());
-          break;
-        case APIStatus.connected:
-          emit(const ConnectionConnectedState());
-          break;
-        case APIStatus.disconnected:
-          emit(const ConnectionDisconnectedState());
-          break;
-      }
-    });
+    _cancelConnectionStatusSubscription();
+
+    _connectionStatusSubscription = _api!.connectionStatus.listen(
+      (APIStatus status) {
+        switch (status) {
+          case APIStatus.connecting:
+            emit(const ConnectionConnectingState());
+            break;
+          case APIStatus.connected:
+            emit(const ConnectionConnectedState());
+            break;
+          case APIStatus.disconnected:
+            emit(const ConnectionDisconnectedState());
+            break;
+        }
+      },
+    );
+  }
+
+  void _cancelConnectionStatusSubscription() {
+    if (_connectionStatusSubscription != null) {
+      _connectionStatusSubscription!.cancel();
+    }
+  }
+
+  @override
+  Future<void> close() {
+    _cancelConnectionStatusSubscription();
+
+    return super.close();
   }
 }