Keep a singleton of connectivity monitor in GRPCHost

tmatsuo · Oct 22, 2016 · ad93106 · ad93106
1 parent 6ac6a52
commit ad93106
Showing 1 changed file with 10 additions and 15 deletions.
diff --git a/src/objective-c/GRPCClient/private/GRPCHost.m b/src/objective-c/GRPCClient/private/GRPCHost.m
@@ -100,19 +100,16 @@ - (nullable instancetype)initWithAddress:(NSString *)address {
       _address = address;
       _secure = YES;
       kHostCache[address] = self;
-
-      // When there is host in the cache, keep a single monitor to the network
-      // to
-      // flush the cache if the connectivity status changed
-      if (!connectivityMonitor) {
-        connectivityMonitor =
-            [GRPCConnectivityMonitor monitorWithHost:hostURL.host];
-        void (^handler)() = ^{
-          [GRPCHost flushChannelCache];
-        };
-        [connectivityMonitor handleLossWithHandler:handler
-                           wifiStatusChangeHandler:handler];
-      }
+    }
+    // Keep a single monitor to flush the cache if the connectivity status changed
+    if (!connectivityMonitor) {
+      connectivityMonitor =
+      [GRPCConnectivityMonitor monitorWithHost:hostURL.host];
+      void (^handler)() = ^{
+        [GRPCHost flushChannelCache];
+      };
+      [connectivityMonitor handleLossWithHandler:handler
+                         wifiStatusChangeHandler:handler];
     }
   }
   return self;
@@ -125,8 +122,6 @@ + (void)flushChannelCache {
                                                     BOOL * _Nonnull stop) {
       [host disconnect];
     }];
-    [kHostCache removeAllObjects];
-    connectivityMonitor = nil;
   }
 }