Allow attaching completion handler for updating context. (#81)

Unleash · Feb 1, 2024 · ebf32e6 · ebf32e6
1 parent f1ae08f
commit ebf32e6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Sources/UnleashProxyClientSwift/UnleashProxyClientSwift.swift b/Sources/UnleashProxyClientSwift/UnleashProxyClientSwift.swift
@@ -101,10 +101,9 @@ public class UnleashClientBase {
         SwiftEventBus.unregister(self, name: name)
     }
 
-    public func updateContext(context: [String: String], properties: [String:String]? = nil) -> Void {
+    public func updateContext(context: [String: String], properties: [String:String]? = nil, completionHandler: ((PollerError?) -> Void)? = nil) {
         self.context = self.calculateContext(context: context, properties: properties)
-        self.stop()
-        self.start()
+        self.start(completionHandler: completionHandler)
     }
 
     func calculateContext(context: [String: String], properties: [String:String]? = nil) -> Context {