Include self when notifying disconnected

dsmurfin · Nov 7, 2022 · a4a04bf · a4a04bf
1 parent 03ca767
commit a4a04bf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/Sources/OTPKit/Components/Consumer.swift b/Sources/OTPKit/Components/Consumer.swift
@@ -1609,7 +1609,7 @@ extension OTPConsumer: ComponentSocketDelegate {
     func socketDidClose(_ socket: ComponentSocket, withError error: Error?) {
         if self._isConnected != false {
             self._isConnected = false
-            delegateQueue.async { self.consumerDelegate?.disconnected(with: error) }
+            delegateQueue.async { self.consumerDelegate?.disconnected(self, with: error) }
         }
     }
 
@@ -1678,10 +1678,11 @@ public protocol OTPConsumerDelegate: AnyObject {
      Notifies the delegate that the consumer was disconnected
 
      - Parameters:
+        - consumer: The consumer which was disconnected.
         - error: An optional error that may have occured, for example if the network interface was removed.
 
     */
-    func disconnected(with error: Error?)
+    func disconnected(_ consumer: OTPConsumer, with error: Error?)
 
 }
 

diff --git a/Sources/OTPKit/Components/Producer.swift b/Sources/OTPKit/Components/Producer.swift
@@ -1885,7 +1885,7 @@ extension OTPProducer: ComponentSocketDelegate {
     func socketDidClose(_ socket: ComponentSocket, withError error: Error?) {
         if self._isConnected != false {
             self._isConnected = false
-            delegateQueue.async { self.producerDelegate?.disconnected(with: error) }
+            delegateQueue.async { self.producerDelegate?.disconnected(self, with: error) }
         }
     }
 
@@ -1927,10 +1927,11 @@ public protocol OTPProducerDelegate: AnyObject {
      Notifies the delegate that the producer was disconnected
 
      - Parameters:
+        - producer: The producer which was disconnected.
         - error: An optional error that may have occured, for example if the network interface was removed.
 
     */
-    func disconnected(with error: Error?)
+    func disconnected(_ producer: OTPProducer, with error: Error?)
 
 }