diff --git a/Sources/WalletConnectRelay/SocketConnectionHandler/AutomaticSocketConnectionHandler.swift b/Sources/WalletConnectRelay/SocketConnectionHandler/AutomaticSocketConnectionHandler.swift index 5a5cec5f6..20eab6208 100644 --- a/Sources/WalletConnectRelay/SocketConnectionHandler/AutomaticSocketConnectionHandler.swift +++ b/Sources/WalletConnectRelay/SocketConnectionHandler/AutomaticSocketConnectionHandler.swift @@ -36,7 +36,6 @@ class AutomaticSocketConnectionHandler: SocketConnectionHandler { } appStateObserver.onWillEnterForeground = { [unowned self] in - backgroundTaskRegistrar.invalidate() socket.connect() } } diff --git a/Sources/WalletConnectRelay/SocketConnectionHandler/BackgroundTaskRegistering.swift b/Sources/WalletConnectRelay/SocketConnectionHandler/BackgroundTaskRegistering.swift index 7b08c94a6..a652bef74 100644 --- a/Sources/WalletConnectRelay/SocketConnectionHandler/BackgroundTaskRegistering.swift +++ b/Sources/WalletConnectRelay/SocketConnectionHandler/BackgroundTaskRegistering.swift @@ -5,7 +5,6 @@ import UIKit protocol BackgroundTaskRegistering { func register(name: String, completion: @escaping () -> Void) - func invalidate() } class BackgroundTaskRegistrar: BackgroundTaskRegistering { @@ -15,7 +14,7 @@ class BackgroundTaskRegistrar: BackgroundTaskRegistering { func register(name: String, completion: @escaping () -> Void) { #if os(iOS) - backgroundTaskID = .invalid + invalidateIfNeeded() backgroundTaskID = UIApplication.shared.beginBackgroundTask(withName: name) { [unowned self] in UIApplication.shared.endBackgroundTask(backgroundTaskID) backgroundTaskID = .invalid @@ -24,7 +23,7 @@ class BackgroundTaskRegistrar: BackgroundTaskRegistering { #endif } - func invalidate() { + private func invalidateIfNeeded() { #if os(iOS) if backgroundTaskID != .invalid { UIApplication.shared.endBackgroundTask(backgroundTaskID)