Merge pull request #481 from nova-wallet/fix/connection-pool-crash

Fix race codition in connection pool
novasamatech · Nov 23, 2022 · dad9e69 · dad9e69
2 parents fe6ce29 + 022d2d5
commit dad9e69
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/novawallet/Common/Services/ChainRegistry/ConnectionPool/ConnectionPool.swift b/novawallet/Common/Services/ChainRegistry/ConnectionPool/ConnectionPool.swift
@@ -135,6 +135,12 @@ extension ConnectionPool: WebSocketEngineDelegate {
 
 extension ConnectionPool: ApplicationHandlerDelegate {
     func didReceiveDidBecomeActive(notification _: Notification) {
+        mutex.lock()
+
+        defer {
+            mutex.unlock()
+        }
+
         connections.values.forEach { wrapper in
             guard let connection = wrapper.target as? ChainConnection else {
                 return
@@ -145,6 +151,12 @@ extension ConnectionPool: ApplicationHandlerDelegate {
     }
 
     func didReceiveDidEnterBackground(notification _: Notification) {
+        mutex.lock()
+
+        defer {
+            mutex.unlock()
+        }
+
         connections.values.forEach { wrapper in
             guard let connection = wrapper.target as? ChainConnection else {
                 return