Update RxRelay multi-relay bind

RxRelay/Observable+Bind.swift

@@ -10,19 +10,43 @@ import RxSwift
 
 extension ObservableType {
     /**
-     Creates new subscription and sends elements to publish relay.
+     Creates new subscription and sends elements to publish relay(s).
+     In case error occurs in debug mode, `fatalError` will be raised.
+     In case error occurs in release mode, `error` will be logged.
+     - parameter to: Target publish relays for sequence elements.
+     - returns: Disposable object that can be used to unsubscribe the observer.
+     */
+    public func bind(to relays: PublishRelay<E>...) -> Disposable {
+        return bind(to: relays)
+    }
+
+    /**
+     Creates new subscription and sends elements to publish relay(s).
 
      In case error occurs in debug mode, `fatalError` will be raised.
      In case error occurs in release mode, `error` will be logged.
 
-     - parameter to: Target publish relay for sequence elements.
+     - parameter to: Target publish relays for sequence elements.
+     - returns: Disposable object that can be used to unsubscribe the observer.
+     */
+    public func bind(to relays: PublishRelay<E?>...) -> Disposable {
+        return self.map { $0 as E? }.bind(to: relays)
+    }
+
+    /**
+     Creates new subscription and sends elements to publish relay(s).
+     In case error occurs in debug mode, `fatalError` will be raised.
+     In case error occurs in release mode, `error` will be logged.
+     - parameter to: Target publish relays for sequence elements.
      - returns: Disposable object that can be used to unsubscribe the observer.
      */
-    public func bind(to relay: PublishRelay<E>) -> Disposable {
-        return self.subscribe { e in
+    private func bind(to relays: [PublishRelay<E>]) -> Disposable {
+        return subscribe { e in
             switch e {
             case let .next(element):
-                relay.accept(element)
+                relays.forEach {
+                    $0.accept(element)
+                }
             case let .error(error):
                 rxFatalErrorInDebug("Binding error to publish relay: \(error)")
             case .completed:
@@ -32,50 +56,48 @@ extension ObservableType {
     }
 
     /**
-     Creates new subscription and sends elements to publish relay.
-
+     Creates new subscription and sends elements to behavior relay(s).
      In case error occurs in debug mode, `fatalError` will be raised.
      In case error occurs in release mode, `error` will be logged.
-
-     - parameter to: Target publish relay for sequence elements.
+     - parameter to: Target behavior relay for sequence elements.
      - returns: Disposable object that can be used to unsubscribe the observer.
      */
-    public func bind(to relay: PublishRelay<E?>) -> Disposable {
-        return self.map { $0 as E? }.bind(to: relay)
+    public func bind(to relays: BehaviorRelay<E>...) -> Disposable {
+        return self.bind(to: relays)
     }
 
     /**
-     Creates new subscription and sends elements to behavior relay.
+     Creates new subscription and sends elements to behavior relay(s).
 
      In case error occurs in debug mode, `fatalError` will be raised.
      In case error occurs in release mode, `error` will be logged.
 
      - parameter to: Target behavior relay for sequence elements.
      - returns: Disposable object that can be used to unsubscribe the observer.
      */
-    public func bind(to relay: BehaviorRelay<E>) -> Disposable {
-        return self.subscribe { e in
-            switch e {
-            case let .next(element):
-                relay.accept(element)
-            case let .error(error):
-                rxFatalErrorInDebug("Binding error to behavior relay: \(error)")
-            case .completed:
-                break
-            }
-        }
+    public func bind(to relays: BehaviorRelay<E?>...) -> Disposable {
+        return self.map { $0 as E? }.bind(to: relays)
     }
 
     /**
-     Creates new subscription and sends elements to behavior relay.
-
+     Creates new subscription and sends elements to behavior relay(s).
      In case error occurs in debug mode, `fatalError` will be raised.
      In case error occurs in release mode, `error` will be logged.
-
      - parameter to: Target behavior relay for sequence elements.
      - returns: Disposable object that can be used to unsubscribe the observer.
      */
-    public func bind(to relay: BehaviorRelay<E?>) -> Disposable {
-        return self.map { $0 as E? }.bind(to: relay)
+    private func bind(to relays: [BehaviorRelay<E>]) -> Disposable {
+        return subscribe { e in
+            switch e {
+            case let .next(element):
+                relays.forEach {
+                    $0.accept(element)
+                }
+            case let .error(error):
+                rxFatalErrorInDebug("Binding error to behavior relay: \(error)")
+            case .completed:
+                break
+            }
+        }
     }
 }