This commit is contained in:
DDIsFriend
2023-08-18 17:28:57 +08:00
commit f0e8a1709d
4282 changed files with 192396 additions and 0 deletions

69
Pods/RxSwift/RxSwift/AnyObserver.swift generated Normal file
View File

@@ -0,0 +1,69 @@
//
// AnyObserver.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/28/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// A type-erased `ObserverType`.
///
/// Forwards operations to an arbitrary underlying observer with the same `Element` type, hiding the specifics of the underlying observer type.
public struct AnyObserver<Element> : ObserverType {
/// Anonymous event handler type.
public typealias EventHandler = (Event<Element>) -> Void
private let observer: EventHandler
/// Construct an instance whose `on(event)` calls `eventHandler(event)`
///
/// - parameter eventHandler: Event handler that observes sequences events.
public init(eventHandler: @escaping EventHandler) {
self.observer = eventHandler
}
/// Construct an instance whose `on(event)` calls `observer.on(event)`
///
/// - parameter observer: Observer that receives sequence events.
public init<Observer: ObserverType>(_ observer: Observer) where Observer.Element == Element {
self.observer = observer.on
}
/// Send `event` to this observer.
///
/// - parameter event: Event instance.
public func on(_ event: Event<Element>) {
self.observer(event)
}
/// Erases type of observer and returns canonical observer.
///
/// - returns: type erased observer.
public func asObserver() -> AnyObserver<Element> {
self
}
}
extension AnyObserver {
/// Collection of `AnyObserver`s
typealias s = Bag<(Event<Element>) -> Void>
}
extension ObserverType {
/// Erases type of observer and returns canonical observer.
///
/// - returns: type erased observer.
public func asObserver() -> AnyObserver<Element> {
AnyObserver(self)
}
/// Transforms observer of type R to type E using custom transform method.
/// Each event sent to result observer is transformed and sent to `self`.
///
/// - returns: observer that transforms events.
public func mapObserver<Result>(_ transform: @escaping (Result) throws -> Element) -> AnyObserver<Result> {
AnyObserver { e in
self.on(e.map(transform))
}
}
}

59
Pods/RxSwift/RxSwift/Binder.swift generated Normal file
View File

@@ -0,0 +1,59 @@
//
// Binder.swift
// RxSwift
//
// Created by Krunoslav Zaher on 9/17/17.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
/**
Observer that enforces interface binding rules:
* can't bind errors (in debug builds binding of errors causes `fatalError` in release builds errors are being logged)
* ensures binding is performed on a specific scheduler
`Binder` doesn't retain target and in case target is released, element isn't bound.
By default it binds elements on main scheduler.
*/
public struct Binder<Value>: ObserverType {
public typealias Element = Value
private let binding: (Event<Value>) -> Void
/// Initializes `Binder`
///
/// - parameter target: Target object.
/// - parameter scheduler: Scheduler used to bind the events.
/// - parameter binding: Binding logic.
public init<Target: AnyObject>(_ target: Target, scheduler: ImmediateSchedulerType = MainScheduler(), binding: @escaping (Target, Value) -> Void) {
weak var weakTarget = target
self.binding = { event in
switch event {
case .next(let element):
_ = scheduler.schedule(element) { element in
if let target = weakTarget {
binding(target, element)
}
return Disposables.create()
}
case .error(let error):
rxFatalErrorInDebug("Binding error: \(error)")
case .completed:
break
}
}
}
/// Binds next element to owner view as described in `binding`.
public func on(_ event: Event<Value>) {
self.binding(event)
}
/// Erases type of observer.
///
/// - returns: type erased observer.
public func asObserver() -> AnyObserver<Value> {
AnyObserver(eventHandler: self.on)
}
}

13
Pods/RxSwift/RxSwift/Cancelable.swift generated Normal file
View File

@@ -0,0 +1,13 @@
//
// Cancelable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/12/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents disposable resource with state tracking.
public protocol Cancelable : Disposable {
/// Was resource disposed.
var isDisposed: Bool { get }
}

View File

@@ -0,0 +1,100 @@
//
// AsyncLock.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/21/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/**
In case nobody holds this lock, the work will be queued and executed immediately
on thread that is requesting lock.
In case there is somebody currently holding that lock, action will be enqueued.
When owned of the lock finishes with it's processing, it will also execute
and pending work.
That means that enqueued work could possibly be executed later on a different thread.
*/
final class AsyncLock<I: InvocableType>
: Disposable
, Lock
, SynchronizedDisposeType {
typealias Action = () -> Void
private var _lock = SpinLock()
private var queue: Queue<I> = Queue(capacity: 0)
private var isExecuting: Bool = false
private var hasFaulted: Bool = false
// lock {
func lock() {
self._lock.lock()
}
func unlock() {
self._lock.unlock()
}
// }
private func enqueue(_ action: I) -> I? {
self.lock(); defer { self.unlock() }
if self.hasFaulted {
return nil
}
if self.isExecuting {
self.queue.enqueue(action)
return nil
}
self.isExecuting = true
return action
}
private func dequeue() -> I? {
self.lock(); defer { self.unlock() }
if !self.queue.isEmpty {
return self.queue.dequeue()
}
else {
self.isExecuting = false
return nil
}
}
func invoke(_ action: I) {
let firstEnqueuedAction = self.enqueue(action)
if let firstEnqueuedAction = firstEnqueuedAction {
firstEnqueuedAction.invoke()
}
else {
// action is enqueued, it's somebody else's concern now
return
}
while true {
let nextAction = self.dequeue()
if let nextAction = nextAction {
nextAction.invoke()
}
else {
return
}
}
}
func dispose() {
self.synchronizedDispose()
}
func synchronized_dispose() {
self.queue = Queue(capacity: 0)
self.hasFaulted = true
}
}

View File

@@ -0,0 +1,23 @@
//
// Lock.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/31/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
protocol Lock {
func lock()
func unlock()
}
// https://lists.swift.org/pipermail/swift-dev/Week-of-Mon-20151214/000321.html
typealias SpinLock = RecursiveLock
extension RecursiveLock : Lock {
@inline(__always)
final func performLocked<T>(_ action: () -> T) -> T {
self.lock(); defer { self.unlock() }
return action()
}
}

View File

@@ -0,0 +1,16 @@
//
// LockOwnerType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 10/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
protocol LockOwnerType: AnyObject, Lock {
var lock: RecursiveLock { get }
}
extension LockOwnerType {
func lock() { self.lock.lock() }
func unlock() { self.lock.unlock() }
}

View File

@@ -0,0 +1,18 @@
//
// SynchronizedDisposeType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 10/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
protocol SynchronizedDisposeType: AnyObject, Disposable, Lock {
func synchronized_dispose()
}
extension SynchronizedDisposeType {
func synchronizedDispose() {
self.lock(); defer { self.unlock() }
self.synchronized_dispose()
}
}

View File

@@ -0,0 +1,18 @@
//
// SynchronizedOnType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 10/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
protocol SynchronizedOnType: AnyObject, ObserverType, Lock {
func synchronized_on(_ event: Event<Element>)
}
extension SynchronizedOnType {
func synchronizedOn(_ event: Event<Element>) {
self.lock(); defer { self.unlock() }
self.synchronized_on(event)
}
}

View File

@@ -0,0 +1,13 @@
//
// SynchronizedUnsubscribeType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 10/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
protocol SynchronizedUnsubscribeType: AnyObject {
associatedtype DisposeKey
func synchronizedUnsubscribe(_ disposeKey: DisposeKey)
}

View File

@@ -0,0 +1,19 @@
//
// ConnectableObservableType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/1/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/**
Represents an observable sequence wrapper that can be connected and disconnected from its underlying observable sequence.
*/
public protocol ConnectableObservableType : ObservableType {
/**
Connects the observable wrapper to its source. All subscribed observers will receive values from the underlying observable sequence as long as the connection is established.
- returns: Disposable used to disconnect the observable wrapper from its source, causing subscribed observer to stop receiving values from the underlying observable sequence.
*/
func connect() -> Disposable
}

64
Pods/RxSwift/RxSwift/Date+Dispatch.swift generated Normal file
View File

@@ -0,0 +1,64 @@
//
// Date+Dispatch.swift
// RxSwift
//
// Created by Krunoslav Zaher on 4/14/19.
// Copyright © 2019 Krunoslav Zaher. All rights reserved.
//
import Dispatch
import Foundation
extension DispatchTimeInterval {
var convertToSecondsFactor: Double {
switch self {
case .nanoseconds: return 1_000_000_000.0
case .microseconds: return 1_000_000.0
case .milliseconds: return 1_000.0
case .seconds: return 1.0
case .never: fatalError()
@unknown default: fatalError()
}
}
func map(_ transform: (Int, Double) -> Int) -> DispatchTimeInterval {
switch self {
case .nanoseconds(let value): return .nanoseconds(transform(value, 1_000_000_000.0))
case .microseconds(let value): return .microseconds(transform(value, 1_000_000.0))
case .milliseconds(let value): return .milliseconds(transform(value, 1_000.0))
case .seconds(let value): return .seconds(transform(value, 1.0))
case .never: return .never
@unknown default: fatalError()
}
}
var isNow: Bool {
switch self {
case .nanoseconds(let value), .microseconds(let value), .milliseconds(let value), .seconds(let value): return value == 0
case .never: return false
@unknown default: fatalError()
}
}
internal func reduceWithSpanBetween(earlierDate: Date, laterDate: Date) -> DispatchTimeInterval {
return self.map { value, factor in
let interval = laterDate.timeIntervalSince(earlierDate)
let remainder = Double(value) - interval * factor
guard remainder > 0 else { return 0 }
return Int(remainder.rounded(.toNearestOrAwayFromZero))
}
}
}
extension Date {
internal func addingDispatchInterval(_ dispatchInterval: DispatchTimeInterval) -> Date {
switch dispatchInterval {
case .nanoseconds(let value), .microseconds(let value), .milliseconds(let value), .seconds(let value):
return self.addingTimeInterval(TimeInterval(value) / dispatchInterval.convertToSecondsFactor)
case .never: return Date.distantFuture
@unknown default: fatalError()
}
}
}

13
Pods/RxSwift/RxSwift/Disposable.swift generated Normal file
View File

@@ -0,0 +1,13 @@
//
// Disposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/8/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a disposable resource.
public protocol Disposable {
/// Dispose resource.
func dispose()
}

View File

@@ -0,0 +1,59 @@
//
// AnonymousDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/15/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents an Action-based disposable.
///
/// When dispose method is called, disposal action will be dereferenced.
private final class AnonymousDisposable : DisposeBase, Cancelable {
public typealias DisposeAction = () -> Void
private let disposed = AtomicInt(0)
private var disposeAction: DisposeAction?
/// - returns: Was resource disposed.
public var isDisposed: Bool {
isFlagSet(self.disposed, 1)
}
/// Constructs a new disposable with the given action used for disposal.
///
/// - parameter disposeAction: Disposal action which will be run upon calling `dispose`.
private init(_ disposeAction: @escaping DisposeAction) {
self.disposeAction = disposeAction
super.init()
}
// Non-deprecated version of the constructor, used by `Disposables.create(with:)`
fileprivate init(disposeAction: @escaping DisposeAction) {
self.disposeAction = disposeAction
super.init()
}
/// Calls the disposal action if and only if the current instance hasn't been disposed yet.
///
/// After invoking disposal action, disposal action will be dereferenced.
fileprivate func dispose() {
if fetchOr(self.disposed, 1) == 0 {
if let action = self.disposeAction {
self.disposeAction = nil
action()
}
}
}
}
extension Disposables {
/// Constructs a new disposable with the given action used for disposal.
///
/// - parameter dispose: Disposal action which will be run upon calling `dispose`.
public static func create(with dispose: @escaping () -> Void) -> Cancelable {
AnonymousDisposable(disposeAction: dispose)
}
}

View File

@@ -0,0 +1,53 @@
//
// BinaryDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/12/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents two disposable resources that are disposed together.
private final class BinaryDisposable : DisposeBase, Cancelable {
private let disposed = AtomicInt(0)
// state
private var disposable1: Disposable?
private var disposable2: Disposable?
/// - returns: Was resource disposed.
var isDisposed: Bool {
isFlagSet(self.disposed, 1)
}
/// Constructs new binary disposable from two disposables.
///
/// - parameter disposable1: First disposable
/// - parameter disposable2: Second disposable
init(_ disposable1: Disposable, _ disposable2: Disposable) {
self.disposable1 = disposable1
self.disposable2 = disposable2
super.init()
}
/// Calls the disposal action if and only if the current instance hasn't been disposed yet.
///
/// After invoking disposal action, disposal action will be dereferenced.
func dispose() {
if fetchOr(self.disposed, 1) == 0 {
self.disposable1?.dispose()
self.disposable2?.dispose()
self.disposable1 = nil
self.disposable2 = nil
}
}
}
extension Disposables {
/// Creates a disposable with the given disposables.
public static func create(_ disposable1: Disposable, _ disposable2: Disposable) -> Cancelable {
BinaryDisposable(disposable1, disposable2)
}
}

View File

@@ -0,0 +1,33 @@
//
// BooleanDisposable.swift
// RxSwift
//
// Created by Junior B. on 10/29/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a disposable resource that can be checked for disposal status.
public final class BooleanDisposable : Cancelable {
internal static let BooleanDisposableTrue = BooleanDisposable(isDisposed: true)
private var disposed = false
/// Initializes a new instance of the `BooleanDisposable` class
public init() {
}
/// Initializes a new instance of the `BooleanDisposable` class with given value
public init(isDisposed: Bool) {
self.disposed = isDisposed
}
/// - returns: Was resource disposed.
public var isDisposed: Bool {
self.disposed
}
/// Sets the status to disposed, which can be observer through the `isDisposed` property.
public func dispose() {
self.disposed = true
}
}

View File

@@ -0,0 +1,147 @@
//
// CompositeDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/20/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a group of disposable resources that are disposed together.
public final class CompositeDisposable : DisposeBase, Cancelable {
/// Key used to remove disposable from composite disposable
public struct DisposeKey {
fileprivate let key: BagKey
fileprivate init(key: BagKey) {
self.key = key
}
}
private var lock = SpinLock()
// state
private var disposables: Bag<Disposable>? = Bag()
public var isDisposed: Bool {
self.lock.performLocked { self.disposables == nil }
}
public override init() {
}
/// Initializes a new instance of composite disposable with the specified number of disposables.
public init(_ disposable1: Disposable, _ disposable2: Disposable) {
// This overload is here to make sure we are using optimized version up to 4 arguments.
_ = self.disposables!.insert(disposable1)
_ = self.disposables!.insert(disposable2)
}
/// Initializes a new instance of composite disposable with the specified number of disposables.
public init(_ disposable1: Disposable, _ disposable2: Disposable, _ disposable3: Disposable) {
// This overload is here to make sure we are using optimized version up to 4 arguments.
_ = self.disposables!.insert(disposable1)
_ = self.disposables!.insert(disposable2)
_ = self.disposables!.insert(disposable3)
}
/// Initializes a new instance of composite disposable with the specified number of disposables.
public init(_ disposable1: Disposable, _ disposable2: Disposable, _ disposable3: Disposable, _ disposable4: Disposable, _ disposables: Disposable...) {
// This overload is here to make sure we are using optimized version up to 4 arguments.
_ = self.disposables!.insert(disposable1)
_ = self.disposables!.insert(disposable2)
_ = self.disposables!.insert(disposable3)
_ = self.disposables!.insert(disposable4)
for disposable in disposables {
_ = self.disposables!.insert(disposable)
}
}
/// Initializes a new instance of composite disposable with the specified number of disposables.
public init(disposables: [Disposable]) {
for disposable in disposables {
_ = self.disposables!.insert(disposable)
}
}
/**
Adds a disposable to the CompositeDisposable or disposes the disposable if the CompositeDisposable is disposed.
- parameter disposable: Disposable to add.
- returns: Key that can be used to remove disposable from composite disposable. In case dispose bag was already
disposed `nil` will be returned.
*/
public func insert(_ disposable: Disposable) -> DisposeKey? {
let key = self._insert(disposable)
if key == nil {
disposable.dispose()
}
return key
}
private func _insert(_ disposable: Disposable) -> DisposeKey? {
self.lock.performLocked {
let bagKey = self.disposables?.insert(disposable)
return bagKey.map(DisposeKey.init)
}
}
/// - returns: Gets the number of disposables contained in the `CompositeDisposable`.
public var count: Int {
self.lock.performLocked { self.disposables?.count ?? 0 }
}
/// Removes and disposes the disposable identified by `disposeKey` from the CompositeDisposable.
///
/// - parameter disposeKey: Key used to identify disposable to be removed.
public func remove(for disposeKey: DisposeKey) {
self._remove(for: disposeKey)?.dispose()
}
private func _remove(for disposeKey: DisposeKey) -> Disposable? {
self.lock.performLocked { self.disposables?.removeKey(disposeKey.key) }
}
/// Disposes all disposables in the group and removes them from the group.
public func dispose() {
if let disposables = self._dispose() {
disposeAll(in: disposables)
}
}
private func _dispose() -> Bag<Disposable>? {
self.lock.performLocked {
let current = self.disposables
self.disposables = nil
return current
}
}
}
extension Disposables {
/// Creates a disposable with the given disposables.
public static func create(_ disposable1: Disposable, _ disposable2: Disposable, _ disposable3: Disposable) -> Cancelable {
CompositeDisposable(disposable1, disposable2, disposable3)
}
/// Creates a disposable with the given disposables.
public static func create(_ disposable1: Disposable, _ disposable2: Disposable, _ disposable3: Disposable, _ disposables: Disposable ...) -> Cancelable {
var disposables = disposables
disposables.append(disposable1)
disposables.append(disposable2)
disposables.append(disposable3)
return CompositeDisposable(disposables: disposables)
}
/// Creates a disposable with the given disposables.
public static func create(_ disposables: [Disposable]) -> Cancelable {
switch disposables.count {
case 2:
return Disposables.create(disposables[0], disposables[1])
default:
return CompositeDisposable(disposables: disposables)
}
}
}

View File

@@ -0,0 +1,13 @@
//
// Disposables.swift
// RxSwift
//
// Created by Mohsen Ramezanpoor on 01/08/2016.
// Copyright © 2016 Krunoslav Zaher. All rights reserved.
//
/// A collection of utility methods for common disposable operations.
public struct Disposables {
private init() {}
}

View File

@@ -0,0 +1,144 @@
//
// DisposeBag.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension Disposable {
/// Adds `self` to `bag`
///
/// - parameter bag: `DisposeBag` to add `self` to.
public func disposed(by bag: DisposeBag) {
bag.insert(self)
}
}
/**
Thread safe bag that disposes added disposables on `deinit`.
This returns ARC (RAII) like resource management to `RxSwift`.
In case contained disposables need to be disposed, just put a different dispose bag
or create a new one in its place.
self.existingDisposeBag = DisposeBag()
In case explicit disposal is necessary, there is also `CompositeDisposable`.
*/
public final class DisposeBag: DisposeBase {
private var lock = SpinLock()
// state
private var disposables = [Disposable]()
private var isDisposed = false
/// Constructs new empty dispose bag.
public override init() {
super.init()
}
/// Adds `disposable` to be disposed when dispose bag is being deinited.
///
/// - parameter disposable: Disposable to add.
public func insert(_ disposable: Disposable) {
self._insert(disposable)?.dispose()
}
private func _insert(_ disposable: Disposable) -> Disposable? {
self.lock.performLocked {
if self.isDisposed {
return disposable
}
self.disposables.append(disposable)
return nil
}
}
/// This is internal on purpose, take a look at `CompositeDisposable` instead.
private func dispose() {
let oldDisposables = self._dispose()
for disposable in oldDisposables {
disposable.dispose()
}
}
private func _dispose() -> [Disposable] {
self.lock.performLocked {
let disposables = self.disposables
self.disposables.removeAll(keepingCapacity: false)
self.isDisposed = true
return disposables
}
}
deinit {
self.dispose()
}
}
extension DisposeBag {
/// Convenience init allows a list of disposables to be gathered for disposal.
public convenience init(disposing disposables: Disposable...) {
self.init()
self.disposables += disposables
}
/// Convenience init which utilizes a function builder to let you pass in a list of
/// disposables to make a DisposeBag of.
public convenience init(@DisposableBuilder builder: () -> [Disposable]) {
self.init(disposing: builder())
}
/// Convenience init allows an array of disposables to be gathered for disposal.
public convenience init(disposing disposables: [Disposable]) {
self.init()
self.disposables += disposables
}
/// Convenience function allows a list of disposables to be gathered for disposal.
public func insert(_ disposables: Disposable...) {
self.insert(disposables)
}
/// Convenience function allows a list of disposables to be gathered for disposal.
public func insert(@DisposableBuilder builder: () -> [Disposable]) {
self.insert(builder())
}
/// Convenience function allows an array of disposables to be gathered for disposal.
public func insert(_ disposables: [Disposable]) {
self.lock.performLocked {
if self.isDisposed {
disposables.forEach { $0.dispose() }
} else {
self.disposables += disposables
}
}
}
/// A function builder accepting a list of Disposables and returning them as an array.
#if swift(>=5.4)
@resultBuilder
public struct DisposableBuilder {
public static func buildBlock(_ disposables: Disposable...) -> [Disposable] {
return disposables
}
}
#else
@_functionBuilder
public struct DisposableBuilder {
public static func buildBlock(_ disposables: Disposable...) -> [Disposable] {
return disposables
}
}
#endif
}

View File

@@ -0,0 +1,22 @@
//
// DisposeBase.swift
// RxSwift
//
// Created by Krunoslav Zaher on 4/4/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Base class for all disposables.
public class DisposeBase {
init() {
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
}
deinit {
#if TRACE_RESOURCES
_ = Resources.decrementTotal()
#endif
}
}

View File

@@ -0,0 +1,30 @@
//
// NopDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/15/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a disposable that does nothing on disposal.
///
/// Nop = No Operation
private struct NopDisposable : Disposable {
fileprivate static let noOp: Disposable = NopDisposable()
private init() {
}
/// Does nothing.
public func dispose() {
}
}
extension Disposables {
/**
Creates a disposable that does nothing on disposal.
*/
static public func create() -> Disposable { NopDisposable.noOp }
}

View File

@@ -0,0 +1,112 @@
//
// RefCountDisposable.swift
// RxSwift
//
// Created by Junior B. on 10/29/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a disposable resource that only disposes its underlying disposable resource when all dependent disposable objects have been disposed.
public final class RefCountDisposable : DisposeBase, Cancelable {
private var lock = SpinLock()
private var disposable = nil as Disposable?
private var primaryDisposed = false
private var count = 0
/// - returns: Was resource disposed.
public var isDisposed: Bool {
self.lock.performLocked { self.disposable == nil }
}
/// Initializes a new instance of the `RefCountDisposable`.
public init(disposable: Disposable) {
self.disposable = disposable
super.init()
}
/**
Holds a dependent disposable that when disposed decreases the refcount on the underlying disposable.
When getter is called, a dependent disposable contributing to the reference count that manages the underlying disposable's lifetime is returned.
*/
public func retain() -> Disposable {
self.lock.performLocked {
if self.disposable != nil {
do {
_ = try incrementChecked(&self.count)
} catch {
rxFatalError("RefCountDisposable increment failed")
}
return RefCountInnerDisposable(self)
} else {
return Disposables.create()
}
}
}
/// Disposes the underlying disposable only when all dependent disposables have been disposed.
public func dispose() {
let oldDisposable: Disposable? = self.lock.performLocked {
if let oldDisposable = self.disposable, !self.primaryDisposed {
self.primaryDisposed = true
if self.count == 0 {
self.disposable = nil
return oldDisposable
}
}
return nil
}
if let disposable = oldDisposable {
disposable.dispose()
}
}
fileprivate func release() {
let oldDisposable: Disposable? = self.lock.performLocked {
if let oldDisposable = self.disposable {
do {
_ = try decrementChecked(&self.count)
} catch {
rxFatalError("RefCountDisposable decrement on release failed")
}
guard self.count >= 0 else {
rxFatalError("RefCountDisposable counter is lower than 0")
}
if self.primaryDisposed && self.count == 0 {
self.disposable = nil
return oldDisposable
}
}
return nil
}
if let disposable = oldDisposable {
disposable.dispose()
}
}
}
internal final class RefCountInnerDisposable: DisposeBase, Disposable
{
private let parent: RefCountDisposable
private let isDisposed = AtomicInt(0)
init(_ parent: RefCountDisposable) {
self.parent = parent
super.init()
}
internal func dispose()
{
if fetchOr(self.isDisposed, 1) == 0 {
self.parent.release()
}
}
}

View File

@@ -0,0 +1,50 @@
//
// ScheduledDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/13/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
private let disposeScheduledDisposable: (ScheduledDisposable) -> Disposable = { sd in
sd.disposeInner()
return Disposables.create()
}
/// Represents a disposable resource whose disposal invocation will be scheduled on the specified scheduler.
public final class ScheduledDisposable : Cancelable {
public let scheduler: ImmediateSchedulerType
private let disposed = AtomicInt(0)
// state
private var disposable: Disposable?
/// - returns: Was resource disposed.
public var isDisposed: Bool {
isFlagSet(self.disposed, 1)
}
/**
Initializes a new instance of the `ScheduledDisposable` that uses a `scheduler` on which to dispose the `disposable`.
- parameter scheduler: Scheduler where the disposable resource will be disposed on.
- parameter disposable: Disposable resource to dispose on the given scheduler.
*/
public init(scheduler: ImmediateSchedulerType, disposable: Disposable) {
self.scheduler = scheduler
self.disposable = disposable
}
/// Disposes the wrapped disposable on the provided scheduler.
public func dispose() {
_ = self.scheduler.schedule(self, action: disposeScheduledDisposable)
}
func disposeInner() {
if fetchOr(self.disposed, 1) == 0 {
self.disposable!.dispose()
self.disposable = nil
}
}
}

View File

@@ -0,0 +1,73 @@
//
// SerialDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/12/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a disposable resource whose underlying disposable resource can be replaced by another disposable resource, causing automatic disposal of the previous underlying disposable resource.
public final class SerialDisposable : DisposeBase, Cancelable {
private var lock = SpinLock()
// state
private var current = nil as Disposable?
private var disposed = false
/// - returns: Was resource disposed.
public var isDisposed: Bool {
self.disposed
}
/// Initializes a new instance of the `SerialDisposable`.
override public init() {
super.init()
}
/**
Gets or sets the underlying disposable.
Assigning this property disposes the previous disposable object.
If the `SerialDisposable` has already been disposed, assignment to this property causes immediate disposal of the given disposable object.
*/
public var disposable: Disposable {
get {
self.lock.performLocked {
self.current ?? Disposables.create()
}
}
set (newDisposable) {
let disposable: Disposable? = self.lock.performLocked {
if self.isDisposed {
return newDisposable
}
else {
let toDispose = self.current
self.current = newDisposable
return toDispose
}
}
if let disposable = disposable {
disposable.dispose()
}
}
}
/// Disposes the underlying disposable as well as all future replacements.
public func dispose() {
self._dispose()?.dispose()
}
private func _dispose() -> Disposable? {
self.lock.performLocked {
guard !self.isDisposed else { return nil }
self.disposed = true
let current = self.current
self.current = nil
return current
}
}
}

View File

@@ -0,0 +1,72 @@
//
// SingleAssignmentDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/15/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/**
Represents a disposable resource which only allows a single assignment of its underlying disposable resource.
If an underlying disposable resource has already been set, future attempts to set the underlying disposable resource will throw an exception.
*/
public final class SingleAssignmentDisposable : DisposeBase, Cancelable {
private struct DisposeState: OptionSet {
let rawValue: Int32
static let disposed = DisposeState(rawValue: 1 << 0)
static let disposableSet = DisposeState(rawValue: 1 << 1)
}
// state
private let state = AtomicInt(0)
private var disposable = nil as Disposable?
/// - returns: A value that indicates whether the object is disposed.
public var isDisposed: Bool {
isFlagSet(self.state, DisposeState.disposed.rawValue)
}
/// Initializes a new instance of the `SingleAssignmentDisposable`.
public override init() {
super.init()
}
/// Gets or sets the underlying disposable. After disposal, the result of getting this property is undefined.
///
/// **Throws exception if the `SingleAssignmentDisposable` has already been assigned to.**
public func setDisposable(_ disposable: Disposable) {
self.disposable = disposable
let previousState = fetchOr(self.state, DisposeState.disposableSet.rawValue)
if (previousState & DisposeState.disposableSet.rawValue) != 0 {
rxFatalError("oldState.disposable != nil")
}
if (previousState & DisposeState.disposed.rawValue) != 0 {
disposable.dispose()
self.disposable = nil
}
}
/// Disposes the underlying disposable.
public func dispose() {
let previousState = fetchOr(self.state, DisposeState.disposed.rawValue)
if (previousState & DisposeState.disposed.rawValue) != 0 {
return
}
if (previousState & DisposeState.disposableSet.rawValue) != 0 {
guard let disposable = self.disposable else {
rxFatalError("Disposable not set")
}
disposable.dispose()
self.disposable = nil
}
}
}

View File

@@ -0,0 +1,21 @@
//
// SubscriptionDisposable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 10/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
struct SubscriptionDisposable<T: SynchronizedUnsubscribeType> : Disposable {
private let key: T.DisposeKey
private weak var owner: T?
init(owner: T, key: T.DisposeKey) {
self.owner = owner
self.key = key
}
func dispose() {
self.owner?.synchronizedUnsubscribe(self.key)
}
}

52
Pods/RxSwift/RxSwift/Errors.swift generated Normal file
View File

@@ -0,0 +1,52 @@
//
// Errors.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/28/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
let RxErrorDomain = "RxErrorDomain"
let RxCompositeFailures = "RxCompositeFailures"
/// Generic Rx error codes.
public enum RxError
: Swift.Error
, CustomDebugStringConvertible {
/// Unknown error occurred.
case unknown
/// Performing an action on disposed object.
case disposed(object: AnyObject)
/// Arithmetic overflow error.
case overflow
/// Argument out of range error.
case argumentOutOfRange
/// Sequence doesn't contain any elements.
case noElements
/// Sequence contains more than one element.
case moreThanOneElement
/// Timeout error.
case timeout
}
extension RxError {
/// A textual representation of `self`, suitable for debugging.
public var debugDescription: String {
switch self {
case .unknown:
return "Unknown error occurred."
case .disposed(let object):
return "Object `\(object)` was already disposed."
case .overflow:
return "Arithmetic overflow occurred."
case .argumentOutOfRange:
return "Argument out of range."
case .noElements:
return "Sequence doesn't contain any elements."
case .moreThanOneElement:
return "Sequence contains more than one element."
case .timeout:
return "Sequence timeout."
}
}
}

104
Pods/RxSwift/RxSwift/Event.swift generated Normal file
View File

@@ -0,0 +1,104 @@
//
// Event.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/8/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a sequence event.
///
/// Sequence grammar:
/// **next\* (error | completed)**
@frozen public enum Event<Element> {
/// Next element is produced.
case next(Element)
/// Sequence terminated with an error.
case error(Swift.Error)
/// Sequence completed successfully.
case completed
}
extension Event: CustomDebugStringConvertible {
/// Description of event.
public var debugDescription: String {
switch self {
case .next(let value):
return "next(\(value))"
case .error(let error):
return "error(\(error))"
case .completed:
return "completed"
}
}
}
extension Event {
/// Is `completed` or `error` event.
public var isStopEvent: Bool {
switch self {
case .next: return false
case .error, .completed: return true
}
}
/// If `next` event, returns element value.
public var element: Element? {
if case .next(let value) = self {
return value
}
return nil
}
/// If `error` event, returns error.
public var error: Swift.Error? {
if case .error(let error) = self {
return error
}
return nil
}
/// If `completed` event, returns `true`.
public var isCompleted: Bool {
if case .completed = self {
return true
}
return false
}
}
extension Event {
/// Maps sequence elements using transform. If error happens during the transform, `.error`
/// will be returned as value.
public func map<Result>(_ transform: (Element) throws -> Result) -> Event<Result> {
do {
switch self {
case let .next(element):
return .next(try transform(element))
case let .error(error):
return .error(error)
case .completed:
return .completed
}
}
catch let e {
return .error(e)
}
}
}
/// A type that can be converted to `Event<Element>`.
public protocol EventConvertible {
/// Type of element in event
associatedtype Element
/// Event representation of this instance
var event: Event<Element> { get }
}
extension Event: EventConvertible {
/// Event representation of this instance
public var event: Event<Element> { self }
}

View File

@@ -0,0 +1,50 @@
//
// Bag+Rx.swift
// RxSwift
//
// Created by Krunoslav Zaher on 10/19/16.
// Copyright © 2016 Krunoslav Zaher. All rights reserved.
//
// MARK: forEach
@inline(__always)
func dispatch<Element>(_ bag: Bag<(Event<Element>) -> Void>, _ event: Event<Element>) {
bag._value0?(event)
if bag._onlyFastPath {
return
}
let pairs = bag._pairs
for i in 0 ..< pairs.count {
pairs[i].value(event)
}
if let dictionary = bag._dictionary {
for element in dictionary.values {
element(event)
}
}
}
/// Dispatches `dispose` to all disposables contained inside bag.
func disposeAll(in bag: Bag<Disposable>) {
bag._value0?.dispose()
if bag._onlyFastPath {
return
}
let pairs = bag._pairs
for i in 0 ..< pairs.count {
pairs[i].value.dispose()
}
if let dictionary = bag._dictionary {
for element in dictionary.values {
element.dispose()
}
}
}

View File

@@ -0,0 +1,35 @@
//
// GroupedObservable.swift
// RxSwift
//
// Created by Tomi Koskinen on 01/12/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents an observable sequence of elements that have a common key.
public struct GroupedObservable<Key, Element> : ObservableType {
/// Gets the common key.
public let key: Key
private let source: Observable<Element>
/// Initializes grouped observable sequence with key and source observable sequence.
///
/// - parameter key: Grouped observable sequence key
/// - parameter source: Observable sequence that represents sequence of elements for the key
/// - returns: Grouped observable sequence of elements for the specific key
public init(key: Key, source: Observable<Element>) {
self.key = key
self.source = source
}
/// Subscribes `observer` to receive events for this sequence.
public func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
self.source.subscribe(observer)
}
/// Converts `self` to `Observable` sequence.
public func asObservable() -> Observable<Element> {
self.source
}
}

View File

@@ -0,0 +1,36 @@
//
// ImmediateSchedulerType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 5/31/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents an object that immediately schedules units of work.
public protocol ImmediateSchedulerType {
/**
Schedules an action to be executed immediately.
- parameter state: State passed to the action to be executed.
- parameter action: Action to be executed.
- returns: The disposable object used to cancel the scheduled action (best effort).
*/
func schedule<StateType>(_ state: StateType, action: @escaping (StateType) -> Disposable) -> Disposable
}
extension ImmediateSchedulerType {
/**
Schedules an action to be executed recursively.
- parameter state: State passed to the action to be executed.
- parameter action: Action to execute recursively. The last parameter passed to the action is used to trigger recursive scheduling of the action, passing in recursive invocation state.
- returns: The disposable object used to cancel the scheduled action (best effort).
*/
public func scheduleRecursive<State>(_ state: State, action: @escaping (_ state: State, _ recurse: (State) -> Void) -> Void) -> Disposable {
let recursiveScheduler = RecursiveImmediateScheduler(action: action, scheduler: self)
recursiveScheduler.schedule(state)
return Disposables.create(with: recursiveScheduler.dispose)
}
}

View File

@@ -0,0 +1,68 @@
//
// Observable+Concurrency.swift
// RxSwift
//
// Created by Shai Mishali on 22/09/2021.
// Copyright © 2021 Krunoslav Zaher. All rights reserved.
//
#if swift(>=5.5.2) && canImport(_Concurrency)
import Foundation
@available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
public extension ObservableConvertibleType {
/// Allows iterating over the values of an Observable
/// asynchronously via Swift's concurrency features (`async/await`)
///
/// A sample usage would look like so:
///
/// ```swift
/// do {
/// for try await value in observable.values {
/// // Handle emitted values
/// }
/// } catch {
/// // Handle error
/// }
/// ```
var values: AsyncThrowingStream<Element, Error> {
AsyncThrowingStream<Element, Error> { continuation in
let disposable = asObservable().subscribe(
onNext: { value in continuation.yield(value) },
onError: { error in continuation.finish(throwing: error) },
onCompleted: { continuation.finish() },
onDisposed: { continuation.onTermination?(.cancelled) }
)
continuation.onTermination = { @Sendable _ in
disposable.dispose()
}
}
}
}
@available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
public extension AsyncSequence {
/// Convert an `AsyncSequence` to an `Observable` emitting
/// values of the asynchronous sequence's type
///
/// - returns: An `Observable` of the async sequence's type
func asObservable() -> Observable<Element> {
Observable.create { observer in
let task = Task {
do {
for try await value in self {
observer.onNext(value)
}
observer.onCompleted()
} catch {
observer.onError(error)
}
}
return Disposables.create { task.cancel() }
}
}
}
#endif

31
Pods/RxSwift/RxSwift/Observable.swift generated Normal file
View File

@@ -0,0 +1,31 @@
//
// Observable.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/8/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// A type-erased `ObservableType`.
///
/// It represents a push style sequence.
public class Observable<Element> : ObservableType {
init() {
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
}
public func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
rxAbstractMethod()
}
public func asObservable() -> Observable<Element> { self }
deinit {
#if TRACE_RESOURCES
_ = Resources.decrementTotal()
#endif
}
}

View File

@@ -0,0 +1,18 @@
//
// ObservableConvertibleType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 9/17/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Type that can be converted to observable sequence (`Observable<Element>`).
public protocol ObservableConvertibleType {
/// Type of elements in sequence.
associatedtype Element
/// Converts `self` to `Observable` sequence.
///
/// - returns: Observable sequence that represents `self`.
func asObservable() -> Observable<Element>
}

View File

@@ -0,0 +1,174 @@
//
// ObservableType+Extensions.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/21/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
#if DEBUG
import Foundation
#endif
extension ObservableType {
/**
Subscribes an event handler to an observable sequence.
- parameter on: Action to invoke for each event in the observable sequence.
- returns: Subscription object used to unsubscribe from the observable sequence.
*/
public func subscribe(_ on: @escaping (Event<Element>) -> Void) -> Disposable {
let observer = AnonymousObserver { e in
on(e)
}
return self.asObservable().subscribe(observer)
}
/**
Subscribes an element handler, an error handler, a completion handler and disposed handler to an observable sequence.
Also, take in an object and provide an unretained, safe to use (i.e. not implicitly unwrapped), reference to it along with the events emitted by the sequence.
- Note: If `object` can't be retained, none of the other closures will be invoked.
- parameter object: The object to provide an unretained reference on.
- parameter onNext: Action to invoke for each element in the observable sequence.
- parameter onError: Action to invoke upon errored termination of the observable sequence.
- parameter onCompleted: Action to invoke upon graceful termination of the observable sequence.
- parameter onDisposed: Action to invoke upon any type of termination of sequence (if the sequence has
gracefully completed, errored, or if the generation is canceled by disposing subscription).
- returns: Subscription object used to unsubscribe from the observable sequence.
*/
public func subscribe<Object: AnyObject>(
with object: Object,
onNext: ((Object, Element) -> Void)? = nil,
onError: ((Object, Swift.Error) -> Void)? = nil,
onCompleted: ((Object) -> Void)? = nil,
onDisposed: ((Object) -> Void)? = nil
) -> Disposable {
subscribe(
onNext: { [weak object] in
guard let object = object else { return }
onNext?(object, $0)
},
onError: { [weak object] in
guard let object = object else { return }
onError?(object, $0)
},
onCompleted: { [weak object] in
guard let object = object else { return }
onCompleted?(object)
},
onDisposed: { [weak object] in
guard let object = object else { return }
onDisposed?(object)
}
)
}
/**
Subscribes an element handler, an error handler, a completion handler and disposed handler to an observable sequence.
- parameter onNext: Action to invoke for each element in the observable sequence.
- parameter onError: Action to invoke upon errored termination of the observable sequence.
- parameter onCompleted: Action to invoke upon graceful termination of the observable sequence.
- parameter onDisposed: Action to invoke upon any type of termination of sequence (if the sequence has
gracefully completed, errored, or if the generation is canceled by disposing subscription).
- returns: Subscription object used to unsubscribe from the observable sequence.
*/
public func subscribe(
onNext: ((Element) -> Void)? = nil,
onError: ((Swift.Error) -> Void)? = nil,
onCompleted: (() -> Void)? = nil,
onDisposed: (() -> Void)? = nil
) -> Disposable {
let disposable: Disposable
if let disposed = onDisposed {
disposable = Disposables.create(with: disposed)
}
else {
disposable = Disposables.create()
}
#if DEBUG
let synchronizationTracker = SynchronizationTracker()
#endif
let callStack = Hooks.recordCallStackOnError ? Hooks.customCaptureSubscriptionCallstack() : []
let observer = AnonymousObserver<Element> { event in
#if DEBUG
synchronizationTracker.register(synchronizationErrorMessage: .default)
defer { synchronizationTracker.unregister() }
#endif
switch event {
case .next(let value):
onNext?(value)
case .error(let error):
if let onError = onError {
onError(error)
}
else {
Hooks.defaultErrorHandler(callStack, error)
}
disposable.dispose()
case .completed:
onCompleted?()
disposable.dispose()
}
}
return Disposables.create(
self.asObservable().subscribe(observer),
disposable
)
}
}
import Foundation
extension Hooks {
public typealias DefaultErrorHandler = (_ subscriptionCallStack: [String], _ error: Error) -> Void
public typealias CustomCaptureSubscriptionCallstack = () -> [String]
private static let lock = RecursiveLock()
private static var _defaultErrorHandler: DefaultErrorHandler = { subscriptionCallStack, error in
#if DEBUG
let serializedCallStack = subscriptionCallStack.joined(separator: "\n")
print("Unhandled error happened: \(error)")
if !serializedCallStack.isEmpty {
print("subscription called from:\n\(serializedCallStack)")
}
#endif
}
private static var _customCaptureSubscriptionCallstack: CustomCaptureSubscriptionCallstack = {
#if DEBUG
return Thread.callStackSymbols
#else
return []
#endif
}
/// Error handler called in case onError handler wasn't provided.
public static var defaultErrorHandler: DefaultErrorHandler {
get {
lock.performLocked { _defaultErrorHandler }
}
set {
lock.performLocked { _defaultErrorHandler = newValue }
}
}
/// Subscription callstack block to fetch custom callstack information.
public static var customCaptureSubscriptionCallstack: CustomCaptureSubscriptionCallstack {
get {
lock.performLocked { _customCaptureSubscriptionCallstack }
}
set {
lock.performLocked { _customCaptureSubscriptionCallstack = newValue }
}
}
}

View File

@@ -0,0 +1,45 @@
//
// ObservableType.swift
// RxSwift
//
// Created by Krunoslav Zaher on 8/8/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/// Represents a push style sequence.
public protocol ObservableType: ObservableConvertibleType {
/**
Subscribes `observer` to receive events for this sequence.
### Grammar
**Next\* (Error | Completed)?**
* sequences can produce zero or more elements so zero or more `Next` events can be sent to `observer`
* once an `Error` or `Completed` event is sent, the sequence terminates and can't produce any other elements
It is possible that events are sent from different threads, but no two events can be sent concurrently to
`observer`.
### Resource Management
When sequence sends `Complete` or `Error` event all internal resources that compute sequence elements
will be freed.
To cancel production of sequence elements and free resources immediately, call `dispose` on returned
subscription.
- returns: Subscription for `observer` that can be used to cancel production of sequence elements and free resources.
*/
func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element
}
extension ObservableType {
/// Default implementation of converting `ObservableType` to `Observable`.
public func asObservable() -> Observable<Element> {
// temporary workaround
//return Observable.create(subscribe: self.subscribe)
Observable.create { o in self.subscribe(o) }
}
}

View File

@@ -0,0 +1,44 @@
//
// AddRef.swift
// RxSwift
//
// Created by Junior B. on 30/10/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
final class AddRefSink<Observer: ObserverType> : Sink<Observer>, ObserverType {
typealias Element = Observer.Element
override init(observer: Observer, cancel: Cancelable) {
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.forwardOn(event)
case .completed, .error:
self.forwardOn(event)
self.dispose()
}
}
}
final class AddRef<Element> : Producer<Element> {
private let source: Observable<Element>
private let refCount: RefCountDisposable
init(source: Observable<Element>, refCount: RefCountDisposable) {
self.source = source
self.refCount = refCount
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let releaseDisposable = self.refCount.retain()
let sink = AddRefSink(observer: observer, cancel: cancel)
let subscription = Disposables.create(releaseDisposable, self.source.subscribe(sink))
return (sink: sink, subscription: subscription)
}
}

157
Pods/RxSwift/RxSwift/Observables/Amb.swift generated Normal file
View File

@@ -0,0 +1,157 @@
//
// Amb.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/14/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Propagates the observable sequence that reacts first.
- seealso: [amb operator on reactivex.io](http://reactivex.io/documentation/operators/amb.html)
- returns: An observable sequence that surfaces any of the given sequences, whichever reacted first.
*/
public static func amb<Sequence: Swift.Sequence>(_ sequence: Sequence) -> Observable<Element>
where Sequence.Element == Observable<Element> {
sequence.reduce(Observable<Sequence.Element.Element>.never()) { a, o in
a.amb(o.asObservable())
}
}
}
extension ObservableType {
/**
Propagates the observable sequence that reacts first.
- seealso: [amb operator on reactivex.io](http://reactivex.io/documentation/operators/amb.html)
- parameter right: Second observable sequence.
- returns: An observable sequence that surfaces either of the given sequences, whichever reacted first.
*/
public func amb<O2: ObservableType>
(_ right: O2)
-> Observable<Element> where O2.Element == Element {
Amb(left: self.asObservable(), right: right.asObservable())
}
}
private enum AmbState {
case neither
case left
case right
}
final private class AmbObserver<Observer: ObserverType>: ObserverType {
typealias Element = Observer.Element
typealias Parent = AmbSink<Observer>
typealias This = AmbObserver<Observer>
typealias Sink = (This, Event<Element>) -> Void
private let parent: Parent
fileprivate var sink: Sink
fileprivate var cancel: Disposable
init(parent: Parent, cancel: Disposable, sink: @escaping Sink) {
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
self.parent = parent
self.sink = sink
self.cancel = cancel
}
func on(_ event: Event<Element>) {
self.sink(self, event)
if event.isStopEvent {
self.cancel.dispose()
}
}
deinit {
#if TRACE_RESOURCES
_ = Resources.decrementTotal()
#endif
}
}
final private class AmbSink<Observer: ObserverType>: Sink<Observer> {
typealias Element = Observer.Element
typealias Parent = Amb<Element>
typealias AmbObserverType = AmbObserver<Observer>
private let parent: Parent
private let lock = RecursiveLock()
// state
private var choice = AmbState.neither
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let disposeAll = Disposables.create(subscription1, subscription2)
let forwardEvent = { (o: AmbObserverType, event: Event<Element>) -> Void in
self.forwardOn(event)
if event.isStopEvent {
self.dispose()
}
}
let decide = { (o: AmbObserverType, event: Event<Element>, me: AmbState, otherSubscription: Disposable) in
self.lock.performLocked {
if self.choice == .neither {
self.choice = me
o.sink = forwardEvent
o.cancel = disposeAll
otherSubscription.dispose()
}
if self.choice == me {
self.forwardOn(event)
if event.isStopEvent {
self.dispose()
}
}
}
}
let sink1 = AmbObserver(parent: self, cancel: subscription1) { o, e in
decide(o, e, .left, subscription2)
}
let sink2 = AmbObserver(parent: self, cancel: subscription1) { o, e in
decide(o, e, .right, subscription1)
}
subscription1.setDisposable(self.parent.left.subscribe(sink1))
subscription2.setDisposable(self.parent.right.subscribe(sink2))
return disposeAll
}
}
final private class Amb<Element>: Producer<Element> {
fileprivate let left: Observable<Element>
fileprivate let right: Observable<Element>
init(left: Observable<Element>, right: Observable<Element>) {
self.left = left
self.right = right
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = AmbSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,48 @@
//
// AsMaybe.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/12/17.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
private final class AsMaybeSink<Observer: ObserverType> : Sink<Observer>, ObserverType {
typealias Element = Observer.Element
private var element: Event<Element>?
func on(_ event: Event<Element>) {
switch event {
case .next:
if self.element != nil {
self.forwardOn(.error(RxError.moreThanOneElement))
self.dispose()
}
self.element = event
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
if let element = self.element {
self.forwardOn(element)
}
self.forwardOn(.completed)
self.dispose()
}
}
}
final class AsMaybe<Element>: Producer<Element> {
private let source: Observable<Element>
init(source: Observable<Element>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = AsMaybeSink(observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,51 @@
//
// AsSingle.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/12/17.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
private final class AsSingleSink<Observer: ObserverType> : Sink<Observer>, ObserverType {
typealias Element = Observer.Element
private var element: Event<Element>?
func on(_ event: Event<Element>) {
switch event {
case .next:
if self.element != nil {
self.forwardOn(.error(RxError.moreThanOneElement))
self.dispose()
}
self.element = event
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
if let element = self.element {
self.forwardOn(element)
self.forwardOn(.completed)
}
else {
self.forwardOn(.error(RxError.noElements))
}
self.dispose()
}
}
}
final class AsSingle<Element>: Producer<Element> {
private let source: Observable<Element>
init(source: Observable<Element>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = AsSingleSink(observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,138 @@
//
// Buffer.swift
// RxSwift
//
// Created by Krunoslav Zaher on 9/13/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Projects each element of an observable sequence into a buffer that's sent out when either it's full or a given amount of time has elapsed, using the specified scheduler to run timers.
A useful real-world analogy of this overload is the behavior of a ferry leaving the dock when all seats are taken, or at the scheduled time of departure, whichever event occurs first.
- seealso: [buffer operator on reactivex.io](http://reactivex.io/documentation/operators/buffer.html)
- parameter timeSpan: Maximum time length of a buffer.
- parameter count: Maximum element count of a buffer.
- parameter scheduler: Scheduler to run buffering timers on.
- returns: An observable sequence of buffers.
*/
public func buffer(timeSpan: RxTimeInterval, count: Int, scheduler: SchedulerType)
-> Observable<[Element]> {
BufferTimeCount(source: self.asObservable(), timeSpan: timeSpan, count: count, scheduler: scheduler)
}
}
final private class BufferTimeCount<Element>: Producer<[Element]> {
fileprivate let timeSpan: RxTimeInterval
fileprivate let count: Int
fileprivate let scheduler: SchedulerType
fileprivate let source: Observable<Element>
init(source: Observable<Element>, timeSpan: RxTimeInterval, count: Int, scheduler: SchedulerType) {
self.source = source
self.timeSpan = timeSpan
self.count = count
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == [Element] {
let sink = BufferTimeCountSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
final private class BufferTimeCountSink<Element, Observer: ObserverType>
: Sink<Observer>
, LockOwnerType
, ObserverType
, SynchronizedOnType where Observer.Element == [Element] {
typealias Parent = BufferTimeCount<Element>
private let parent: Parent
let lock = RecursiveLock()
// state
private let timerD = SerialDisposable()
private var buffer = [Element]()
private var windowID = 0
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
self.createTimer(self.windowID)
return Disposables.create(timerD, parent.source.subscribe(self))
}
func startNewWindowAndSendCurrentOne() {
self.windowID = self.windowID &+ 1
let windowID = self.windowID
let buffer = self.buffer
self.buffer = []
self.forwardOn(.next(buffer))
self.createTimer(windowID)
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next(let element):
self.buffer.append(element)
if self.buffer.count == self.parent.count {
self.startNewWindowAndSendCurrentOne()
}
case .error(let error):
self.buffer = []
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.forwardOn(.next(self.buffer))
self.forwardOn(.completed)
self.dispose()
}
}
func createTimer(_ windowID: Int) {
if self.timerD.isDisposed {
return
}
if self.windowID != windowID {
return
}
let nextTimer = SingleAssignmentDisposable()
self.timerD.disposable = nextTimer
let disposable = self.parent.scheduler.scheduleRelative(windowID, dueTime: self.parent.timeSpan) { previousWindowID in
self.lock.performLocked {
if previousWindowID != self.windowID {
return
}
self.startNewWindowAndSendCurrentOne()
}
return Disposables.create()
}
nextTimer.setDisposable(disposable)
}
}

View File

@@ -0,0 +1,275 @@
//
// Catch.swift
// RxSwift
//
// Created by Krunoslav Zaher on 4/19/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Continues an observable sequence that is terminated by an error with the observable sequence produced by the handler.
- seealso: [catch operator on reactivex.io](http://reactivex.io/documentation/operators/catch.html)
- parameter handler: Error handler function, producing another observable sequence.
- returns: An observable sequence containing the source sequence's elements, followed by the elements produced by the handler's resulting observable sequence in case an error occurred.
*/
public func `catch`(_ handler: @escaping (Swift.Error) throws -> Observable<Element>)
-> Observable<Element> {
Catch(source: self.asObservable(), handler: handler)
}
/**
Continues an observable sequence that is terminated by an error with the observable sequence produced by the handler.
- seealso: [catch operator on reactivex.io](http://reactivex.io/documentation/operators/catch.html)
- parameter handler: Error handler function, producing another observable sequence.
- returns: An observable sequence containing the source sequence's elements, followed by the elements produced by the handler's resulting observable sequence in case an error occurred.
*/
@available(*, deprecated, renamed: "catch(_:)")
public func catchError(_ handler: @escaping (Swift.Error) throws -> Observable<Element>)
-> Observable<Element> {
`catch`(handler)
}
/**
Continues an observable sequence that is terminated by an error with a single element.
- seealso: [catch operator on reactivex.io](http://reactivex.io/documentation/operators/catch.html)
- parameter element: Last element in an observable sequence in case error occurs.
- returns: An observable sequence containing the source sequence's elements, followed by the `element` in case an error occurred.
*/
public func catchAndReturn(_ element: Element)
-> Observable<Element> {
Catch(source: self.asObservable(), handler: { _ in Observable.just(element) })
}
/**
Continues an observable sequence that is terminated by an error with a single element.
- seealso: [catch operator on reactivex.io](http://reactivex.io/documentation/operators/catch.html)
- parameter element: Last element in an observable sequence in case error occurs.
- returns: An observable sequence containing the source sequence's elements, followed by the `element` in case an error occurred.
*/
@available(*, deprecated, renamed: "catchAndReturn(_:)")
public func catchErrorJustReturn(_ element: Element)
-> Observable<Element> {
catchAndReturn(element)
}
}
extension ObservableType {
/**
Continues an observable sequence that is terminated by an error with the next observable sequence.
- seealso: [catch operator on reactivex.io](http://reactivex.io/documentation/operators/catch.html)
- returns: An observable sequence containing elements from consecutive source sequences until a source sequence terminates successfully.
*/
@available(*, deprecated, renamed: "catch(onSuccess:onFailure:onDisposed:)")
public static func catchError<Sequence: Swift.Sequence>(_ sequence: Sequence) -> Observable<Element>
where Sequence.Element == Observable<Element> {
`catch`(sequence: sequence)
}
/**
Continues an observable sequence that is terminated by an error with the next observable sequence.
- seealso: [catch operator on reactivex.io](http://reactivex.io/documentation/operators/catch.html)
- returns: An observable sequence containing elements from consecutive source sequences until a source sequence terminates successfully.
*/
public static func `catch`<Sequence: Swift.Sequence>(sequence: Sequence) -> Observable<Element>
where Sequence.Element == Observable<Element> {
CatchSequence(sources: sequence)
}
}
extension ObservableType {
/**
Repeats the source observable sequence until it successfully terminates.
**This could potentially create an infinite sequence.**
- seealso: [retry operator on reactivex.io](http://reactivex.io/documentation/operators/retry.html)
- returns: Observable sequence to repeat until it successfully terminates.
*/
public func retry() -> Observable<Element> {
CatchSequence(sources: InfiniteSequence(repeatedValue: self.asObservable()))
}
/**
Repeats the source observable sequence the specified number of times in case of an error or until it successfully terminates.
If you encounter an error and want it to retry once, then you must use `retry(2)`
- seealso: [retry operator on reactivex.io](http://reactivex.io/documentation/operators/retry.html)
- parameter maxAttemptCount: Maximum number of times to repeat the sequence.
- returns: An observable sequence producing the elements of the given sequence repeatedly until it terminates successfully.
*/
public func retry(_ maxAttemptCount: Int)
-> Observable<Element> {
CatchSequence(sources: Swift.repeatElement(self.asObservable(), count: maxAttemptCount))
}
}
// catch with callback
final private class CatchSinkProxy<Observer: ObserverType>: ObserverType {
typealias Element = Observer.Element
typealias Parent = CatchSink<Observer>
private let parent: Parent
init(parent: Parent) {
self.parent = parent
}
func on(_ event: Event<Element>) {
self.parent.forwardOn(event)
switch event {
case .next:
break
case .error, .completed:
self.parent.dispose()
}
}
}
final private class CatchSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = Catch<Element>
private let parent: Parent
private let subscription = SerialDisposable()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
let d1 = SingleAssignmentDisposable()
self.subscription.disposable = d1
d1.setDisposable(self.parent.source.subscribe(self))
return self.subscription
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.forwardOn(event)
case .completed:
self.forwardOn(event)
self.dispose()
case .error(let error):
do {
let catchSequence = try self.parent.handler(error)
let observer = CatchSinkProxy(parent: self)
self.subscription.disposable = catchSequence.subscribe(observer)
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
}
}
}
final private class Catch<Element>: Producer<Element> {
typealias Handler = (Swift.Error) throws -> Observable<Element>
fileprivate let source: Observable<Element>
fileprivate let handler: Handler
init(source: Observable<Element>, handler: @escaping Handler) {
self.source = source
self.handler = handler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = CatchSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// catch enumerable
final private class CatchSequenceSink<Sequence: Swift.Sequence, Observer: ObserverType>
: TailRecursiveSink<Sequence, Observer>
, ObserverType where Sequence.Element: ObservableConvertibleType, Sequence.Element.Element == Observer.Element {
typealias Element = Observer.Element
typealias Parent = CatchSequence<Sequence>
private var lastError: Swift.Error?
override init(observer: Observer, cancel: Cancelable) {
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.forwardOn(event)
case .error(let error):
self.lastError = error
self.schedule(.moveNext)
case .completed:
self.forwardOn(event)
self.dispose()
}
}
override func subscribeToNext(_ source: Observable<Element>) -> Disposable {
source.subscribe(self)
}
override func done() {
if let lastError = self.lastError {
self.forwardOn(.error(lastError))
}
else {
self.forwardOn(.completed)
}
self.dispose()
}
override func extract(_ observable: Observable<Element>) -> SequenceGenerator? {
if let onError = observable as? CatchSequence<Sequence> {
return (onError.sources.makeIterator(), nil)
}
else {
return nil
}
}
}
final private class CatchSequence<Sequence: Swift.Sequence>: Producer<Sequence.Element.Element> where Sequence.Element: ObservableConvertibleType {
typealias Element = Sequence.Element.Element
let sources: Sequence
init(sources: Sequence) {
self.sources = sources
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = CatchSequenceSink<Sequence, Observer>(observer: observer, cancel: cancel)
let subscription = sink.run((self.sources.makeIterator(), nil))
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,165 @@
//
// CombineLatest+Collection.swift
// RxSwift
//
// Created by Krunoslav Zaher on 8/29/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combinelatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<Collection: Swift.Collection>(_ collection: Collection, resultSelector: @escaping ([Collection.Element.Element]) throws -> Element) -> Observable<Element>
where Collection.Element: ObservableType {
CombineLatestCollectionType(sources: collection, resultSelector: resultSelector)
}
/**
Merges the specified observable sequences into one observable sequence whenever any of the observable sequences produces an element.
- seealso: [combinelatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<Collection: Swift.Collection>(_ collection: Collection) -> Observable<[Element]>
where Collection.Element: ObservableType, Collection.Element.Element == Element {
CombineLatestCollectionType(sources: collection, resultSelector: { $0 })
}
}
final private class CombineLatestCollectionTypeSink<Collection: Swift.Collection, Observer: ObserverType>
: Sink<Observer> where Collection.Element: ObservableConvertibleType {
typealias Result = Observer.Element
typealias Parent = CombineLatestCollectionType<Collection, Result>
typealias SourceElement = Collection.Element.Element
let parent: Parent
let lock = RecursiveLock()
// state
var numberOfValues = 0
var values: [SourceElement?]
var isDone: [Bool]
var numberOfDone = 0
var subscriptions: [SingleAssignmentDisposable]
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.values = [SourceElement?](repeating: nil, count: parent.count)
self.isDone = [Bool](repeating: false, count: parent.count)
self.subscriptions = [SingleAssignmentDisposable]()
self.subscriptions.reserveCapacity(parent.count)
for _ in 0 ..< parent.count {
self.subscriptions.append(SingleAssignmentDisposable())
}
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<SourceElement>, atIndex: Int) {
self.lock.lock(); defer { self.lock.unlock() }
switch event {
case .next(let element):
if self.values[atIndex] == nil {
self.numberOfValues += 1
}
self.values[atIndex] = element
if self.numberOfValues < self.parent.count {
let numberOfOthersThatAreDone = self.numberOfDone - (self.isDone[atIndex] ? 1 : 0)
if numberOfOthersThatAreDone == self.parent.count - 1 {
self.forwardOn(.completed)
self.dispose()
}
return
}
do {
let result = try self.parent.resultSelector(self.values.map { $0! })
self.forwardOn(.next(result))
}
catch let error {
self.forwardOn(.error(error))
self.dispose()
}
case .error(let error):
self.forwardOn(.error(error))
self.dispose()
case .completed:
if self.isDone[atIndex] {
return
}
self.isDone[atIndex] = true
self.numberOfDone += 1
if self.numberOfDone == self.parent.count {
self.forwardOn(.completed)
self.dispose()
}
else {
self.subscriptions[atIndex].dispose()
}
}
}
func run() -> Disposable {
var j = 0
for i in self.parent.sources {
let index = j
let source = i.asObservable()
let disposable = source.subscribe(AnyObserver { event in
self.on(event, atIndex: index)
})
self.subscriptions[j].setDisposable(disposable)
j += 1
}
if self.parent.sources.isEmpty {
do {
let result = try self.parent.resultSelector([])
self.forwardOn(.next(result))
self.forwardOn(.completed)
self.dispose()
}
catch let error {
self.forwardOn(.error(error))
self.dispose()
}
}
return Disposables.create(subscriptions)
}
}
final private class CombineLatestCollectionType<Collection: Swift.Collection, Result>: Producer<Result> where Collection.Element: ObservableConvertibleType {
typealias ResultSelector = ([Collection.Element.Element]) throws -> Result
let sources: Collection
let resultSelector: ResultSelector
let count: Int
init(sources: Collection, resultSelector: @escaping ResultSelector) {
self.sources = sources
self.resultSelector = resultSelector
self.count = self.sources.count
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestCollectionTypeSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,843 @@
// This file is autogenerated. Take a look at `Preprocessor` target in RxSwift project
//
// CombineLatest+arity.swift
// RxSwift
//
// Created by Krunoslav Zaher on 4/22/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
// 2
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType>
(_ source1: O1, _ source2: O2, resultSelector: @escaping (O1.Element, O2.Element) throws -> Element)
-> Observable<Element> {
return CombineLatest2(
source1: source1.asObservable(), source2: source2.asObservable(),
resultSelector: resultSelector
)
}
}
extension ObservableType where Element == Any {
/**
Merges the specified observable sequences into one observable sequence of tuples whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType>
(_ source1: O1, _ source2: O2)
-> Observable<(O1.Element, O2.Element)> {
return CombineLatest2(
source1: source1.asObservable(), source2: source2.asObservable(),
resultSelector: { ($0, $1) }
)
}
}
final class CombineLatestSink2_<E1, E2, Observer: ObserverType> : CombineLatestSink<Observer> {
typealias Result = Observer.Element
typealias Parent = CombineLatest2<E1, E2, Result>
let parent: Parent
var latestElement1: E1! = nil
var latestElement2: E2! = nil
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(arity: 2, observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let observer1 = CombineLatestObserver(lock: self.lock, parent: self, index: 0, setLatestValue: { (e: E1) -> Void in self.latestElement1 = e }, this: subscription1)
let observer2 = CombineLatestObserver(lock: self.lock, parent: self, index: 1, setLatestValue: { (e: E2) -> Void in self.latestElement2 = e }, this: subscription2)
subscription1.setDisposable(self.parent.source1.subscribe(observer1))
subscription2.setDisposable(self.parent.source2.subscribe(observer2))
return Disposables.create([
subscription1,
subscription2
])
}
override func getResult() throws -> Result {
try self.parent.resultSelector(self.latestElement1, self.latestElement2)
}
}
final class CombineLatest2<E1, E2, Result> : Producer<Result> {
typealias ResultSelector = (E1, E2) throws -> Result
let source1: Observable<E1>
let source2: Observable<E2>
let resultSelector: ResultSelector
init(source1: Observable<E1>, source2: Observable<E2>, resultSelector: @escaping ResultSelector) {
self.source1 = source1
self.source2 = source2
self.resultSelector = resultSelector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestSink2_(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// 3
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, resultSelector: @escaping (O1.Element, O2.Element, O3.Element) throws -> Element)
-> Observable<Element> {
return CombineLatest3(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(),
resultSelector: resultSelector
)
}
}
extension ObservableType where Element == Any {
/**
Merges the specified observable sequences into one observable sequence of tuples whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3)
-> Observable<(O1.Element, O2.Element, O3.Element)> {
return CombineLatest3(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(),
resultSelector: { ($0, $1, $2) }
)
}
}
final class CombineLatestSink3_<E1, E2, E3, Observer: ObserverType> : CombineLatestSink<Observer> {
typealias Result = Observer.Element
typealias Parent = CombineLatest3<E1, E2, E3, Result>
let parent: Parent
var latestElement1: E1! = nil
var latestElement2: E2! = nil
var latestElement3: E3! = nil
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(arity: 3, observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let subscription3 = SingleAssignmentDisposable()
let observer1 = CombineLatestObserver(lock: self.lock, parent: self, index: 0, setLatestValue: { (e: E1) -> Void in self.latestElement1 = e }, this: subscription1)
let observer2 = CombineLatestObserver(lock: self.lock, parent: self, index: 1, setLatestValue: { (e: E2) -> Void in self.latestElement2 = e }, this: subscription2)
let observer3 = CombineLatestObserver(lock: self.lock, parent: self, index: 2, setLatestValue: { (e: E3) -> Void in self.latestElement3 = e }, this: subscription3)
subscription1.setDisposable(self.parent.source1.subscribe(observer1))
subscription2.setDisposable(self.parent.source2.subscribe(observer2))
subscription3.setDisposable(self.parent.source3.subscribe(observer3))
return Disposables.create([
subscription1,
subscription2,
subscription3
])
}
override func getResult() throws -> Result {
try self.parent.resultSelector(self.latestElement1, self.latestElement2, self.latestElement3)
}
}
final class CombineLatest3<E1, E2, E3, Result> : Producer<Result> {
typealias ResultSelector = (E1, E2, E3) throws -> Result
let source1: Observable<E1>
let source2: Observable<E2>
let source3: Observable<E3>
let resultSelector: ResultSelector
init(source1: Observable<E1>, source2: Observable<E2>, source3: Observable<E3>, resultSelector: @escaping ResultSelector) {
self.source1 = source1
self.source2 = source2
self.source3 = source3
self.resultSelector = resultSelector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestSink3_(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// 4
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, resultSelector: @escaping (O1.Element, O2.Element, O3.Element, O4.Element) throws -> Element)
-> Observable<Element> {
return CombineLatest4(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(),
resultSelector: resultSelector
)
}
}
extension ObservableType where Element == Any {
/**
Merges the specified observable sequences into one observable sequence of tuples whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4)
-> Observable<(O1.Element, O2.Element, O3.Element, O4.Element)> {
return CombineLatest4(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(),
resultSelector: { ($0, $1, $2, $3) }
)
}
}
final class CombineLatestSink4_<E1, E2, E3, E4, Observer: ObserverType> : CombineLatestSink<Observer> {
typealias Result = Observer.Element
typealias Parent = CombineLatest4<E1, E2, E3, E4, Result>
let parent: Parent
var latestElement1: E1! = nil
var latestElement2: E2! = nil
var latestElement3: E3! = nil
var latestElement4: E4! = nil
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(arity: 4, observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let subscription3 = SingleAssignmentDisposable()
let subscription4 = SingleAssignmentDisposable()
let observer1 = CombineLatestObserver(lock: self.lock, parent: self, index: 0, setLatestValue: { (e: E1) -> Void in self.latestElement1 = e }, this: subscription1)
let observer2 = CombineLatestObserver(lock: self.lock, parent: self, index: 1, setLatestValue: { (e: E2) -> Void in self.latestElement2 = e }, this: subscription2)
let observer3 = CombineLatestObserver(lock: self.lock, parent: self, index: 2, setLatestValue: { (e: E3) -> Void in self.latestElement3 = e }, this: subscription3)
let observer4 = CombineLatestObserver(lock: self.lock, parent: self, index: 3, setLatestValue: { (e: E4) -> Void in self.latestElement4 = e }, this: subscription4)
subscription1.setDisposable(self.parent.source1.subscribe(observer1))
subscription2.setDisposable(self.parent.source2.subscribe(observer2))
subscription3.setDisposable(self.parent.source3.subscribe(observer3))
subscription4.setDisposable(self.parent.source4.subscribe(observer4))
return Disposables.create([
subscription1,
subscription2,
subscription3,
subscription4
])
}
override func getResult() throws -> Result {
try self.parent.resultSelector(self.latestElement1, self.latestElement2, self.latestElement3, self.latestElement4)
}
}
final class CombineLatest4<E1, E2, E3, E4, Result> : Producer<Result> {
typealias ResultSelector = (E1, E2, E3, E4) throws -> Result
let source1: Observable<E1>
let source2: Observable<E2>
let source3: Observable<E3>
let source4: Observable<E4>
let resultSelector: ResultSelector
init(source1: Observable<E1>, source2: Observable<E2>, source3: Observable<E3>, source4: Observable<E4>, resultSelector: @escaping ResultSelector) {
self.source1 = source1
self.source2 = source2
self.source3 = source3
self.source4 = source4
self.resultSelector = resultSelector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestSink4_(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// 5
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5, resultSelector: @escaping (O1.Element, O2.Element, O3.Element, O4.Element, O5.Element) throws -> Element)
-> Observable<Element> {
return CombineLatest5(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(),
resultSelector: resultSelector
)
}
}
extension ObservableType where Element == Any {
/**
Merges the specified observable sequences into one observable sequence of tuples whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5)
-> Observable<(O1.Element, O2.Element, O3.Element, O4.Element, O5.Element)> {
return CombineLatest5(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(),
resultSelector: { ($0, $1, $2, $3, $4) }
)
}
}
final class CombineLatestSink5_<E1, E2, E3, E4, E5, Observer: ObserverType> : CombineLatestSink<Observer> {
typealias Result = Observer.Element
typealias Parent = CombineLatest5<E1, E2, E3, E4, E5, Result>
let parent: Parent
var latestElement1: E1! = nil
var latestElement2: E2! = nil
var latestElement3: E3! = nil
var latestElement4: E4! = nil
var latestElement5: E5! = nil
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(arity: 5, observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let subscription3 = SingleAssignmentDisposable()
let subscription4 = SingleAssignmentDisposable()
let subscription5 = SingleAssignmentDisposable()
let observer1 = CombineLatestObserver(lock: self.lock, parent: self, index: 0, setLatestValue: { (e: E1) -> Void in self.latestElement1 = e }, this: subscription1)
let observer2 = CombineLatestObserver(lock: self.lock, parent: self, index: 1, setLatestValue: { (e: E2) -> Void in self.latestElement2 = e }, this: subscription2)
let observer3 = CombineLatestObserver(lock: self.lock, parent: self, index: 2, setLatestValue: { (e: E3) -> Void in self.latestElement3 = e }, this: subscription3)
let observer4 = CombineLatestObserver(lock: self.lock, parent: self, index: 3, setLatestValue: { (e: E4) -> Void in self.latestElement4 = e }, this: subscription4)
let observer5 = CombineLatestObserver(lock: self.lock, parent: self, index: 4, setLatestValue: { (e: E5) -> Void in self.latestElement5 = e }, this: subscription5)
subscription1.setDisposable(self.parent.source1.subscribe(observer1))
subscription2.setDisposable(self.parent.source2.subscribe(observer2))
subscription3.setDisposable(self.parent.source3.subscribe(observer3))
subscription4.setDisposable(self.parent.source4.subscribe(observer4))
subscription5.setDisposable(self.parent.source5.subscribe(observer5))
return Disposables.create([
subscription1,
subscription2,
subscription3,
subscription4,
subscription5
])
}
override func getResult() throws -> Result {
try self.parent.resultSelector(self.latestElement1, self.latestElement2, self.latestElement3, self.latestElement4, self.latestElement5)
}
}
final class CombineLatest5<E1, E2, E3, E4, E5, Result> : Producer<Result> {
typealias ResultSelector = (E1, E2, E3, E4, E5) throws -> Result
let source1: Observable<E1>
let source2: Observable<E2>
let source3: Observable<E3>
let source4: Observable<E4>
let source5: Observable<E5>
let resultSelector: ResultSelector
init(source1: Observable<E1>, source2: Observable<E2>, source3: Observable<E3>, source4: Observable<E4>, source5: Observable<E5>, resultSelector: @escaping ResultSelector) {
self.source1 = source1
self.source2 = source2
self.source3 = source3
self.source4 = source4
self.source5 = source5
self.resultSelector = resultSelector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestSink5_(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// 6
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType, O6: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5, _ source6: O6, resultSelector: @escaping (O1.Element, O2.Element, O3.Element, O4.Element, O5.Element, O6.Element) throws -> Element)
-> Observable<Element> {
return CombineLatest6(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(), source6: source6.asObservable(),
resultSelector: resultSelector
)
}
}
extension ObservableType where Element == Any {
/**
Merges the specified observable sequences into one observable sequence of tuples whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType, O6: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5, _ source6: O6)
-> Observable<(O1.Element, O2.Element, O3.Element, O4.Element, O5.Element, O6.Element)> {
return CombineLatest6(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(), source6: source6.asObservable(),
resultSelector: { ($0, $1, $2, $3, $4, $5) }
)
}
}
final class CombineLatestSink6_<E1, E2, E3, E4, E5, E6, Observer: ObserverType> : CombineLatestSink<Observer> {
typealias Result = Observer.Element
typealias Parent = CombineLatest6<E1, E2, E3, E4, E5, E6, Result>
let parent: Parent
var latestElement1: E1! = nil
var latestElement2: E2! = nil
var latestElement3: E3! = nil
var latestElement4: E4! = nil
var latestElement5: E5! = nil
var latestElement6: E6! = nil
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(arity: 6, observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let subscription3 = SingleAssignmentDisposable()
let subscription4 = SingleAssignmentDisposable()
let subscription5 = SingleAssignmentDisposable()
let subscription6 = SingleAssignmentDisposable()
let observer1 = CombineLatestObserver(lock: self.lock, parent: self, index: 0, setLatestValue: { (e: E1) -> Void in self.latestElement1 = e }, this: subscription1)
let observer2 = CombineLatestObserver(lock: self.lock, parent: self, index: 1, setLatestValue: { (e: E2) -> Void in self.latestElement2 = e }, this: subscription2)
let observer3 = CombineLatestObserver(lock: self.lock, parent: self, index: 2, setLatestValue: { (e: E3) -> Void in self.latestElement3 = e }, this: subscription3)
let observer4 = CombineLatestObserver(lock: self.lock, parent: self, index: 3, setLatestValue: { (e: E4) -> Void in self.latestElement4 = e }, this: subscription4)
let observer5 = CombineLatestObserver(lock: self.lock, parent: self, index: 4, setLatestValue: { (e: E5) -> Void in self.latestElement5 = e }, this: subscription5)
let observer6 = CombineLatestObserver(lock: self.lock, parent: self, index: 5, setLatestValue: { (e: E6) -> Void in self.latestElement6 = e }, this: subscription6)
subscription1.setDisposable(self.parent.source1.subscribe(observer1))
subscription2.setDisposable(self.parent.source2.subscribe(observer2))
subscription3.setDisposable(self.parent.source3.subscribe(observer3))
subscription4.setDisposable(self.parent.source4.subscribe(observer4))
subscription5.setDisposable(self.parent.source5.subscribe(observer5))
subscription6.setDisposable(self.parent.source6.subscribe(observer6))
return Disposables.create([
subscription1,
subscription2,
subscription3,
subscription4,
subscription5,
subscription6
])
}
override func getResult() throws -> Result {
try self.parent.resultSelector(self.latestElement1, self.latestElement2, self.latestElement3, self.latestElement4, self.latestElement5, self.latestElement6)
}
}
final class CombineLatest6<E1, E2, E3, E4, E5, E6, Result> : Producer<Result> {
typealias ResultSelector = (E1, E2, E3, E4, E5, E6) throws -> Result
let source1: Observable<E1>
let source2: Observable<E2>
let source3: Observable<E3>
let source4: Observable<E4>
let source5: Observable<E5>
let source6: Observable<E6>
let resultSelector: ResultSelector
init(source1: Observable<E1>, source2: Observable<E2>, source3: Observable<E3>, source4: Observable<E4>, source5: Observable<E5>, source6: Observable<E6>, resultSelector: @escaping ResultSelector) {
self.source1 = source1
self.source2 = source2
self.source3 = source3
self.source4 = source4
self.source5 = source5
self.source6 = source6
self.resultSelector = resultSelector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestSink6_(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// 7
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType, O6: ObservableType, O7: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5, _ source6: O6, _ source7: O7, resultSelector: @escaping (O1.Element, O2.Element, O3.Element, O4.Element, O5.Element, O6.Element, O7.Element) throws -> Element)
-> Observable<Element> {
return CombineLatest7(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(), source6: source6.asObservable(), source7: source7.asObservable(),
resultSelector: resultSelector
)
}
}
extension ObservableType where Element == Any {
/**
Merges the specified observable sequences into one observable sequence of tuples whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType, O6: ObservableType, O7: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5, _ source6: O6, _ source7: O7)
-> Observable<(O1.Element, O2.Element, O3.Element, O4.Element, O5.Element, O6.Element, O7.Element)> {
return CombineLatest7(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(), source6: source6.asObservable(), source7: source7.asObservable(),
resultSelector: { ($0, $1, $2, $3, $4, $5, $6) }
)
}
}
final class CombineLatestSink7_<E1, E2, E3, E4, E5, E6, E7, Observer: ObserverType> : CombineLatestSink<Observer> {
typealias Result = Observer.Element
typealias Parent = CombineLatest7<E1, E2, E3, E4, E5, E6, E7, Result>
let parent: Parent
var latestElement1: E1! = nil
var latestElement2: E2! = nil
var latestElement3: E3! = nil
var latestElement4: E4! = nil
var latestElement5: E5! = nil
var latestElement6: E6! = nil
var latestElement7: E7! = nil
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(arity: 7, observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let subscription3 = SingleAssignmentDisposable()
let subscription4 = SingleAssignmentDisposable()
let subscription5 = SingleAssignmentDisposable()
let subscription6 = SingleAssignmentDisposable()
let subscription7 = SingleAssignmentDisposable()
let observer1 = CombineLatestObserver(lock: self.lock, parent: self, index: 0, setLatestValue: { (e: E1) -> Void in self.latestElement1 = e }, this: subscription1)
let observer2 = CombineLatestObserver(lock: self.lock, parent: self, index: 1, setLatestValue: { (e: E2) -> Void in self.latestElement2 = e }, this: subscription2)
let observer3 = CombineLatestObserver(lock: self.lock, parent: self, index: 2, setLatestValue: { (e: E3) -> Void in self.latestElement3 = e }, this: subscription3)
let observer4 = CombineLatestObserver(lock: self.lock, parent: self, index: 3, setLatestValue: { (e: E4) -> Void in self.latestElement4 = e }, this: subscription4)
let observer5 = CombineLatestObserver(lock: self.lock, parent: self, index: 4, setLatestValue: { (e: E5) -> Void in self.latestElement5 = e }, this: subscription5)
let observer6 = CombineLatestObserver(lock: self.lock, parent: self, index: 5, setLatestValue: { (e: E6) -> Void in self.latestElement6 = e }, this: subscription6)
let observer7 = CombineLatestObserver(lock: self.lock, parent: self, index: 6, setLatestValue: { (e: E7) -> Void in self.latestElement7 = e }, this: subscription7)
subscription1.setDisposable(self.parent.source1.subscribe(observer1))
subscription2.setDisposable(self.parent.source2.subscribe(observer2))
subscription3.setDisposable(self.parent.source3.subscribe(observer3))
subscription4.setDisposable(self.parent.source4.subscribe(observer4))
subscription5.setDisposable(self.parent.source5.subscribe(observer5))
subscription6.setDisposable(self.parent.source6.subscribe(observer6))
subscription7.setDisposable(self.parent.source7.subscribe(observer7))
return Disposables.create([
subscription1,
subscription2,
subscription3,
subscription4,
subscription5,
subscription6,
subscription7
])
}
override func getResult() throws -> Result {
try self.parent.resultSelector(self.latestElement1, self.latestElement2, self.latestElement3, self.latestElement4, self.latestElement5, self.latestElement6, self.latestElement7)
}
}
final class CombineLatest7<E1, E2, E3, E4, E5, E6, E7, Result> : Producer<Result> {
typealias ResultSelector = (E1, E2, E3, E4, E5, E6, E7) throws -> Result
let source1: Observable<E1>
let source2: Observable<E2>
let source3: Observable<E3>
let source4: Observable<E4>
let source5: Observable<E5>
let source6: Observable<E6>
let source7: Observable<E7>
let resultSelector: ResultSelector
init(source1: Observable<E1>, source2: Observable<E2>, source3: Observable<E3>, source4: Observable<E4>, source5: Observable<E5>, source6: Observable<E6>, source7: Observable<E7>, resultSelector: @escaping ResultSelector) {
self.source1 = source1
self.source2 = source2
self.source3 = source3
self.source4 = source4
self.source5 = source5
self.source6 = source6
self.source7 = source7
self.resultSelector = resultSelector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestSink7_(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// 8
extension ObservableType {
/**
Merges the specified observable sequences into one observable sequence by using the selector function whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- parameter resultSelector: Function to invoke whenever any of the sources produces an element.
- returns: An observable sequence containing the result of combining elements of the sources using the specified result selector function.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType, O6: ObservableType, O7: ObservableType, O8: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5, _ source6: O6, _ source7: O7, _ source8: O8, resultSelector: @escaping (O1.Element, O2.Element, O3.Element, O4.Element, O5.Element, O6.Element, O7.Element, O8.Element) throws -> Element)
-> Observable<Element> {
return CombineLatest8(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(), source6: source6.asObservable(), source7: source7.asObservable(), source8: source8.asObservable(),
resultSelector: resultSelector
)
}
}
extension ObservableType where Element == Any {
/**
Merges the specified observable sequences into one observable sequence of tuples whenever any of the observable sequences produces an element.
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
- returns: An observable sequence containing the result of combining elements of the sources.
*/
public static func combineLatest<O1: ObservableType, O2: ObservableType, O3: ObservableType, O4: ObservableType, O5: ObservableType, O6: ObservableType, O7: ObservableType, O8: ObservableType>
(_ source1: O1, _ source2: O2, _ source3: O3, _ source4: O4, _ source5: O5, _ source6: O6, _ source7: O7, _ source8: O8)
-> Observable<(O1.Element, O2.Element, O3.Element, O4.Element, O5.Element, O6.Element, O7.Element, O8.Element)> {
return CombineLatest8(
source1: source1.asObservable(), source2: source2.asObservable(), source3: source3.asObservable(), source4: source4.asObservable(), source5: source5.asObservable(), source6: source6.asObservable(), source7: source7.asObservable(), source8: source8.asObservable(),
resultSelector: { ($0, $1, $2, $3, $4, $5, $6, $7) }
)
}
}
final class CombineLatestSink8_<E1, E2, E3, E4, E5, E6, E7, E8, Observer: ObserverType> : CombineLatestSink<Observer> {
typealias Result = Observer.Element
typealias Parent = CombineLatest8<E1, E2, E3, E4, E5, E6, E7, E8, Result>
let parent: Parent
var latestElement1: E1! = nil
var latestElement2: E2! = nil
var latestElement3: E3! = nil
var latestElement4: E4! = nil
var latestElement5: E5! = nil
var latestElement6: E6! = nil
var latestElement7: E7! = nil
var latestElement8: E8! = nil
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(arity: 8, observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription1 = SingleAssignmentDisposable()
let subscription2 = SingleAssignmentDisposable()
let subscription3 = SingleAssignmentDisposable()
let subscription4 = SingleAssignmentDisposable()
let subscription5 = SingleAssignmentDisposable()
let subscription6 = SingleAssignmentDisposable()
let subscription7 = SingleAssignmentDisposable()
let subscription8 = SingleAssignmentDisposable()
let observer1 = CombineLatestObserver(lock: self.lock, parent: self, index: 0, setLatestValue: { (e: E1) -> Void in self.latestElement1 = e }, this: subscription1)
let observer2 = CombineLatestObserver(lock: self.lock, parent: self, index: 1, setLatestValue: { (e: E2) -> Void in self.latestElement2 = e }, this: subscription2)
let observer3 = CombineLatestObserver(lock: self.lock, parent: self, index: 2, setLatestValue: { (e: E3) -> Void in self.latestElement3 = e }, this: subscription3)
let observer4 = CombineLatestObserver(lock: self.lock, parent: self, index: 3, setLatestValue: { (e: E4) -> Void in self.latestElement4 = e }, this: subscription4)
let observer5 = CombineLatestObserver(lock: self.lock, parent: self, index: 4, setLatestValue: { (e: E5) -> Void in self.latestElement5 = e }, this: subscription5)
let observer6 = CombineLatestObserver(lock: self.lock, parent: self, index: 5, setLatestValue: { (e: E6) -> Void in self.latestElement6 = e }, this: subscription6)
let observer7 = CombineLatestObserver(lock: self.lock, parent: self, index: 6, setLatestValue: { (e: E7) -> Void in self.latestElement7 = e }, this: subscription7)
let observer8 = CombineLatestObserver(lock: self.lock, parent: self, index: 7, setLatestValue: { (e: E8) -> Void in self.latestElement8 = e }, this: subscription8)
subscription1.setDisposable(self.parent.source1.subscribe(observer1))
subscription2.setDisposable(self.parent.source2.subscribe(observer2))
subscription3.setDisposable(self.parent.source3.subscribe(observer3))
subscription4.setDisposable(self.parent.source4.subscribe(observer4))
subscription5.setDisposable(self.parent.source5.subscribe(observer5))
subscription6.setDisposable(self.parent.source6.subscribe(observer6))
subscription7.setDisposable(self.parent.source7.subscribe(observer7))
subscription8.setDisposable(self.parent.source8.subscribe(observer8))
return Disposables.create([
subscription1,
subscription2,
subscription3,
subscription4,
subscription5,
subscription6,
subscription7,
subscription8
])
}
override func getResult() throws -> Result {
try self.parent.resultSelector(self.latestElement1, self.latestElement2, self.latestElement3, self.latestElement4, self.latestElement5, self.latestElement6, self.latestElement7, self.latestElement8)
}
}
final class CombineLatest8<E1, E2, E3, E4, E5, E6, E7, E8, Result> : Producer<Result> {
typealias ResultSelector = (E1, E2, E3, E4, E5, E6, E7, E8) throws -> Result
let source1: Observable<E1>
let source2: Observable<E2>
let source3: Observable<E3>
let source4: Observable<E4>
let source5: Observable<E5>
let source6: Observable<E6>
let source7: Observable<E7>
let source8: Observable<E8>
let resultSelector: ResultSelector
init(source1: Observable<E1>, source2: Observable<E2>, source3: Observable<E3>, source4: Observable<E4>, source5: Observable<E5>, source6: Observable<E6>, source7: Observable<E7>, source8: Observable<E8>, resultSelector: @escaping ResultSelector) {
self.source1 = source1
self.source2 = source2
self.source3 = source3
self.source4 = source4
self.source5 = source5
self.source6 = source6
self.source7 = source7
self.source8 = source8
self.resultSelector = resultSelector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = CombineLatestSink8_(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,131 @@
//
// CombineLatest.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/21/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
protocol CombineLatestProtocol: AnyObject {
func next(_ index: Int)
func fail(_ error: Swift.Error)
func done(_ index: Int)
}
class CombineLatestSink<Observer: ObserverType>
: Sink<Observer>
, CombineLatestProtocol {
typealias Element = Observer.Element
let lock = RecursiveLock()
private let arity: Int
private var numberOfValues = 0
private var numberOfDone = 0
private var hasValue: [Bool]
private var isDone: [Bool]
init(arity: Int, observer: Observer, cancel: Cancelable) {
self.arity = arity
self.hasValue = [Bool](repeating: false, count: arity)
self.isDone = [Bool](repeating: false, count: arity)
super.init(observer: observer, cancel: cancel)
}
func getResult() throws -> Element {
rxAbstractMethod()
}
func next(_ index: Int) {
if !self.hasValue[index] {
self.hasValue[index] = true
self.numberOfValues += 1
}
if self.numberOfValues == self.arity {
do {
let result = try self.getResult()
self.forwardOn(.next(result))
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
}
else {
var allOthersDone = true
for i in 0 ..< self.arity {
if i != index && !self.isDone[i] {
allOthersDone = false
break
}
}
if allOthersDone {
self.forwardOn(.completed)
self.dispose()
}
}
}
func fail(_ error: Swift.Error) {
self.forwardOn(.error(error))
self.dispose()
}
func done(_ index: Int) {
if self.isDone[index] {
return
}
self.isDone[index] = true
self.numberOfDone += 1
if self.numberOfDone == self.arity {
self.forwardOn(.completed)
self.dispose()
}
}
}
final class CombineLatestObserver<Element>
: ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias ValueSetter = (Element) -> Void
private let parent: CombineLatestProtocol
let lock: RecursiveLock
private let index: Int
private let this: Disposable
private let setLatestValue: ValueSetter
init(lock: RecursiveLock, parent: CombineLatestProtocol, index: Int, setLatestValue: @escaping ValueSetter, this: Disposable) {
self.lock = lock
self.parent = parent
self.index = index
self.this = this
self.setLatestValue = setLatestValue
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next(let value):
self.setLatestValue(value)
self.parent.next(self.index)
case .error(let error):
self.this.dispose()
self.parent.fail(error)
case .completed:
self.this.dispose()
self.parent.done(self.index)
}
}
}

View File

@@ -0,0 +1,76 @@
//
// CompactMap.swift
// RxSwift
//
// Created by Michael Long on 04/09/2019.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Projects each element of an observable sequence into an optional form and filters all optional results.
- parameter transform: A transform function to apply to each source element and which returns an element or nil.
- returns: An observable sequence whose elements are the result of filtering the transform function for each element of the source.
*/
public func compactMap<Result>(_ transform: @escaping (Element) throws -> Result?)
-> Observable<Result> {
CompactMap(source: self.asObservable(), transform: transform)
}
}
final private class CompactMapSink<SourceType, Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Transform = (SourceType) throws -> ResultType?
typealias ResultType = Observer.Element
typealias Element = SourceType
private let transform: Transform
init(transform: @escaping Transform, observer: Observer, cancel: Cancelable) {
self.transform = transform
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<SourceType>) {
switch event {
case .next(let element):
do {
if let mappedElement = try self.transform(element) {
self.forwardOn(.next(mappedElement))
}
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
case .error(let error):
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class CompactMap<SourceType, ResultType>: Producer<ResultType> {
typealias Transform = (SourceType) throws -> ResultType?
private let source: Observable<SourceType>
private let transform: Transform
init(source: Observable<SourceType>, transform: @escaping Transform) {
self.source = source
self.transform = transform
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == ResultType {
let sink = CompactMapSink(transform: self.transform, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,131 @@
//
// Concat.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/21/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Concatenates the second observable sequence to `self` upon successful termination of `self`.
- seealso: [concat operator on reactivex.io](http://reactivex.io/documentation/operators/concat.html)
- parameter second: Second observable sequence.
- returns: An observable sequence that contains the elements of `self`, followed by those of the second sequence.
*/
public func concat<Source: ObservableConvertibleType>(_ second: Source) -> Observable<Element> where Source.Element == Element {
Observable.concat([self.asObservable(), second.asObservable()])
}
}
extension ObservableType {
/**
Concatenates all observable sequences in the given sequence, as long as the previous observable sequence terminated successfully.
This operator has tail recursive optimizations that will prevent stack overflow.
Optimizations will be performed in cases equivalent to following:
[1, [2, [3, .....].concat()].concat].concat()
- seealso: [concat operator on reactivex.io](http://reactivex.io/documentation/operators/concat.html)
- returns: An observable sequence that contains the elements of each given sequence, in sequential order.
*/
public static func concat<Sequence: Swift.Sequence>(_ sequence: Sequence) -> Observable<Element>
where Sequence.Element == Observable<Element> {
return Concat(sources: sequence, count: nil)
}
/**
Concatenates all observable sequences in the given collection, as long as the previous observable sequence terminated successfully.
This operator has tail recursive optimizations that will prevent stack overflow.
Optimizations will be performed in cases equivalent to following:
[1, [2, [3, .....].concat()].concat].concat()
- seealso: [concat operator on reactivex.io](http://reactivex.io/documentation/operators/concat.html)
- returns: An observable sequence that contains the elements of each given sequence, in sequential order.
*/
public static func concat<Collection: Swift.Collection>(_ collection: Collection) -> Observable<Element>
where Collection.Element == Observable<Element> {
return Concat(sources: collection, count: Int64(collection.count))
}
/**
Concatenates all observable sequences in the given collection, as long as the previous observable sequence terminated successfully.
This operator has tail recursive optimizations that will prevent stack overflow.
Optimizations will be performed in cases equivalent to following:
[1, [2, [3, .....].concat()].concat].concat()
- seealso: [concat operator on reactivex.io](http://reactivex.io/documentation/operators/concat.html)
- returns: An observable sequence that contains the elements of each given sequence, in sequential order.
*/
public static func concat(_ sources: Observable<Element> ...) -> Observable<Element> {
Concat(sources: sources, count: Int64(sources.count))
}
}
final private class ConcatSink<Sequence: Swift.Sequence, Observer: ObserverType>
: TailRecursiveSink<Sequence, Observer>
, ObserverType where Sequence.Element: ObservableConvertibleType, Sequence.Element.Element == Observer.Element {
typealias Element = Observer.Element
override init(observer: Observer, cancel: Cancelable) {
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>){
switch event {
case .next:
self.forwardOn(event)
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
self.schedule(.moveNext)
}
}
override func subscribeToNext(_ source: Observable<Element>) -> Disposable {
source.subscribe(self)
}
override func extract(_ observable: Observable<Element>) -> SequenceGenerator? {
if let source = observable as? Concat<Sequence> {
return (source.sources.makeIterator(), source.count)
}
else {
return nil
}
}
}
final private class Concat<Sequence: Swift.Sequence>: Producer<Sequence.Element.Element> where Sequence.Element: ObservableConvertibleType {
typealias Element = Sequence.Element.Element
fileprivate let sources: Sequence
fileprivate let count: IntMax?
init(sources: Sequence, count: IntMax?) {
self.sources = sources
self.count = count
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = ConcatSink<Sequence, Observer>(observer: observer, cancel: cancel)
let subscription = sink.run((self.sources.makeIterator(), self.count))
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,78 @@
//
// Create.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/8/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
// MARK: create
/**
Creates an observable sequence from a specified subscribe method implementation.
- seealso: [create operator on reactivex.io](http://reactivex.io/documentation/operators/create.html)
- parameter subscribe: Implementation of the resulting observable sequence's `subscribe` method.
- returns: The observable sequence with the specified implementation for the `subscribe` method.
*/
public static func create(_ subscribe: @escaping (AnyObserver<Element>) -> Disposable) -> Observable<Element> {
AnonymousObservable(subscribe)
}
}
final private class AnonymousObservableSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = AnonymousObservable<Element>
// state
private let isStopped = AtomicInt(0)
#if DEBUG
private let synchronizationTracker = SynchronizationTracker()
#endif
override init(observer: Observer, cancel: Cancelable) {
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
#if DEBUG
self.synchronizationTracker.register(synchronizationErrorMessage: .default)
defer { self.synchronizationTracker.unregister() }
#endif
switch event {
case .next:
if load(self.isStopped) == 1 {
return
}
self.forwardOn(event)
case .error, .completed:
if fetchOr(self.isStopped, 1) == 0 {
self.forwardOn(event)
self.dispose()
}
}
}
func run(_ parent: Parent) -> Disposable {
parent.subscribeHandler(AnyObserver(self))
}
}
final private class AnonymousObservable<Element>: Producer<Element> {
typealias SubscribeHandler = (AnyObserver<Element>) -> Disposable
let subscribeHandler: SubscribeHandler
init(_ subscribeHandler: @escaping SubscribeHandler) {
self.subscribeHandler = subscribeHandler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = AnonymousObservableSink(observer: observer, cancel: cancel)
let subscription = sink.run(self)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,119 @@
//
// Debounce.swift
// RxSwift
//
// Created by Krunoslav Zaher on 9/11/16.
// Copyright © 2016 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Ignores elements from an observable sequence which are followed by another element within a specified relative time duration, using the specified scheduler to run throttling timers.
- seealso: [debounce operator on reactivex.io](http://reactivex.io/documentation/operators/debounce.html)
- parameter dueTime: Throttling duration for each element.
- parameter scheduler: Scheduler to run the throttle timers on.
- returns: The throttled sequence.
*/
public func debounce(_ dueTime: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
return Debounce(source: self.asObservable(), dueTime: dueTime, scheduler: scheduler)
}
}
final private class DebounceSink<Observer: ObserverType>
: Sink<Observer>
, ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias Element = Observer.Element
typealias ParentType = Debounce<Element>
private let parent: ParentType
let lock = RecursiveLock()
// state
private var id = 0 as UInt64
private var value: Element?
let cancellable = SerialDisposable()
init(parent: ParentType, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription = self.parent.source.subscribe(self)
return Disposables.create(subscription, cancellable)
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next(let element):
self.id = self.id &+ 1
let currentId = self.id
self.value = element
let scheduler = self.parent.scheduler
let dueTime = self.parent.dueTime
let d = SingleAssignmentDisposable()
self.cancellable.disposable = d
d.setDisposable(scheduler.scheduleRelative(currentId, dueTime: dueTime, action: self.propagate))
case .error:
self.value = nil
self.forwardOn(event)
self.dispose()
case .completed:
if let value = self.value {
self.value = nil
self.forwardOn(.next(value))
}
self.forwardOn(.completed)
self.dispose()
}
}
func propagate(_ currentId: UInt64) -> Disposable {
self.lock.performLocked {
let originalValue = self.value
if let value = originalValue, self.id == currentId {
self.value = nil
self.forwardOn(.next(value))
}
return Disposables.create()
}
}
}
final private class Debounce<Element>: Producer<Element> {
fileprivate let source: Observable<Element>
fileprivate let dueTime: RxTimeInterval
fileprivate let scheduler: SchedulerType
init(source: Observable<Element>, dueTime: RxTimeInterval, scheduler: SchedulerType) {
self.source = source
self.dueTime = dueTime
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = DebounceSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,102 @@
//
// Debug.swift
// RxSwift
//
// Created by Krunoslav Zaher on 5/2/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
import Foundation
extension ObservableType {
/**
Prints received events for all observers on standard output.
- seealso: [do operator on reactivex.io](http://reactivex.io/documentation/operators/do.html)
- parameter identifier: Identifier that is printed together with event description to standard output.
- parameter trimOutput: Should output be trimmed to max 40 characters.
- returns: An observable sequence whose events are printed to standard output.
*/
public func debug(_ identifier: String? = nil, trimOutput: Bool = false, file: String = #file, line: UInt = #line, function: String = #function)
-> Observable<Element> {
return Debug(source: self, identifier: identifier, trimOutput: trimOutput, file: file, line: line, function: function)
}
}
private let dateFormat = "yyyy-MM-dd HH:mm:ss.SSS"
private func logEvent(_ identifier: String, dateFormat: DateFormatter, content: String) {
print("\(dateFormat.string(from: Date())): \(identifier) -> \(content)")
}
final private class DebugSink<Source: ObservableType, Observer: ObserverType>: Sink<Observer>, ObserverType where Observer.Element == Source.Element {
typealias Element = Observer.Element
typealias Parent = Debug<Source>
private let parent: Parent
private let timestampFormatter = DateFormatter()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.timestampFormatter.dateFormat = dateFormat
logEvent(self.parent.identifier, dateFormat: self.timestampFormatter, content: "subscribed")
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
let maxEventTextLength = 40
let eventText = "\(event)"
let eventNormalized = (eventText.count > maxEventTextLength) && self.parent.trimOutput
? String(eventText.prefix(maxEventTextLength / 2)) + "..." + String(eventText.suffix(maxEventTextLength / 2))
: eventText
logEvent(self.parent.identifier, dateFormat: self.timestampFormatter, content: "Event \(eventNormalized)")
self.forwardOn(event)
if event.isStopEvent {
self.dispose()
}
}
override func dispose() {
if !self.isDisposed {
logEvent(self.parent.identifier, dateFormat: self.timestampFormatter, content: "isDisposed")
}
super.dispose()
}
}
final private class Debug<Source: ObservableType>: Producer<Source.Element> {
fileprivate let identifier: String
fileprivate let trimOutput: Bool
private let source: Source
init(source: Source, identifier: String?, trimOutput: Bool, file: String, line: UInt, function: String) {
self.trimOutput = trimOutput
if let identifier = identifier {
self.identifier = identifier
}
else {
let trimmedFile: String
if let lastIndex = file.lastIndex(of: "/") {
trimmedFile = String(file[file.index(after: lastIndex) ..< file.endIndex])
}
else {
trimmedFile = file
}
self.identifier = "\(trimmedFile):\(line) (\(function))"
}
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Source.Element {
let sink = DebugSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,34 @@
//
// Decode.swift
// RxSwift
//
// Created by Shai Mishali on 24/07/2020.
// Copyright © 2020 Krunoslav Zaher. All rights reserved.
//
import Foundation
public extension ObservableType where Element == Data {
/// Attempt to decode the emitted `Data` using a provided decoder.
///
/// - parameter type: A `Decodable`-conforming type to attempt to decode to
/// - parameter decoder: A capable decoder, e.g. `JSONDecoder` or `PropertyListDecoder`
///
/// - note: If using a custom decoder, it must conform to the `DataDecoder` protocol.
///
/// - returns: An `Observable` of the decoded type
func decode<Item: Decodable,
Decoder: DataDecoder>(type: Item.Type,
decoder: Decoder) -> Observable<Item> {
map { try decoder.decode(type, from: $0) }
}
}
/// Represents an entity capable of decoding raw `Data`
/// into a concrete `Decodable` type
public protocol DataDecoder {
func decode<Item: Decodable>(_ type: Item.Type, from data: Data) throws -> Item
}
extension JSONDecoder: DataDecoder {}
extension PropertyListDecoder: DataDecoder {}

View File

@@ -0,0 +1,66 @@
//
// DefaultIfEmpty.swift
// RxSwift
//
// Created by sergdort on 23/12/2016.
// Copyright © 2016 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Emits elements from the source observable sequence, or a default element if the source observable sequence is empty.
- seealso: [DefaultIfEmpty operator on reactivex.io](http://reactivex.io/documentation/operators/defaultifempty.html)
- parameter default: Default element to be sent if the source does not emit any elements
- returns: An observable sequence which emits default element end completes in case the original sequence is empty
*/
public func ifEmpty(default: Element) -> Observable<Element> {
DefaultIfEmpty(source: self.asObservable(), default: `default`)
}
}
final private class DefaultIfEmptySink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
private let `default`: Element
private var isEmpty = true
init(default: Element, observer: Observer, cancel: Cancelable) {
self.default = `default`
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.isEmpty = false
self.forwardOn(event)
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
if self.isEmpty {
self.forwardOn(.next(self.default))
}
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class DefaultIfEmpty<SourceType>: Producer<SourceType> {
private let source: Observable<SourceType>
private let `default`: SourceType
init(source: Observable<SourceType>, `default`: SourceType) {
self.source = source
self.default = `default`
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == SourceType {
let sink = DefaultIfEmptySink(default: self.default, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,75 @@
//
// Deferred.swift
// RxSwift
//
// Created by Krunoslav Zaher on 4/19/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns an observable sequence that invokes the specified factory function whenever a new observer subscribes.
- seealso: [defer operator on reactivex.io](http://reactivex.io/documentation/operators/defer.html)
- parameter observableFactory: Observable factory function to invoke for each observer that subscribes to the resulting sequence.
- returns: An observable sequence whose observers trigger an invocation of the given observable factory function.
*/
public static func deferred(_ observableFactory: @escaping () throws -> Observable<Element>)
-> Observable<Element> {
Deferred(observableFactory: observableFactory)
}
}
final private class DeferredSink<Source: ObservableType, Observer: ObserverType>: Sink<Observer>, ObserverType where Source.Element == Observer.Element {
typealias Element = Observer.Element
private let observableFactory: () throws -> Source
init(observableFactory: @escaping () throws -> Source, observer: Observer, cancel: Cancelable) {
self.observableFactory = observableFactory
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
do {
let result = try self.observableFactory()
return result.subscribe(self)
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
return Disposables.create()
}
}
func on(_ event: Event<Element>) {
self.forwardOn(event)
switch event {
case .next:
break
case .error:
self.dispose()
case .completed:
self.dispose()
}
}
}
final private class Deferred<Source: ObservableType>: Producer<Source.Element> {
typealias Factory = () throws -> Source
private let observableFactory : Factory
init(observableFactory: @escaping Factory) {
self.observableFactory = observableFactory
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable)
where Observer.Element == Source.Element {
let sink = DeferredSink(observableFactory: self.observableFactory, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,174 @@
//
// Delay.swift
// RxSwift
//
// Created by tarunon on 2016/02/09.
// Copyright © 2016 Krunoslav Zaher. All rights reserved.
//
import Foundation
extension ObservableType {
/**
Returns an observable sequence by the source observable sequence shifted forward in time by a specified delay. Error events from the source observable sequence are not delayed.
- seealso: [delay operator on reactivex.io](http://reactivex.io/documentation/operators/delay.html)
- parameter dueTime: Relative time shift of the source by.
- parameter scheduler: Scheduler to run the subscription delay timer on.
- returns: the source Observable shifted in time by the specified delay.
*/
public func delay(_ dueTime: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
return Delay(source: self.asObservable(), dueTime: dueTime, scheduler: scheduler)
}
}
final private class DelaySink<Observer: ObserverType>
: Sink<Observer>
, ObserverType {
typealias Element = Observer.Element
typealias Source = Observable<Element>
typealias DisposeKey = Bag<Disposable>.KeyType
private let lock = RecursiveLock()
private let dueTime: RxTimeInterval
private let scheduler: SchedulerType
private let sourceSubscription = SingleAssignmentDisposable()
private let cancelable = SerialDisposable()
// is scheduled some action
private var active = false
// is "run loop" on different scheduler running
private var running = false
private var errorEvent: Event<Element>?
// state
private var queue = Queue<(eventTime: RxTime, event: Event<Element>)>(capacity: 0)
init(observer: Observer, dueTime: RxTimeInterval, scheduler: SchedulerType, cancel: Cancelable) {
self.dueTime = dueTime
self.scheduler = scheduler
super.init(observer: observer, cancel: cancel)
}
// All of these complications in this method are caused by the fact that
// error should be propagated immediately. Error can be potentially received on different
// scheduler so this process needs to be synchronized somehow.
//
// Another complication is that scheduler is potentially concurrent so internal queue is used.
func drainQueue(state: (), scheduler: AnyRecursiveScheduler<()>) {
self.lock.lock()
let hasFailed = self.errorEvent != nil
if !hasFailed {
self.running = true
}
self.lock.unlock()
if hasFailed {
return
}
var ranAtLeastOnce = false
while true {
self.lock.lock()
let errorEvent = self.errorEvent
let eventToForwardImmediately = ranAtLeastOnce ? nil : self.queue.dequeue()?.event
let nextEventToScheduleOriginalTime: Date? = ranAtLeastOnce && !self.queue.isEmpty ? self.queue.peek().eventTime : nil
if errorEvent == nil {
if eventToForwardImmediately != nil {
}
else if nextEventToScheduleOriginalTime != nil {
self.running = false
}
else {
self.running = false
self.active = false
}
}
self.lock.unlock()
if let errorEvent = errorEvent {
self.forwardOn(errorEvent)
self.dispose()
return
}
else {
if let eventToForwardImmediately = eventToForwardImmediately {
ranAtLeastOnce = true
self.forwardOn(eventToForwardImmediately)
if case .completed = eventToForwardImmediately {
self.dispose()
return
}
}
else if let nextEventToScheduleOriginalTime = nextEventToScheduleOriginalTime {
scheduler.schedule((), dueTime: self.dueTime.reduceWithSpanBetween(earlierDate: nextEventToScheduleOriginalTime, laterDate: self.scheduler.now))
return
}
else {
return
}
}
}
}
func on(_ event: Event<Element>) {
if event.isStopEvent {
self.sourceSubscription.dispose()
}
switch event {
case .error:
self.lock.lock()
let shouldSendImmediately = !self.running
self.queue = Queue(capacity: 0)
self.errorEvent = event
self.lock.unlock()
if shouldSendImmediately {
self.forwardOn(event)
self.dispose()
}
default:
self.lock.lock()
let shouldSchedule = !self.active
self.active = true
self.queue.enqueue((self.scheduler.now, event))
self.lock.unlock()
if shouldSchedule {
self.cancelable.disposable = self.scheduler.scheduleRecursive((), dueTime: self.dueTime, action: self.drainQueue)
}
}
}
func run(source: Observable<Element>) -> Disposable {
self.sourceSubscription.setDisposable(source.subscribe(self))
return Disposables.create(sourceSubscription, cancelable)
}
}
final private class Delay<Element>: Producer<Element> {
private let source: Observable<Element>
private let dueTime: RxTimeInterval
private let scheduler: SchedulerType
init(source: Observable<Element>, dueTime: RxTimeInterval, scheduler: SchedulerType) {
self.source = source
self.dueTime = dueTime
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = DelaySink(observer: observer, dueTime: self.dueTime, scheduler: self.scheduler, cancel: cancel)
let subscription = sink.run(source: self.source)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,58 @@
//
// DelaySubscription.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/14/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Time shifts the observable sequence by delaying the subscription with the specified relative time duration, using the specified scheduler to run timers.
- seealso: [delay operator on reactivex.io](http://reactivex.io/documentation/operators/delay.html)
- parameter dueTime: Relative time shift of the subscription.
- parameter scheduler: Scheduler to run the subscription delay timer on.
- returns: Time-shifted sequence.
*/
public func delaySubscription(_ dueTime: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
DelaySubscription(source: self.asObservable(), dueTime: dueTime, scheduler: scheduler)
}
}
final private class DelaySubscriptionSink<Observer: ObserverType>
: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
func on(_ event: Event<Element>) {
self.forwardOn(event)
if event.isStopEvent {
self.dispose()
}
}
}
final private class DelaySubscription<Element>: Producer<Element> {
private let source: Observable<Element>
private let dueTime: RxTimeInterval
private let scheduler: SchedulerType
init(source: Observable<Element>, dueTime: RxTimeInterval, scheduler: SchedulerType) {
self.source = source
self.dueTime = dueTime
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = DelaySubscriptionSink(observer: observer, cancel: cancel)
let subscription = self.scheduler.scheduleRelative((), dueTime: self.dueTime) { _ in
return self.source.subscribe(sink)
}
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,51 @@
//
// Dematerialize.swift
// RxSwift
//
// Created by Jamie Pinkham on 3/13/17.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
extension ObservableType where Element: EventConvertible {
/**
Convert any previously materialized Observable into it's original form.
- seealso: [materialize operator on reactivex.io](http://reactivex.io/documentation/operators/materialize-dematerialize.html)
- returns: The dematerialized observable sequence.
*/
public func dematerialize() -> Observable<Element.Element> {
Dematerialize(source: self.asObservable())
}
}
private final class DematerializeSink<T: EventConvertible, Observer: ObserverType>: Sink<Observer>, ObserverType where Observer.Element == T.Element {
fileprivate func on(_ event: Event<T>) {
switch event {
case .next(let element):
self.forwardOn(element.event)
if element.event.isStopEvent {
self.dispose()
}
case .completed:
self.forwardOn(.completed)
self.dispose()
case .error(let error):
self.forwardOn(.error(error))
self.dispose()
}
}
}
final private class Dematerialize<T: EventConvertible>: Producer<T.Element> {
private let source: Observable<T>
init(source: Observable<T>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == T.Element {
let sink = DematerializeSink<T, Observer>(observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,137 @@
//
// DistinctUntilChanged.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/15/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType where Element: Equatable {
/**
Returns an observable sequence that contains only distinct contiguous elements according to equality operator.
- seealso: [distinct operator on reactivex.io](http://reactivex.io/documentation/operators/distinct.html)
- returns: An observable sequence only containing the distinct contiguous elements, based on equality operator, from the source sequence.
*/
public func distinctUntilChanged()
-> Observable<Element> {
self.distinctUntilChanged({ $0 }, comparer: { ($0 == $1) })
}
}
extension ObservableType {
/**
Returns an observable sequence that contains only distinct contiguous elements according to the `keySelector`.
- seealso: [distinct operator on reactivex.io](http://reactivex.io/documentation/operators/distinct.html)
- parameter keySelector: A function to compute the comparison key for each element.
- returns: An observable sequence only containing the distinct contiguous elements, based on a computed key value, from the source sequence.
*/
public func distinctUntilChanged<Key: Equatable>(_ keySelector: @escaping (Element) throws -> Key)
-> Observable<Element> {
self.distinctUntilChanged(keySelector, comparer: { $0 == $1 })
}
/**
Returns an observable sequence that contains only distinct contiguous elements according to the `comparer`.
- seealso: [distinct operator on reactivex.io](http://reactivex.io/documentation/operators/distinct.html)
- parameter comparer: Equality comparer for computed key values.
- returns: An observable sequence only containing the distinct contiguous elements, based on `comparer`, from the source sequence.
*/
public func distinctUntilChanged(_ comparer: @escaping (Element, Element) throws -> Bool)
-> Observable<Element> {
self.distinctUntilChanged({ $0 }, comparer: comparer)
}
/**
Returns an observable sequence that contains only distinct contiguous elements according to the keySelector and the comparer.
- seealso: [distinct operator on reactivex.io](http://reactivex.io/documentation/operators/distinct.html)
- parameter keySelector: A function to compute the comparison key for each element.
- parameter comparer: Equality comparer for computed key values.
- returns: An observable sequence only containing the distinct contiguous elements, based on a computed key value and the comparer, from the source sequence.
*/
public func distinctUntilChanged<K>(_ keySelector: @escaping (Element) throws -> K, comparer: @escaping (K, K) throws -> Bool)
-> Observable<Element> {
return DistinctUntilChanged(source: self.asObservable(), selector: keySelector, comparer: comparer)
}
/**
Returns an observable sequence that contains only contiguous elements with distinct values in the provided key path on each object.
- seealso: [distinct operator on reactivex.io](http://reactivex.io/documentation/operators/distinct.html)
- returns: An observable sequence only containing the distinct contiguous elements, based on equality operator on the provided key path
*/
public func distinctUntilChanged<Property: Equatable>(at keyPath: KeyPath<Element, Property>) ->
Observable<Element> {
self.distinctUntilChanged { $0[keyPath: keyPath] == $1[keyPath: keyPath] }
}
}
final private class DistinctUntilChangedSink<Observer: ObserverType, Key>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
private let parent: DistinctUntilChanged<Element, Key>
private var currentKey: Key?
init(parent: DistinctUntilChanged<Element, Key>, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
do {
let key = try self.parent.selector(value)
var areEqual = false
if let currentKey = self.currentKey {
areEqual = try self.parent.comparer(currentKey, key)
}
if areEqual {
return
}
self.currentKey = key
self.forwardOn(event)
}
catch let error {
self.forwardOn(.error(error))
self.dispose()
}
case .error, .completed:
self.forwardOn(event)
self.dispose()
}
}
}
final private class DistinctUntilChanged<Element, Key>: Producer<Element> {
typealias KeySelector = (Element) throws -> Key
typealias EqualityComparer = (Key, Key) throws -> Bool
private let source: Observable<Element>
fileprivate let selector: KeySelector
fileprivate let comparer: EqualityComparer
init(source: Observable<Element>, selector: @escaping KeySelector, comparer: @escaping EqualityComparer) {
self.source = source
self.selector = selector
self.comparer = comparer
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = DistinctUntilChangedSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

112
Pods/RxSwift/RxSwift/Observables/Do.swift generated Normal file
View File

@@ -0,0 +1,112 @@
//
// Do.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/21/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Invokes an action for each event in the observable sequence, and propagates all observer messages through the result sequence.
- seealso: [do operator on reactivex.io](http://reactivex.io/documentation/operators/do.html)
- parameter onNext: Action to invoke for each element in the observable sequence.
- parameter afterNext: Action to invoke for each element after the observable has passed an onNext event along to its downstream.
- parameter onError: Action to invoke upon errored termination of the observable sequence.
- parameter afterError: Action to invoke after errored termination of the observable sequence.
- parameter onCompleted: Action to invoke upon graceful termination of the observable sequence.
- parameter afterCompleted: Action to invoke after graceful termination of the observable sequence.
- parameter onSubscribe: Action to invoke before subscribing to source observable sequence.
- parameter onSubscribed: Action to invoke after subscribing to source observable sequence.
- parameter onDispose: Action to invoke after subscription to source observable has been disposed for any reason. It can be either because sequence terminates for some reason or observer subscription being disposed.
- returns: The source sequence with the side-effecting behavior applied.
*/
public func `do`(onNext: ((Element) throws -> Void)? = nil, afterNext: ((Element) throws -> Void)? = nil, onError: ((Swift.Error) throws -> Void)? = nil, afterError: ((Swift.Error) throws -> Void)? = nil, onCompleted: (() throws -> Void)? = nil, afterCompleted: (() throws -> Void)? = nil, onSubscribe: (() -> Void)? = nil, onSubscribed: (() -> Void)? = nil, onDispose: (() -> Void)? = nil)
-> Observable<Element> {
return Do(source: self.asObservable(), eventHandler: { e in
switch e {
case .next(let element):
try onNext?(element)
case .error(let e):
try onError?(e)
case .completed:
try onCompleted?()
}
}, afterEventHandler: { e in
switch e {
case .next(let element):
try afterNext?(element)
case .error(let e):
try afterError?(e)
case .completed:
try afterCompleted?()
}
}, onSubscribe: onSubscribe, onSubscribed: onSubscribed, onDispose: onDispose)
}
}
final private class DoSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias EventHandler = (Event<Element>) throws -> Void
typealias AfterEventHandler = (Event<Element>) throws -> Void
private let eventHandler: EventHandler
private let afterEventHandler: AfterEventHandler
init(eventHandler: @escaping EventHandler, afterEventHandler: @escaping AfterEventHandler, observer: Observer, cancel: Cancelable) {
self.eventHandler = eventHandler
self.afterEventHandler = afterEventHandler
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
do {
try self.eventHandler(event)
self.forwardOn(event)
try self.afterEventHandler(event)
if event.isStopEvent {
self.dispose()
}
}
catch let error {
self.forwardOn(.error(error))
self.dispose()
}
}
}
final private class Do<Element>: Producer<Element> {
typealias EventHandler = (Event<Element>) throws -> Void
typealias AfterEventHandler = (Event<Element>) throws -> Void
private let source: Observable<Element>
private let eventHandler: EventHandler
private let afterEventHandler: AfterEventHandler
private let onSubscribe: (() -> Void)?
private let onSubscribed: (() -> Void)?
private let onDispose: (() -> Void)?
init(source: Observable<Element>, eventHandler: @escaping EventHandler, afterEventHandler: @escaping AfterEventHandler, onSubscribe: (() -> Void)?, onSubscribed: (() -> Void)?, onDispose: (() -> Void)?) {
self.source = source
self.eventHandler = eventHandler
self.afterEventHandler = afterEventHandler
self.onSubscribe = onSubscribe
self.onSubscribed = onSubscribed
self.onDispose = onDispose
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
self.onSubscribe?()
let sink = DoSink(eventHandler: self.eventHandler, afterEventHandler: self.afterEventHandler, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
self.onSubscribed?()
let onDispose = self.onDispose
let allSubscriptions = Disposables.create {
subscription.dispose()
onDispose?()
}
return (sink: sink, subscription: allSubscriptions)
}
}

View File

@@ -0,0 +1,105 @@
//
// ElementAt.swift
// RxSwift
//
// Created by Junior B. on 21/10/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns a sequence emitting only element _n_ emitted by an Observable
- seealso: [elementAt operator on reactivex.io](http://reactivex.io/documentation/operators/elementat.html)
- parameter index: The index of the required element (starting from 0).
- returns: An observable sequence that emits the desired element as its own sole emission.
*/
@available(*, deprecated, renamed: "element(at:)")
public func elementAt(_ index: Int)
-> Observable<Element> {
element(at: index)
}
/**
Returns a sequence emitting only element _n_ emitted by an Observable
- seealso: [elementAt operator on reactivex.io](http://reactivex.io/documentation/operators/elementat.html)
- parameter index: The index of the required element (starting from 0).
- returns: An observable sequence that emits the desired element as its own sole emission.
*/
public func element(at index: Int)
-> Observable<Element> {
ElementAt(source: self.asObservable(), index: index, throwOnEmpty: true)
}
}
final private class ElementAtSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias SourceType = Observer.Element
typealias Parent = ElementAt<SourceType>
let parent: Parent
var i: Int
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.i = parent.index
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<SourceType>) {
switch event {
case .next:
if self.i == 0 {
self.forwardOn(event)
self.forwardOn(.completed)
self.dispose()
}
do {
_ = try decrementChecked(&self.i)
} catch let e {
self.forwardOn(.error(e))
self.dispose()
return
}
case .error(let e):
self.forwardOn(.error(e))
self.dispose()
case .completed:
if self.parent.throwOnEmpty {
self.forwardOn(.error(RxError.argumentOutOfRange))
} else {
self.forwardOn(.completed)
}
self.dispose()
}
}
}
final private class ElementAt<SourceType>: Producer<SourceType> {
let source: Observable<SourceType>
let throwOnEmpty: Bool
let index: Int
init(source: Observable<SourceType>, index: Int, throwOnEmpty: Bool) {
if index < 0 {
rxFatalError("index can't be negative")
}
self.source = source
self.index = index
self.throwOnEmpty = throwOnEmpty
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == SourceType {
let sink = ElementAtSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,27 @@
//
// Empty.swift
// RxSwift
//
// Created by Krunoslav Zaher on 8/30/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns an empty observable sequence, using the specified scheduler to send out the single `Completed` message.
- seealso: [empty operator on reactivex.io](http://reactivex.io/documentation/operators/empty-never-throw.html)
- returns: An observable sequence with no elements.
*/
public static func empty() -> Observable<Element> {
EmptyProducer<Element>()
}
}
final private class EmptyProducer<Element>: Producer<Element> {
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
observer.on(.completed)
return Disposables.create()
}
}

View File

@@ -0,0 +1,61 @@
//
// Enumerated.swift
// RxSwift
//
// Created by Krunoslav Zaher on 8/6/17.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Enumerates the elements of an observable sequence.
- seealso: [map operator on reactivex.io](http://reactivex.io/documentation/operators/map.html)
- returns: An observable sequence that contains tuples of source sequence elements and their indexes.
*/
public func enumerated()
-> Observable<(index: Int, element: Element)> {
Enumerated(source: self.asObservable())
}
}
final private class EnumeratedSink<Element, Observer: ObserverType>: Sink<Observer>, ObserverType where Observer.Element == (index: Int, element: Element) {
var index = 0
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
do {
let nextIndex = try incrementChecked(&self.index)
let next = (index: nextIndex, element: value)
self.forwardOn(.next(next))
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
case .completed:
self.forwardOn(.completed)
self.dispose()
case .error(let error):
self.forwardOn(.error(error))
self.dispose()
}
}
}
final private class Enumerated<Element>: Producer<(index: Int, element: Element)> {
private let source: Observable<Element>
init(source: Observable<Element>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == (index: Int, element: Element) {
let sink = EnumeratedSink<Element, Observer>(observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,33 @@
//
// Error.swift
// RxSwift
//
// Created by Krunoslav Zaher on 8/30/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns an observable sequence that terminates with an `error`.
- seealso: [throw operator on reactivex.io](http://reactivex.io/documentation/operators/empty-never-throw.html)
- returns: The observable sequence that terminates with specified error.
*/
public static func error(_ error: Swift.Error) -> Observable<Element> {
ErrorProducer(error: error)
}
}
final private class ErrorProducer<Element>: Producer<Element> {
private let error: Swift.Error
init(error: Swift.Error) {
self.error = error
}
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
observer.on(.error(self.error))
return Disposables.create()
}
}

View File

@@ -0,0 +1,86 @@
//
// Filter.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/17/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Filters the elements of an observable sequence based on a predicate.
- seealso: [filter operator on reactivex.io](http://reactivex.io/documentation/operators/filter.html)
- parameter predicate: A function to test each source element for a condition.
- returns: An observable sequence that contains elements from the input sequence that satisfy the condition.
*/
public func filter(_ predicate: @escaping (Element) throws -> Bool)
-> Observable<Element> {
Filter(source: self.asObservable(), predicate: predicate)
}
}
extension ObservableType {
/**
Skips elements and completes (or errors) when the observable sequence completes (or errors). Equivalent to filter that always returns false.
- seealso: [ignoreElements operator on reactivex.io](http://reactivex.io/documentation/operators/ignoreelements.html)
- returns: An observable sequence that skips all elements of the source sequence.
*/
public func ignoreElements()
-> Observable<Never> {
self.flatMap { _ in Observable<Never>.empty() }
}
}
final private class FilterSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Predicate = (Element) throws -> Bool
typealias Element = Observer.Element
private let predicate: Predicate
init(predicate: @escaping Predicate, observer: Observer, cancel: Cancelable) {
self.predicate = predicate
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
do {
let satisfies = try self.predicate(value)
if satisfies {
self.forwardOn(.next(value))
}
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
case .completed, .error:
self.forwardOn(event)
self.dispose()
}
}
}
final private class Filter<Element>: Producer<Element> {
typealias Predicate = (Element) throws -> Bool
private let source: Observable<Element>
private let predicate: Predicate
init(source: Observable<Element>, predicate: @escaping Predicate) {
self.source = source
self.predicate = predicate
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = FilterSink(predicate: self.predicate, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,41 @@
//
// First.swift
// RxSwift
//
// Created by Krunoslav Zaher on 7/31/17.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
private final class FirstSink<Element, Observer: ObserverType> : Sink<Observer>, ObserverType where Observer.Element == Element? {
typealias Parent = First<Element>
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
self.forwardOn(.next(value))
self.forwardOn(.completed)
self.dispose()
case .error(let error):
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.forwardOn(.next(nil))
self.forwardOn(.completed)
self.dispose()
}
}
}
final class First<Element>: Producer<Element?> {
private let source: Observable<Element>
init(source: Observable<Element>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element? {
let sink = FirstSink(observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,87 @@
//
// Generate.swift
// RxSwift
//
// Created by Krunoslav Zaher on 9/2/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Generates an observable sequence by running a state-driven loop producing the sequence's elements, using the specified scheduler
to run the loop send out observer messages.
- seealso: [create operator on reactivex.io](http://reactivex.io/documentation/operators/create.html)
- parameter initialState: Initial state.
- parameter condition: Condition to terminate generation (upon returning `false`).
- parameter iterate: Iteration step function.
- parameter scheduler: Scheduler on which to run the generator loop.
- returns: The generated sequence.
*/
public static func generate(initialState: Element, condition: @escaping (Element) throws -> Bool, scheduler: ImmediateSchedulerType = CurrentThreadScheduler.instance, iterate: @escaping (Element) throws -> Element) -> Observable<Element> {
Generate(initialState: initialState, condition: condition, iterate: iterate, resultSelector: { $0 }, scheduler: scheduler)
}
}
final private class GenerateSink<Sequence, Observer: ObserverType>: Sink<Observer> {
typealias Parent = Generate<Sequence, Observer.Element>
private let parent: Parent
private var state: Sequence
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.state = parent.initialState
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
return self.parent.scheduler.scheduleRecursive(true) { isFirst, recurse -> Void in
do {
if !isFirst {
self.state = try self.parent.iterate(self.state)
}
if try self.parent.condition(self.state) {
let result = try self.parent.resultSelector(self.state)
self.forwardOn(.next(result))
recurse(false)
}
else {
self.forwardOn(.completed)
self.dispose()
}
}
catch let error {
self.forwardOn(.error(error))
self.dispose()
}
}
}
}
final private class Generate<Sequence, Element>: Producer<Element> {
fileprivate let initialState: Sequence
fileprivate let condition: (Sequence) throws -> Bool
fileprivate let iterate: (Sequence) throws -> Sequence
fileprivate let resultSelector: (Sequence) throws -> Element
fileprivate let scheduler: ImmediateSchedulerType
init(initialState: Sequence, condition: @escaping (Sequence) throws -> Bool, iterate: @escaping (Sequence) throws -> Sequence, resultSelector: @escaping (Sequence) throws -> Element, scheduler: ImmediateSchedulerType) {
self.initialState = initialState
self.condition = condition
self.iterate = iterate
self.resultSelector = resultSelector
self.scheduler = scheduler
super.init()
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = GenerateSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,133 @@
//
// GroupBy.swift
// RxSwift
//
// Created by Tomi Koskinen on 01/12/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/*
Groups the elements of an observable sequence according to a specified key selector function.
- seealso: [groupBy operator on reactivex.io](http://reactivex.io/documentation/operators/groupby.html)
- parameter keySelector: A function to extract the key for each element.
- returns: A sequence of observable groups, each of which corresponds to a unique key value, containing all elements that share that same key value.
*/
public func groupBy<Key: Hashable>(keySelector: @escaping (Element) throws -> Key)
-> Observable<GroupedObservable<Key, Element>> {
GroupBy(source: self.asObservable(), selector: keySelector)
}
}
final private class GroupedObservableImpl<Element>: Observable<Element> {
private var subject: PublishSubject<Element>
private var refCount: RefCountDisposable
init(subject: PublishSubject<Element>, refCount: RefCountDisposable) {
self.subject = subject
self.refCount = refCount
}
override public func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
let release = self.refCount.retain()
let subscription = self.subject.subscribe(observer)
return Disposables.create(release, subscription)
}
}
final private class GroupBySink<Key: Hashable, Element, Observer: ObserverType>
: Sink<Observer>
, ObserverType where Observer.Element == GroupedObservable<Key, Element> {
typealias ResultType = Observer.Element
typealias Parent = GroupBy<Key, Element>
private let parent: Parent
private let subscription = SingleAssignmentDisposable()
private var refCountDisposable: RefCountDisposable!
private var groupedSubjectTable: [Key: PublishSubject<Element>]
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.groupedSubjectTable = [Key: PublishSubject<Element>]()
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
self.refCountDisposable = RefCountDisposable(disposable: self.subscription)
self.subscription.setDisposable(self.parent.source.subscribe(self))
return self.refCountDisposable
}
private func onGroupEvent(key: Key, value: Element) {
if let writer = self.groupedSubjectTable[key] {
writer.on(.next(value))
} else {
let writer = PublishSubject<Element>()
self.groupedSubjectTable[key] = writer
let group = GroupedObservable(
key: key,
source: GroupedObservableImpl(subject: writer, refCount: refCountDisposable)
)
self.forwardOn(.next(group))
writer.on(.next(value))
}
}
final func on(_ event: Event<Element>) {
switch event {
case let .next(value):
do {
let groupKey = try self.parent.selector(value)
self.onGroupEvent(key: groupKey, value: value)
}
catch let e {
self.error(e)
return
}
case let .error(e):
self.error(e)
case .completed:
self.forwardOnGroups(event: .completed)
self.forwardOn(.completed)
self.subscription.dispose()
self.dispose()
}
}
final func error(_ error: Swift.Error) {
self.forwardOnGroups(event: .error(error))
self.forwardOn(.error(error))
self.subscription.dispose()
self.dispose()
}
final func forwardOnGroups(event: Event<Element>) {
for writer in self.groupedSubjectTable.values {
writer.on(event)
}
}
}
final private class GroupBy<Key: Hashable, Element>: Producer<GroupedObservable<Key,Element>> {
typealias KeySelector = (Element) throws -> Key
fileprivate let source: Observable<Element>
fileprivate let selector: KeySelector
init(source: Observable<Element>, selector: @escaping KeySelector) {
self.source = source
self.selector = selector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == GroupedObservable<Key,Element> {
let sink = GroupBySink(parent: self, observer: observer, cancel: cancel)
return (sink: sink, subscription: sink.run())
}
}

View File

@@ -0,0 +1,87 @@
//
// Just.swift
// RxSwift
//
// Created by Krunoslav Zaher on 8/30/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns an observable sequence that contains a single element.
- seealso: [just operator on reactivex.io](http://reactivex.io/documentation/operators/just.html)
- parameter element: Single element in the resulting observable sequence.
- returns: An observable sequence containing the single specified element.
*/
public static func just(_ element: Element) -> Observable<Element> {
Just(element: element)
}
/**
Returns an observable sequence that contains a single element.
- seealso: [just operator on reactivex.io](http://reactivex.io/documentation/operators/just.html)
- parameter element: Single element in the resulting observable sequence.
- parameter scheduler: Scheduler to send the single element on.
- returns: An observable sequence containing the single specified element.
*/
public static func just(_ element: Element, scheduler: ImmediateSchedulerType) -> Observable<Element> {
JustScheduled(element: element, scheduler: scheduler)
}
}
final private class JustScheduledSink<Observer: ObserverType>: Sink<Observer> {
typealias Parent = JustScheduled<Observer.Element>
private let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
let scheduler = self.parent.scheduler
return scheduler.schedule(self.parent.element) { element in
self.forwardOn(.next(element))
return scheduler.schedule(()) { _ in
self.forwardOn(.completed)
self.dispose()
return Disposables.create()
}
}
}
}
final private class JustScheduled<Element>: Producer<Element> {
fileprivate let scheduler: ImmediateSchedulerType
fileprivate let element: Element
init(element: Element, scheduler: ImmediateSchedulerType) {
self.scheduler = scheduler
self.element = element
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = JustScheduledSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
final private class Just<Element>: Producer<Element> {
private let element: Element
init(element: Element) {
self.element = element
}
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
observer.on(.next(self.element))
observer.on(.completed)
return Disposables.create()
}
}

View File

@@ -0,0 +1,77 @@
//
// Map.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/15/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Projects each element of an observable sequence into a new form.
- seealso: [map operator on reactivex.io](http://reactivex.io/documentation/operators/map.html)
- parameter transform: A transform function to apply to each source element.
- returns: An observable sequence whose elements are the result of invoking the transform function on each element of source.
*/
public func map<Result>(_ transform: @escaping (Element) throws -> Result)
-> Observable<Result> {
Map(source: self.asObservable(), transform: transform)
}
}
final private class MapSink<SourceType, Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Transform = (SourceType) throws -> ResultType
typealias ResultType = Observer.Element
typealias Element = SourceType
private let transform: Transform
init(transform: @escaping Transform, observer: Observer, cancel: Cancelable) {
self.transform = transform
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<SourceType>) {
switch event {
case .next(let element):
do {
let mappedElement = try self.transform(element)
self.forwardOn(.next(mappedElement))
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
case .error(let error):
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class Map<SourceType, ResultType>: Producer<ResultType> {
typealias Transform = (SourceType) throws -> ResultType
private let source: Observable<SourceType>
private let transform: Transform
init(source: Observable<SourceType>, transform: @escaping Transform) {
self.source = source
self.transform = transform
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == ResultType {
let sink = MapSink(transform: self.transform, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,44 @@
//
// Materialize.swift
// RxSwift
//
// Created by sergdort on 08/03/2017.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Convert any Observable into an Observable of its events.
- seealso: [materialize operator on reactivex.io](http://reactivex.io/documentation/operators/materialize-dematerialize.html)
- returns: An observable sequence that wraps events in an Event<E>. The returned Observable never errors, but it does complete after observing all of the events of the underlying Observable.
*/
public func materialize() -> Observable<Event<Element>> {
Materialize(source: self.asObservable())
}
}
private final class MaterializeSink<Element, Observer: ObserverType>: Sink<Observer>, ObserverType where Observer.Element == Event<Element> {
func on(_ event: Event<Element>) {
self.forwardOn(.next(event))
if event.isStopEvent {
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class Materialize<T>: Producer<Event<T>> {
private let source: Observable<T>
init(source: Observable<T>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = MaterializeSink(observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,600 @@
//
// Merge.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/28/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Projects each element of an observable sequence to an observable sequence and merges the resulting observable sequences into one observable sequence.
- seealso: [flatMap operator on reactivex.io](http://reactivex.io/documentation/operators/flatmap.html)
- parameter selector: A transform function to apply to each element.
- returns: An observable sequence whose elements are the result of invoking the one-to-many transform function on each element of the input sequence.
*/
public func flatMap<Source: ObservableConvertibleType>(_ selector: @escaping (Element) throws -> Source)
-> Observable<Source.Element> {
return FlatMap(source: self.asObservable(), selector: selector)
}
}
extension ObservableType {
/**
Projects each element of an observable sequence to an observable sequence and merges the resulting observable sequences into one observable sequence.
If element is received while there is some projected observable sequence being merged it will simply be ignored.
- seealso: [flatMapFirst operator on reactivex.io](http://reactivex.io/documentation/operators/flatmap.html)
- parameter selector: A transform function to apply to element that was observed while no observable is executing in parallel.
- returns: An observable sequence whose elements are the result of invoking the one-to-many transform function on each element of the input sequence that was received while no other sequence was being calculated.
*/
public func flatMapFirst<Source: ObservableConvertibleType>(_ selector: @escaping (Element) throws -> Source)
-> Observable<Source.Element> {
return FlatMapFirst(source: self.asObservable(), selector: selector)
}
}
extension ObservableType where Element: ObservableConvertibleType {
/**
Merges elements from all observable sequences in the given enumerable sequence into a single observable sequence.
- seealso: [merge operator on reactivex.io](http://reactivex.io/documentation/operators/merge.html)
- returns: The observable sequence that merges the elements of the observable sequences.
*/
public func merge() -> Observable<Element.Element> {
Merge(source: self.asObservable())
}
/**
Merges elements from all inner observable sequences into a single observable sequence, limiting the number of concurrent subscriptions to inner sequences.
- seealso: [merge operator on reactivex.io](http://reactivex.io/documentation/operators/merge.html)
- parameter maxConcurrent: Maximum number of inner observable sequences being subscribed to concurrently.
- returns: The observable sequence that merges the elements of the inner sequences.
*/
public func merge(maxConcurrent: Int)
-> Observable<Element.Element> {
MergeLimited(source: self.asObservable(), maxConcurrent: maxConcurrent)
}
}
extension ObservableType where Element: ObservableConvertibleType {
/**
Concatenates all inner observable sequences, as long as the previous observable sequence terminated successfully.
- seealso: [concat operator on reactivex.io](http://reactivex.io/documentation/operators/concat.html)
- returns: An observable sequence that contains the elements of each observed inner sequence, in sequential order.
*/
public func concat() -> Observable<Element.Element> {
self.merge(maxConcurrent: 1)
}
}
extension ObservableType {
/**
Merges elements from all observable sequences from collection into a single observable sequence.
- seealso: [merge operator on reactivex.io](http://reactivex.io/documentation/operators/merge.html)
- parameter sources: Collection of observable sequences to merge.
- returns: The observable sequence that merges the elements of the observable sequences.
*/
public static func merge<Collection: Swift.Collection>(_ sources: Collection) -> Observable<Element> where Collection.Element == Observable<Element> {
MergeArray(sources: Array(sources))
}
/**
Merges elements from all observable sequences from array into a single observable sequence.
- seealso: [merge operator on reactivex.io](http://reactivex.io/documentation/operators/merge.html)
- parameter sources: Array of observable sequences to merge.
- returns: The observable sequence that merges the elements of the observable sequences.
*/
public static func merge(_ sources: [Observable<Element>]) -> Observable<Element> {
MergeArray(sources: sources)
}
/**
Merges elements from all observable sequences into a single observable sequence.
- seealso: [merge operator on reactivex.io](http://reactivex.io/documentation/operators/merge.html)
- parameter sources: Collection of observable sequences to merge.
- returns: The observable sequence that merges the elements of the observable sequences.
*/
public static func merge(_ sources: Observable<Element>...) -> Observable<Element> {
MergeArray(sources: sources)
}
}
// MARK: concatMap
extension ObservableType {
/**
Projects each element of an observable sequence to an observable sequence and concatenates the resulting observable sequences into one observable sequence.
- seealso: [concat operator on reactivex.io](http://reactivex.io/documentation/operators/concat.html)
- returns: An observable sequence that contains the elements of each observed inner sequence, in sequential order.
*/
public func concatMap<Source: ObservableConvertibleType>(_ selector: @escaping (Element) throws -> Source)
-> Observable<Source.Element> {
return ConcatMap(source: self.asObservable(), selector: selector)
}
}
private final class MergeLimitedSinkIter<SourceElement, SourceSequence: ObservableConvertibleType, Observer: ObserverType>
: ObserverType
, LockOwnerType
, SynchronizedOnType where SourceSequence.Element == Observer.Element {
typealias Element = Observer.Element
typealias DisposeKey = CompositeDisposable.DisposeKey
typealias Parent = MergeLimitedSink<SourceElement, SourceSequence, Observer>
private let parent: Parent
private let disposeKey: DisposeKey
var lock: RecursiveLock {
self.parent.lock
}
init(parent: Parent, disposeKey: DisposeKey) {
self.parent = parent
self.disposeKey = disposeKey
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next:
self.parent.forwardOn(event)
case .error:
self.parent.forwardOn(event)
self.parent.dispose()
case .completed:
self.parent.group.remove(for: self.disposeKey)
if let next = self.parent.queue.dequeue() {
self.parent.subscribe(next, group: self.parent.group)
}
else {
self.parent.activeCount -= 1
if self.parent.stopped && self.parent.activeCount == 0 {
self.parent.forwardOn(.completed)
self.parent.dispose()
}
}
}
}
}
private final class ConcatMapSink<SourceElement, SourceSequence: ObservableConvertibleType, Observer: ObserverType>: MergeLimitedSink<SourceElement, SourceSequence, Observer> where Observer.Element == SourceSequence.Element {
typealias Selector = (SourceElement) throws -> SourceSequence
private let selector: Selector
init(selector: @escaping Selector, observer: Observer, cancel: Cancelable) {
self.selector = selector
super.init(maxConcurrent: 1, observer: observer, cancel: cancel)
}
override func performMap(_ element: SourceElement) throws -> SourceSequence {
try self.selector(element)
}
}
private final class MergeLimitedBasicSink<SourceSequence: ObservableConvertibleType, Observer: ObserverType>: MergeLimitedSink<SourceSequence, SourceSequence, Observer> where Observer.Element == SourceSequence.Element {
override func performMap(_ element: SourceSequence) throws -> SourceSequence {
element
}
}
private class MergeLimitedSink<SourceElement, SourceSequence: ObservableConvertibleType, Observer: ObserverType>
: Sink<Observer>
, ObserverType where Observer.Element == SourceSequence.Element {
typealias QueueType = Queue<SourceSequence>
let maxConcurrent: Int
let lock = RecursiveLock()
// state
var stopped = false
var activeCount = 0
var queue = QueueType(capacity: 2)
let sourceSubscription = SingleAssignmentDisposable()
let group = CompositeDisposable()
init(maxConcurrent: Int, observer: Observer, cancel: Cancelable) {
self.maxConcurrent = maxConcurrent
super.init(observer: observer, cancel: cancel)
}
func run(_ source: Observable<SourceElement>) -> Disposable {
_ = self.group.insert(self.sourceSubscription)
let disposable = source.subscribe(self)
self.sourceSubscription.setDisposable(disposable)
return self.group
}
func subscribe(_ innerSource: SourceSequence, group: CompositeDisposable) {
let subscription = SingleAssignmentDisposable()
let key = group.insert(subscription)
if let key = key {
let observer = MergeLimitedSinkIter(parent: self, disposeKey: key)
let disposable = innerSource.asObservable().subscribe(observer)
subscription.setDisposable(disposable)
}
}
func performMap(_ element: SourceElement) throws -> SourceSequence {
rxAbstractMethod()
}
@inline(__always)
final private func nextElementArrived(element: SourceElement) -> SourceSequence? {
self.lock.performLocked {
let subscribe: Bool
if self.activeCount < self.maxConcurrent {
self.activeCount += 1
subscribe = true
}
else {
do {
let value = try self.performMap(element)
self.queue.enqueue(value)
} catch {
self.forwardOn(.error(error))
self.dispose()
}
subscribe = false
}
if subscribe {
do {
return try self.performMap(element)
} catch {
self.forwardOn(.error(error))
self.dispose()
}
}
return nil
}
}
func on(_ event: Event<SourceElement>) {
switch event {
case .next(let element):
if let sequence = self.nextElementArrived(element: element) {
self.subscribe(sequence, group: self.group)
}
case .error(let error):
self.lock.performLocked {
self.forwardOn(.error(error))
self.dispose()
}
case .completed:
self.lock.performLocked {
if self.activeCount == 0 {
self.forwardOn(.completed)
self.dispose()
}
else {
self.sourceSubscription.dispose()
}
self.stopped = true
}
}
}
}
final private class MergeLimited<SourceSequence: ObservableConvertibleType>: Producer<SourceSequence.Element> {
private let source: Observable<SourceSequence>
private let maxConcurrent: Int
init(source: Observable<SourceSequence>, maxConcurrent: Int) {
self.source = source
self.maxConcurrent = maxConcurrent
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == SourceSequence.Element {
let sink = MergeLimitedBasicSink<SourceSequence, Observer>(maxConcurrent: self.maxConcurrent, observer: observer, cancel: cancel)
let subscription = sink.run(self.source)
return (sink: sink, subscription: subscription)
}
}
// MARK: Merge
private final class MergeBasicSink<Source: ObservableConvertibleType, Observer: ObserverType> : MergeSink<Source, Source, Observer> where Observer.Element == Source.Element {
override func performMap(_ element: Source) throws -> Source {
element
}
}
// MARK: flatMap
private final class FlatMapSink<SourceElement, SourceSequence: ObservableConvertibleType, Observer: ObserverType> : MergeSink<SourceElement, SourceSequence, Observer> where Observer.Element == SourceSequence.Element {
typealias Selector = (SourceElement) throws -> SourceSequence
private let selector: Selector
init(selector: @escaping Selector, observer: Observer, cancel: Cancelable) {
self.selector = selector
super.init(observer: observer, cancel: cancel)
}
override func performMap(_ element: SourceElement) throws -> SourceSequence {
try self.selector(element)
}
}
// MARK: FlatMapFirst
private final class FlatMapFirstSink<SourceElement, SourceSequence: ObservableConvertibleType, Observer: ObserverType> : MergeSink<SourceElement, SourceSequence, Observer> where Observer.Element == SourceSequence.Element {
typealias Selector = (SourceElement) throws -> SourceSequence
private let selector: Selector
override var subscribeNext: Bool {
self.activeCount == 0
}
init(selector: @escaping Selector, observer: Observer, cancel: Cancelable) {
self.selector = selector
super.init(observer: observer, cancel: cancel)
}
override func performMap(_ element: SourceElement) throws -> SourceSequence {
try self.selector(element)
}
}
private final class MergeSinkIter<SourceElement, SourceSequence: ObservableConvertibleType, Observer: ObserverType> : ObserverType where Observer.Element == SourceSequence.Element {
typealias Parent = MergeSink<SourceElement, SourceSequence, Observer>
typealias DisposeKey = CompositeDisposable.DisposeKey
typealias Element = Observer.Element
private let parent: Parent
private let disposeKey: DisposeKey
init(parent: Parent, disposeKey: DisposeKey) {
self.parent = parent
self.disposeKey = disposeKey
}
func on(_ event: Event<Element>) {
self.parent.lock.performLocked {
switch event {
case .next(let value):
self.parent.forwardOn(.next(value))
case .error(let error):
self.parent.forwardOn(.error(error))
self.parent.dispose()
case .completed:
self.parent.group.remove(for: self.disposeKey)
self.parent.activeCount -= 1
self.parent.checkCompleted()
}
}
}
}
private class MergeSink<SourceElement, SourceSequence: ObservableConvertibleType, Observer: ObserverType>
: Sink<Observer>
, ObserverType where Observer.Element == SourceSequence.Element {
typealias ResultType = Observer.Element
typealias Element = SourceElement
let lock = RecursiveLock()
var subscribeNext: Bool {
true
}
// state
let group = CompositeDisposable()
let sourceSubscription = SingleAssignmentDisposable()
var activeCount = 0
var stopped = false
override init(observer: Observer, cancel: Cancelable) {
super.init(observer: observer, cancel: cancel)
}
func performMap(_ element: SourceElement) throws -> SourceSequence {
rxAbstractMethod()
}
@inline(__always)
final private func nextElementArrived(element: SourceElement) -> SourceSequence? {
self.lock.performLocked {
if !self.subscribeNext {
return nil
}
do {
let value = try self.performMap(element)
self.activeCount += 1
return value
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
return nil
}
}
}
func on(_ event: Event<SourceElement>) {
switch event {
case .next(let element):
if let value = self.nextElementArrived(element: element) {
self.subscribeInner(value.asObservable())
}
case .error(let error):
self.lock.performLocked {
self.forwardOn(.error(error))
self.dispose()
}
case .completed:
self.lock.performLocked {
self.stopped = true
self.sourceSubscription.dispose()
self.checkCompleted()
}
}
}
func subscribeInner(_ source: Observable<Observer.Element>) {
let iterDisposable = SingleAssignmentDisposable()
if let disposeKey = self.group.insert(iterDisposable) {
let iter = MergeSinkIter(parent: self, disposeKey: disposeKey)
let subscription = source.subscribe(iter)
iterDisposable.setDisposable(subscription)
}
}
func run(_ sources: [Observable<Observer.Element>]) -> Disposable {
self.activeCount += sources.count
for source in sources {
self.subscribeInner(source)
}
self.stopped = true
self.checkCompleted()
return self.group
}
@inline(__always)
func checkCompleted() {
if self.stopped && self.activeCount == 0 {
self.forwardOn(.completed)
self.dispose()
}
}
func run(_ source: Observable<SourceElement>) -> Disposable {
_ = self.group.insert(self.sourceSubscription)
let subscription = source.subscribe(self)
self.sourceSubscription.setDisposable(subscription)
return self.group
}
}
// MARK: Producers
final private class FlatMap<SourceElement, SourceSequence: ObservableConvertibleType>: Producer<SourceSequence.Element> {
typealias Selector = (SourceElement) throws -> SourceSequence
private let source: Observable<SourceElement>
private let selector: Selector
init(source: Observable<SourceElement>, selector: @escaping Selector) {
self.source = source
self.selector = selector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == SourceSequence.Element {
let sink = FlatMapSink(selector: self.selector, observer: observer, cancel: cancel)
let subscription = sink.run(self.source)
return (sink: sink, subscription: subscription)
}
}
final private class FlatMapFirst<SourceElement, SourceSequence: ObservableConvertibleType>: Producer<SourceSequence.Element> {
typealias Selector = (SourceElement) throws -> SourceSequence
private let source: Observable<SourceElement>
private let selector: Selector
init(source: Observable<SourceElement>, selector: @escaping Selector) {
self.source = source
self.selector = selector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == SourceSequence.Element {
let sink = FlatMapFirstSink<SourceElement, SourceSequence, Observer>(selector: self.selector, observer: observer, cancel: cancel)
let subscription = sink.run(self.source)
return (sink: sink, subscription: subscription)
}
}
final class ConcatMap<SourceElement, SourceSequence: ObservableConvertibleType>: Producer<SourceSequence.Element> {
typealias Selector = (SourceElement) throws -> SourceSequence
private let source: Observable<SourceElement>
private let selector: Selector
init(source: Observable<SourceElement>, selector: @escaping Selector) {
self.source = source
self.selector = selector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == SourceSequence.Element {
let sink = ConcatMapSink<SourceElement, SourceSequence, Observer>(selector: self.selector, observer: observer, cancel: cancel)
let subscription = sink.run(self.source)
return (sink: sink, subscription: subscription)
}
}
final class Merge<SourceSequence: ObservableConvertibleType> : Producer<SourceSequence.Element> {
private let source: Observable<SourceSequence>
init(source: Observable<SourceSequence>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == SourceSequence.Element {
let sink = MergeBasicSink<SourceSequence, Observer>(observer: observer, cancel: cancel)
let subscription = sink.run(self.source)
return (sink: sink, subscription: subscription)
}
}
final private class MergeArray<Element>: Producer<Element> {
private let sources: [Observable<Element>]
init(sources: [Observable<Element>]) {
self.sources = sources
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = MergeBasicSink<Observable<Element>, Observer>(observer: observer, cancel: cancel)
let subscription = sink.run(self.sources)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,405 @@
//
// Multicast.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/27/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
/**
Represents an observable wrapper that can be connected and disconnected from its underlying observable sequence.
*/
public class ConnectableObservable<Element>
: Observable<Element>
, ConnectableObservableType {
/**
Connects the observable wrapper to its source. All subscribed observers will receive values from the underlying observable sequence as long as the connection is established.
- returns: Disposable used to disconnect the observable wrapper from its source, causing subscribed observer to stop receiving values from the underlying observable sequence.
*/
public func connect() -> Disposable {
rxAbstractMethod()
}
}
extension ObservableType {
/**
Multicasts the source sequence notifications through an instantiated subject into all uses of the sequence within a selector function.
Each subscription to the resulting sequence causes a separate multicast invocation, exposing the sequence resulting from the selector function's invocation.
For specializations with fixed subject types, see `publish` and `replay`.
- seealso: [multicast operator on reactivex.io](http://reactivex.io/documentation/operators/publish.html)
- parameter subjectSelector: Factory function to create an intermediate subject through which the source sequence's elements will be multicast to the selector function.
- parameter selector: Selector function which can use the multicasted source sequence subject to the policies enforced by the created subject.
- returns: An observable sequence that contains the elements of a sequence produced by multicasting the source sequence within a selector function.
*/
public func multicast<Subject: SubjectType, Result>(_ subjectSelector: @escaping () throws -> Subject, selector: @escaping (Observable<Subject.Element>) throws -> Observable<Result>)
-> Observable<Result> where Subject.Observer.Element == Element {
return Multicast(
source: self.asObservable(),
subjectSelector: subjectSelector,
selector: selector
)
}
}
extension ObservableType {
/**
Returns a connectable observable sequence that shares a single subscription to the underlying sequence.
This operator is a specialization of `multicast` using a `PublishSubject`.
- seealso: [publish operator on reactivex.io](http://reactivex.io/documentation/operators/publish.html)
- returns: A connectable observable sequence that shares a single subscription to the underlying sequence.
*/
public func publish() -> ConnectableObservable<Element> {
self.multicast { PublishSubject() }
}
}
extension ObservableType {
/**
Returns a connectable observable sequence that shares a single subscription to the underlying sequence replaying bufferSize elements.
This operator is a specialization of `multicast` using a `ReplaySubject`.
- seealso: [replay operator on reactivex.io](http://reactivex.io/documentation/operators/replay.html)
- parameter bufferSize: Maximum element count of the replay buffer.
- returns: A connectable observable sequence that shares a single subscription to the underlying sequence.
*/
public func replay(_ bufferSize: Int)
-> ConnectableObservable<Element> {
self.multicast { ReplaySubject.create(bufferSize: bufferSize) }
}
/**
Returns a connectable observable sequence that shares a single subscription to the underlying sequence replaying all elements.
This operator is a specialization of `multicast` using a `ReplaySubject`.
- seealso: [replay operator on reactivex.io](http://reactivex.io/documentation/operators/replay.html)
- returns: A connectable observable sequence that shares a single subscription to the underlying sequence.
*/
public func replayAll()
-> ConnectableObservable<Element> {
self.multicast { ReplaySubject.createUnbounded() }
}
}
extension ConnectableObservableType {
/**
Returns an observable sequence that stays connected to the source as long as there is at least one subscription to the observable sequence.
- seealso: [refCount operator on reactivex.io](http://reactivex.io/documentation/operators/refcount.html)
- returns: An observable sequence that stays connected to the source as long as there is at least one subscription to the observable sequence.
*/
public func refCount() -> Observable<Element> {
RefCount(source: self)
}
}
extension ObservableType {
/**
Multicasts the source sequence notifications through the specified subject to the resulting connectable observable.
Upon connection of the connectable observable, the subject is subscribed to the source exactly one, and messages are forwarded to the observers registered with the connectable observable.
For specializations with fixed subject types, see `publish` and `replay`.
- seealso: [multicast operator on reactivex.io](http://reactivex.io/documentation/operators/publish.html)
- parameter subject: Subject to push source elements into.
- returns: A connectable observable sequence that upon connection causes the source sequence to push results into the specified subject.
*/
public func multicast<Subject: SubjectType>(_ subject: Subject)
-> ConnectableObservable<Subject.Element> where Subject.Observer.Element == Element {
ConnectableObservableAdapter(source: self.asObservable(), makeSubject: { subject })
}
/**
Multicasts the source sequence notifications through an instantiated subject to the resulting connectable observable.
Upon connection of the connectable observable, the subject is subscribed to the source exactly one, and messages are forwarded to the observers registered with the connectable observable.
Subject is cleared on connection disposal or in case source sequence produces terminal event.
- seealso: [multicast operator on reactivex.io](http://reactivex.io/documentation/operators/publish.html)
- parameter makeSubject: Factory function used to instantiate a subject for each connection.
- returns: A connectable observable sequence that upon connection causes the source sequence to push results into the specified subject.
*/
public func multicast<Subject: SubjectType>(makeSubject: @escaping () -> Subject)
-> ConnectableObservable<Subject.Element> where Subject.Observer.Element == Element {
ConnectableObservableAdapter(source: self.asObservable(), makeSubject: makeSubject)
}
}
final private class Connection<Subject: SubjectType>: ObserverType, Disposable {
typealias Element = Subject.Observer.Element
private var lock: RecursiveLock
// state
private var parent: ConnectableObservableAdapter<Subject>?
private var subscription : Disposable?
private var subjectObserver: Subject.Observer
private let disposed = AtomicInt(0)
init(parent: ConnectableObservableAdapter<Subject>, subjectObserver: Subject.Observer, lock: RecursiveLock, subscription: Disposable) {
self.parent = parent
self.subscription = subscription
self.lock = lock
self.subjectObserver = subjectObserver
}
func on(_ event: Event<Subject.Observer.Element>) {
if isFlagSet(self.disposed, 1) {
return
}
if event.isStopEvent {
self.dispose()
}
self.subjectObserver.on(event)
}
func dispose() {
lock.lock(); defer { lock.unlock() }
fetchOr(self.disposed, 1)
guard let parent = self.parent else {
return
}
if parent.connection === self {
parent.connection = nil
parent.subject = nil
}
self.parent = nil
self.subscription?.dispose()
self.subscription = nil
}
}
final private class ConnectableObservableAdapter<Subject: SubjectType>
: ConnectableObservable<Subject.Element> {
typealias ConnectionType = Connection<Subject>
private let source: Observable<Subject.Observer.Element>
private let makeSubject: () -> Subject
fileprivate let lock = RecursiveLock()
fileprivate var subject: Subject?
// state
fileprivate var connection: ConnectionType?
init(source: Observable<Subject.Observer.Element>, makeSubject: @escaping () -> Subject) {
self.source = source
self.makeSubject = makeSubject
self.subject = nil
self.connection = nil
}
override func connect() -> Disposable {
return self.lock.performLocked {
if let connection = self.connection {
return connection
}
let singleAssignmentDisposable = SingleAssignmentDisposable()
let connection = Connection(parent: self, subjectObserver: self.lazySubject.asObserver(), lock: self.lock, subscription: singleAssignmentDisposable)
self.connection = connection
let subscription = self.source.subscribe(connection)
singleAssignmentDisposable.setDisposable(subscription)
return connection
}
}
private var lazySubject: Subject {
if let subject = self.subject {
return subject
}
let subject = self.makeSubject()
self.subject = subject
return subject
}
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Subject.Element {
self.lazySubject.subscribe(observer)
}
}
final private class RefCountSink<ConnectableSource: ConnectableObservableType, Observer: ObserverType>
: Sink<Observer>
, ObserverType where ConnectableSource.Element == Observer.Element {
typealias Element = Observer.Element
typealias Parent = RefCount<ConnectableSource>
private let parent: Parent
private var connectionIdSnapshot: Int64 = -1
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription = self.parent.source.subscribe(self)
self.parent.lock.lock(); defer { self.parent.lock.unlock() }
self.connectionIdSnapshot = self.parent.connectionId
if self.isDisposed {
return Disposables.create()
}
if self.parent.count == 0 {
self.parent.count = 1
self.parent.connectableSubscription = self.parent.source.connect()
}
else {
self.parent.count += 1
}
return Disposables.create {
subscription.dispose()
self.parent.lock.lock(); defer { self.parent.lock.unlock() }
if self.parent.connectionId != self.connectionIdSnapshot {
return
}
if self.parent.count == 1 {
self.parent.count = 0
guard let connectableSubscription = self.parent.connectableSubscription else {
return
}
connectableSubscription.dispose()
self.parent.connectableSubscription = nil
}
else if self.parent.count > 1 {
self.parent.count -= 1
}
else {
rxFatalError("Something went wrong with RefCount disposing mechanism")
}
}
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.forwardOn(event)
case .error, .completed:
self.parent.lock.lock()
if self.parent.connectionId == self.connectionIdSnapshot {
let connection = self.parent.connectableSubscription
defer { connection?.dispose() }
self.parent.count = 0
self.parent.connectionId = self.parent.connectionId &+ 1
self.parent.connectableSubscription = nil
}
self.parent.lock.unlock()
self.forwardOn(event)
self.dispose()
}
}
}
final private class RefCount<ConnectableSource: ConnectableObservableType>: Producer<ConnectableSource.Element> {
fileprivate let lock = RecursiveLock()
// state
fileprivate var count = 0
fileprivate var connectionId: Int64 = 0
fileprivate var connectableSubscription = nil as Disposable?
fileprivate let source: ConnectableSource
init(source: ConnectableSource) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable)
where Observer.Element == ConnectableSource.Element {
let sink = RefCountSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
final private class MulticastSink<Subject: SubjectType, Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias ResultType = Element
typealias MutlicastType = Multicast<Subject, Observer.Element>
private let parent: MutlicastType
init(parent: MutlicastType, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
do {
let subject = try self.parent.subjectSelector()
let connectable = ConnectableObservableAdapter(source: self.parent.source, makeSubject: { subject })
let observable = try self.parent.selector(connectable)
let subscription = observable.subscribe(self)
let connection = connectable.connect()
return Disposables.create(subscription, connection)
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
return Disposables.create()
}
}
func on(_ event: Event<ResultType>) {
self.forwardOn(event)
switch event {
case .next: break
case .error, .completed:
self.dispose()
}
}
}
final private class Multicast<Subject: SubjectType, Result>: Producer<Result> {
typealias SubjectSelectorType = () throws -> Subject
typealias SelectorType = (Observable<Subject.Element>) throws -> Observable<Result>
fileprivate let source: Observable<Subject.Observer.Element>
fileprivate let subjectSelector: SubjectSelectorType
fileprivate let selector: SelectorType
init(source: Observable<Subject.Observer.Element>, subjectSelector: @escaping SubjectSelectorType, selector: @escaping SelectorType) {
self.source = source
self.subjectSelector = subjectSelector
self.selector = selector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Result {
let sink = MulticastSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,27 @@
//
// Never.swift
// RxSwift
//
// Created by Krunoslav Zaher on 8/30/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns a non-terminating observable sequence, which can be used to denote an infinite duration.
- seealso: [never operator on reactivex.io](http://reactivex.io/documentation/operators/empty-never-throw.html)
- returns: An observable sequence whose observers will never get called.
*/
public static func never() -> Observable<Element> {
NeverProducer()
}
}
final private class NeverProducer<Element>: Producer<Element> {
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
Disposables.create()
}
}

View File

@@ -0,0 +1,243 @@
//
// ObserveOn.swift
// RxSwift
//
// Created by Krunoslav Zaher on 7/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Wraps the source sequence in order to run its observer callbacks on the specified scheduler.
This only invokes observer callbacks on a `scheduler`. In case the subscription and/or unsubscription
actions have side-effects that require to be run on a scheduler, use `subscribeOn`.
- seealso: [observeOn operator on reactivex.io](http://reactivex.io/documentation/operators/observeon.html)
- parameter scheduler: Scheduler to notify observers on.
- returns: The source sequence whose observations happen on the specified scheduler.
*/
public func observe(on scheduler: ImmediateSchedulerType)
-> Observable<Element> {
guard let serialScheduler = scheduler as? SerialDispatchQueueScheduler else {
return ObserveOn(source: self.asObservable(), scheduler: scheduler)
}
return ObserveOnSerialDispatchQueue(source: self.asObservable(),
scheduler: serialScheduler)
}
/**
Wraps the source sequence in order to run its observer callbacks on the specified scheduler.
This only invokes observer callbacks on a `scheduler`. In case the subscription and/or unsubscription
actions have side-effects that require to be run on a scheduler, use `subscribeOn`.
- seealso: [observeOn operator on reactivex.io](http://reactivex.io/documentation/operators/observeon.html)
- parameter scheduler: Scheduler to notify observers on.
- returns: The source sequence whose observations happen on the specified scheduler.
*/
@available(*, deprecated, renamed: "observe(on:)")
public func observeOn(_ scheduler: ImmediateSchedulerType)
-> Observable<Element> {
observe(on: scheduler)
}
}
final private class ObserveOn<Element>: Producer<Element> {
let scheduler: ImmediateSchedulerType
let source: Observable<Element>
init(source: Observable<Element>, scheduler: ImmediateSchedulerType) {
self.scheduler = scheduler
self.source = source
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = ObserveOnSink(scheduler: self.scheduler, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
#if TRACE_RESOURCES
deinit {
_ = Resources.decrementTotal()
}
#endif
}
enum ObserveOnState : Int32 {
// pump is not running
case stopped = 0
// pump is running
case running = 1
}
final private class ObserveOnSink<Observer: ObserverType>: ObserverBase<Observer.Element> {
typealias Element = Observer.Element
let scheduler: ImmediateSchedulerType
var lock = SpinLock()
let observer: Observer
// state
var state = ObserveOnState.stopped
var queue = Queue<Event<Element>>(capacity: 10)
let scheduleDisposable = SerialDisposable()
let cancel: Cancelable
init(scheduler: ImmediateSchedulerType, observer: Observer, cancel: Cancelable) {
self.scheduler = scheduler
self.observer = observer
self.cancel = cancel
}
override func onCore(_ event: Event<Element>) {
let shouldStart = self.lock.performLocked { () -> Bool in
self.queue.enqueue(event)
switch self.state {
case .stopped:
self.state = .running
return true
case .running:
return false
}
}
if shouldStart {
self.scheduleDisposable.disposable = self.scheduler.scheduleRecursive((), action: self.run)
}
}
func run(_ state: (), _ recurse: (()) -> Void) {
let (nextEvent, observer) = self.lock.performLocked { () -> (Event<Element>?, Observer) in
if !self.queue.isEmpty {
return (self.queue.dequeue(), self.observer)
}
else {
self.state = .stopped
return (nil, self.observer)
}
}
if let nextEvent = nextEvent, !self.cancel.isDisposed {
observer.on(nextEvent)
if nextEvent.isStopEvent {
self.dispose()
}
}
else {
return
}
let shouldContinue = self.shouldContinue_synchronized()
if shouldContinue {
recurse(())
}
}
func shouldContinue_synchronized() -> Bool {
self.lock.performLocked {
let isEmpty = self.queue.isEmpty
if isEmpty { self.state = .stopped }
return !isEmpty
}
}
override func dispose() {
super.dispose()
self.cancel.dispose()
self.scheduleDisposable.dispose()
}
}
#if TRACE_RESOURCES
private let numberOfSerialDispatchObservables = AtomicInt(0)
extension Resources {
/**
Counts number of `SerialDispatchQueueObservables`.
Purposed for unit tests.
*/
public static var numberOfSerialDispatchQueueObservables: Int32 {
return load(numberOfSerialDispatchObservables)
}
}
#endif
final private class ObserveOnSerialDispatchQueueSink<Observer: ObserverType>: ObserverBase<Observer.Element> {
let scheduler: SerialDispatchQueueScheduler
let observer: Observer
let cancel: Cancelable
var cachedScheduleLambda: (((sink: ObserveOnSerialDispatchQueueSink<Observer>, event: Event<Element>)) -> Disposable)!
init(scheduler: SerialDispatchQueueScheduler, observer: Observer, cancel: Cancelable) {
self.scheduler = scheduler
self.observer = observer
self.cancel = cancel
super.init()
self.cachedScheduleLambda = { pair in
guard !cancel.isDisposed else { return Disposables.create() }
pair.sink.observer.on(pair.event)
if pair.event.isStopEvent {
pair.sink.dispose()
}
return Disposables.create()
}
}
override func onCore(_ event: Event<Element>) {
_ = self.scheduler.schedule((self, event), action: self.cachedScheduleLambda!)
}
override func dispose() {
super.dispose()
self.cancel.dispose()
}
}
final private class ObserveOnSerialDispatchQueue<Element>: Producer<Element> {
let scheduler: SerialDispatchQueueScheduler
let source: Observable<Element>
init(source: Observable<Element>, scheduler: SerialDispatchQueueScheduler) {
self.scheduler = scheduler
self.source = source
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
_ = increment(numberOfSerialDispatchObservables)
#endif
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = ObserveOnSerialDispatchQueueSink(scheduler: self.scheduler, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
#if TRACE_RESOURCES
deinit {
_ = Resources.decrementTotal()
_ = decrement(numberOfSerialDispatchObservables)
}
#endif
}

View File

@@ -0,0 +1,95 @@
//
// Optional.swift
// RxSwift
//
// Created by tarunon on 2016/12/13.
// Copyright © 2016 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Converts a optional to an observable sequence.
- seealso: [from operator on reactivex.io](http://reactivex.io/documentation/operators/from.html)
- parameter optional: Optional element in the resulting observable sequence.
- returns: An observable sequence containing the wrapped value or not from given optional.
*/
public static func from(optional: Element?) -> Observable<Element> {
ObservableOptional(optional: optional)
}
/**
Converts a optional to an observable sequence.
- seealso: [from operator on reactivex.io](http://reactivex.io/documentation/operators/from.html)
- parameter optional: Optional element in the resulting observable sequence.
- parameter scheduler: Scheduler to send the optional element on.
- returns: An observable sequence containing the wrapped value or not from given optional.
*/
public static func from(optional: Element?, scheduler: ImmediateSchedulerType) -> Observable<Element> {
ObservableOptionalScheduled(optional: optional, scheduler: scheduler)
}
}
final private class ObservableOptionalScheduledSink<Observer: ObserverType>: Sink<Observer> {
typealias Element = Observer.Element
typealias Parent = ObservableOptionalScheduled<Element>
private let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
return self.parent.scheduler.schedule(self.parent.optional) { (optional: Element?) -> Disposable in
if let next = optional {
self.forwardOn(.next(next))
return self.parent.scheduler.schedule(()) { _ in
self.forwardOn(.completed)
self.dispose()
return Disposables.create()
}
} else {
self.forwardOn(.completed)
self.dispose()
return Disposables.create()
}
}
}
}
final private class ObservableOptionalScheduled<Element>: Producer<Element> {
fileprivate let optional: Element?
fileprivate let scheduler: ImmediateSchedulerType
init(optional: Element?, scheduler: ImmediateSchedulerType) {
self.optional = optional
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = ObservableOptionalScheduledSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
final private class ObservableOptional<Element>: Producer<Element> {
private let optional: Element?
init(optional: Element?) {
self.optional = optional
}
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
if let element = self.optional {
observer.on(.next(element))
}
observer.on(.completed)
return Disposables.create()
}
}

View File

@@ -0,0 +1,92 @@
//
// Producer.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/20/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
class Producer<Element>: Observable<Element> {
override init() {
super.init()
}
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
if !CurrentThreadScheduler.isScheduleRequired {
// The returned disposable needs to release all references once it was disposed.
let disposer = SinkDisposer()
let sinkAndSubscription = self.run(observer, cancel: disposer)
disposer.setSinkAndSubscription(sink: sinkAndSubscription.sink, subscription: sinkAndSubscription.subscription)
return disposer
}
else {
return CurrentThreadScheduler.instance.schedule(()) { _ in
let disposer = SinkDisposer()
let sinkAndSubscription = self.run(observer, cancel: disposer)
disposer.setSinkAndSubscription(sink: sinkAndSubscription.sink, subscription: sinkAndSubscription.subscription)
return disposer
}
}
}
func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
rxAbstractMethod()
}
}
private final class SinkDisposer: Cancelable {
private enum DisposeState: Int32 {
case disposed = 1
case sinkAndSubscriptionSet = 2
}
private let state = AtomicInt(0)
private var sink: Disposable?
private var subscription: Disposable?
var isDisposed: Bool {
isFlagSet(self.state, DisposeState.disposed.rawValue)
}
func setSinkAndSubscription(sink: Disposable, subscription: Disposable) {
self.sink = sink
self.subscription = subscription
let previousState = fetchOr(self.state, DisposeState.sinkAndSubscriptionSet.rawValue)
if (previousState & DisposeState.sinkAndSubscriptionSet.rawValue) != 0 {
rxFatalError("Sink and subscription were already set")
}
if (previousState & DisposeState.disposed.rawValue) != 0 {
sink.dispose()
subscription.dispose()
self.sink = nil
self.subscription = nil
}
}
func dispose() {
let previousState = fetchOr(self.state, DisposeState.disposed.rawValue)
if (previousState & DisposeState.disposed.rawValue) != 0 {
return
}
if (previousState & DisposeState.sinkAndSubscriptionSet.rawValue) != 0 {
guard let sink = self.sink else {
rxFatalError("Sink not set")
}
guard let subscription = self.subscription else {
rxFatalError("Subscription not set")
}
sink.dispose()
subscription.dispose()
self.sink = nil
self.subscription = nil
}
}
}

View File

@@ -0,0 +1,73 @@
//
// Range.swift
// RxSwift
//
// Created by Krunoslav Zaher on 9/13/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType where Element: RxAbstractInteger {
/**
Generates an observable sequence of integral numbers within a specified range, using the specified scheduler to generate and send out observer messages.
- seealso: [range operator on reactivex.io](http://reactivex.io/documentation/operators/range.html)
- parameter start: The value of the first integer in the sequence.
- parameter count: The number of sequential integers to generate.
- parameter scheduler: Scheduler to run the generator loop on.
- returns: An observable sequence that contains a range of sequential integral numbers.
*/
public static func range(start: Element, count: Element, scheduler: ImmediateSchedulerType = CurrentThreadScheduler.instance) -> Observable<Element> {
RangeProducer<Element>(start: start, count: count, scheduler: scheduler)
}
}
final private class RangeProducer<Element: RxAbstractInteger>: Producer<Element> {
fileprivate let start: Element
fileprivate let count: Element
fileprivate let scheduler: ImmediateSchedulerType
init(start: Element, count: Element, scheduler: ImmediateSchedulerType) {
guard count >= 0 else {
rxFatalError("count can't be negative")
}
guard start &+ (count - 1) >= start || count == 0 else {
rxFatalError("overflow of count")
}
self.start = start
self.count = count
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = RangeSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
final private class RangeSink<Observer: ObserverType>: Sink<Observer> where Observer.Element: RxAbstractInteger {
typealias Parent = RangeProducer<Observer.Element>
private let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
return self.parent.scheduler.scheduleRecursive(0 as Observer.Element) { i, recurse in
if i < self.parent.count {
self.forwardOn(.next(self.parent.start + i))
recurse(i + 1)
}
else {
self.forwardOn(.completed)
self.dispose()
}
}
}
}

View File

@@ -0,0 +1,109 @@
//
// Reduce.swift
// RxSwift
//
// Created by Krunoslav Zaher on 4/1/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Applies an `accumulator` function over an observable sequence, returning the result of the aggregation as a single element in the result sequence. The specified `seed` value is used as the initial accumulator value.
For aggregation behavior with incremental intermediate results, see `scan`.
- seealso: [reduce operator on reactivex.io](http://reactivex.io/documentation/operators/reduce.html)
- parameter seed: The initial accumulator value.
- parameter accumulator: A accumulator function to be invoked on each element.
- parameter mapResult: A function to transform the final accumulator value into the result value.
- returns: An observable sequence containing a single element with the final accumulator value.
*/
public func reduce<A, Result>(_ seed: A, accumulator: @escaping (A, Element) throws -> A, mapResult: @escaping (A) throws -> Result)
-> Observable<Result> {
Reduce(source: self.asObservable(), seed: seed, accumulator: accumulator, mapResult: mapResult)
}
/**
Applies an `accumulator` function over an observable sequence, returning the result of the aggregation as a single element in the result sequence. The specified `seed` value is used as the initial accumulator value.
For aggregation behavior with incremental intermediate results, see `scan`.
- seealso: [reduce operator on reactivex.io](http://reactivex.io/documentation/operators/reduce.html)
- parameter seed: The initial accumulator value.
- parameter accumulator: A accumulator function to be invoked on each element.
- returns: An observable sequence containing a single element with the final accumulator value.
*/
public func reduce<A>(_ seed: A, accumulator: @escaping (A, Element) throws -> A)
-> Observable<A> {
Reduce(source: self.asObservable(), seed: seed, accumulator: accumulator, mapResult: { $0 })
}
}
final private class ReduceSink<SourceType, AccumulateType, Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias ResultType = Observer.Element
typealias Parent = Reduce<SourceType, AccumulateType, ResultType>
private let parent: Parent
private var accumulation: AccumulateType
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.accumulation = parent.seed
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<SourceType>) {
switch event {
case .next(let value):
do {
self.accumulation = try self.parent.accumulator(self.accumulation, value)
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
case .error(let e):
self.forwardOn(.error(e))
self.dispose()
case .completed:
do {
let result = try self.parent.mapResult(self.accumulation)
self.forwardOn(.next(result))
self.forwardOn(.completed)
self.dispose()
}
catch let e {
self.forwardOn(.error(e))
self.dispose()
}
}
}
}
final private class Reduce<SourceType, AccumulateType, ResultType>: Producer<ResultType> {
typealias AccumulatorType = (AccumulateType, SourceType) throws -> AccumulateType
typealias ResultSelectorType = (AccumulateType) throws -> ResultType
private let source: Observable<SourceType>
fileprivate let seed: AccumulateType
fileprivate let accumulator: AccumulatorType
fileprivate let mapResult: ResultSelectorType
init(source: Observable<SourceType>, seed: AccumulateType, accumulator: @escaping AccumulatorType, mapResult: @escaping ResultSelectorType) {
self.source = source
self.seed = seed
self.accumulator = accumulator
self.mapResult = mapResult
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == ResultType {
let sink = ReduceSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,57 @@
//
// Repeat.swift
// RxSwift
//
// Created by Krunoslav Zaher on 9/13/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Generates an observable sequence that repeats the given element infinitely, using the specified scheduler to send out observer messages.
- seealso: [repeat operator on reactivex.io](http://reactivex.io/documentation/operators/repeat.html)
- parameter element: Element to repeat.
- parameter scheduler: Scheduler to run the producer loop on.
- returns: An observable sequence that repeats the given element infinitely.
*/
public static func repeatElement(_ element: Element, scheduler: ImmediateSchedulerType = CurrentThreadScheduler.instance) -> Observable<Element> {
RepeatElement(element: element, scheduler: scheduler)
}
}
final private class RepeatElement<Element>: Producer<Element> {
fileprivate let element: Element
fileprivate let scheduler: ImmediateSchedulerType
init(element: Element, scheduler: ImmediateSchedulerType) {
self.element = element
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = RepeatElementSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
final private class RepeatElementSink<Observer: ObserverType>: Sink<Observer> {
typealias Parent = RepeatElement<Observer.Element>
private let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
return self.parent.scheduler.scheduleRecursive(self.parent.element) { e, recurse in
self.forwardOn(.next(e))
recurse(e)
}
}
}

View File

@@ -0,0 +1,211 @@
//
// RetryWhen.swift
// RxSwift
//
// Created by Junior B. on 06/10/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Repeats the source observable sequence on error when the notifier emits a next value.
If the source observable errors and the notifier completes, it will complete the source sequence.
- seealso: [retry operator on reactivex.io](http://reactivex.io/documentation/operators/retry.html)
- parameter notificationHandler: A handler that is passed an observable sequence of errors raised by the source observable and returns and observable that either continues, completes or errors. This behavior is then applied to the source observable.
- returns: An observable sequence producing the elements of the given sequence repeatedly until it terminates successfully or is notified to error or complete.
*/
public func retry<TriggerObservable: ObservableType, Error: Swift.Error>(when notificationHandler: @escaping (Observable<Error>) -> TriggerObservable)
-> Observable<Element> {
RetryWhenSequence(sources: InfiniteSequence(repeatedValue: self.asObservable()), notificationHandler: notificationHandler)
}
/**
Repeats the source observable sequence on error when the notifier emits a next value.
If the source observable errors and the notifier completes, it will complete the source sequence.
- seealso: [retry operator on reactivex.io](http://reactivex.io/documentation/operators/retry.html)
- parameter notificationHandler: A handler that is passed an observable sequence of errors raised by the source observable and returns and observable that either continues, completes or errors. This behavior is then applied to the source observable.
- returns: An observable sequence producing the elements of the given sequence repeatedly until it terminates successfully or is notified to error or complete.
*/
@available(*, deprecated, renamed: "retry(when:)")
public func retryWhen<TriggerObservable: ObservableType, Error: Swift.Error>(_ notificationHandler: @escaping (Observable<Error>) -> TriggerObservable)
-> Observable<Element> {
retry(when: notificationHandler)
}
/**
Repeats the source observable sequence on error when the notifier emits a next value.
If the source observable errors and the notifier completes, it will complete the source sequence.
- seealso: [retry operator on reactivex.io](http://reactivex.io/documentation/operators/retry.html)
- parameter notificationHandler: A handler that is passed an observable sequence of errors raised by the source observable and returns and observable that either continues, completes or errors. This behavior is then applied to the source observable.
- returns: An observable sequence producing the elements of the given sequence repeatedly until it terminates successfully or is notified to error or complete.
*/
public func retry<TriggerObservable: ObservableType>(when notificationHandler: @escaping (Observable<Swift.Error>) -> TriggerObservable)
-> Observable<Element> {
RetryWhenSequence(sources: InfiniteSequence(repeatedValue: self.asObservable()), notificationHandler: notificationHandler)
}
/**
Repeats the source observable sequence on error when the notifier emits a next value.
If the source observable errors and the notifier completes, it will complete the source sequence.
- seealso: [retry operator on reactivex.io](http://reactivex.io/documentation/operators/retry.html)
- parameter notificationHandler: A handler that is passed an observable sequence of errors raised by the source observable and returns and observable that either continues, completes or errors. This behavior is then applied to the source observable.
- returns: An observable sequence producing the elements of the given sequence repeatedly until it terminates successfully or is notified to error or complete.
*/
@available(*, deprecated, renamed: "retry(when:)")
public func retryWhen<TriggerObservable: ObservableType>(_ notificationHandler: @escaping (Observable<Swift.Error>) -> TriggerObservable)
-> Observable<Element> {
RetryWhenSequence(sources: InfiniteSequence(repeatedValue: self.asObservable()), notificationHandler: notificationHandler)
}
}
final private class RetryTriggerSink<Sequence: Swift.Sequence, Observer: ObserverType, TriggerObservable: ObservableType, Error>
: ObserverType where Sequence.Element: ObservableType, Sequence.Element.Element == Observer.Element {
typealias Element = TriggerObservable.Element
typealias Parent = RetryWhenSequenceSinkIter<Sequence, Observer, TriggerObservable, Error>
private let parent: Parent
init(parent: Parent) {
self.parent = parent
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.parent.parent.lastError = nil
self.parent.parent.schedule(.moveNext)
case .error(let e):
self.parent.parent.forwardOn(.error(e))
self.parent.parent.dispose()
case .completed:
self.parent.parent.forwardOn(.completed)
self.parent.parent.dispose()
}
}
}
final private class RetryWhenSequenceSinkIter<Sequence: Swift.Sequence, Observer: ObserverType, TriggerObservable: ObservableType, Error>
: ObserverType
, Disposable where Sequence.Element: ObservableType, Sequence.Element.Element == Observer.Element {
typealias Element = Observer.Element
typealias Parent = RetryWhenSequenceSink<Sequence, Observer, TriggerObservable, Error>
fileprivate let parent: Parent
private let errorHandlerSubscription = SingleAssignmentDisposable()
private let subscription: Disposable
init(parent: Parent, subscription: Disposable) {
self.parent = parent
self.subscription = subscription
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.parent.forwardOn(event)
case .error(let error):
self.parent.lastError = error
if let failedWith = error as? Error {
// dispose current subscription
self.subscription.dispose()
let errorHandlerSubscription = self.parent.notifier.subscribe(RetryTriggerSink(parent: self))
self.errorHandlerSubscription.setDisposable(errorHandlerSubscription)
self.parent.errorSubject.on(.next(failedWith))
}
else {
self.parent.forwardOn(.error(error))
self.parent.dispose()
}
case .completed:
self.parent.forwardOn(event)
self.parent.dispose()
}
}
final func dispose() {
self.subscription.dispose()
self.errorHandlerSubscription.dispose()
}
}
final private class RetryWhenSequenceSink<Sequence: Swift.Sequence, Observer: ObserverType, TriggerObservable: ObservableType, Error>
: TailRecursiveSink<Sequence, Observer> where Sequence.Element: ObservableType, Sequence.Element.Element == Observer.Element {
typealias Element = Observer.Element
typealias Parent = RetryWhenSequence<Sequence, TriggerObservable, Error>
let lock = RecursiveLock()
private let parent: Parent
fileprivate var lastError: Swift.Error?
fileprivate let errorSubject = PublishSubject<Error>()
private let handler: Observable<TriggerObservable.Element>
fileprivate let notifier = PublishSubject<TriggerObservable.Element>()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.handler = parent.notificationHandler(self.errorSubject).asObservable()
super.init(observer: observer, cancel: cancel)
}
override func done() {
if let lastError = self.lastError {
self.forwardOn(.error(lastError))
self.lastError = nil
}
else {
self.forwardOn(.completed)
}
self.dispose()
}
override func extract(_ observable: Observable<Element>) -> SequenceGenerator? {
// It is important to always return `nil` here because there are side effects in the `run` method
// that are dependant on particular `retryWhen` operator so single operator stack can't be reused in this
// case.
return nil
}
override func subscribeToNext(_ source: Observable<Element>) -> Disposable {
let subscription = SingleAssignmentDisposable()
let iter = RetryWhenSequenceSinkIter(parent: self, subscription: subscription)
subscription.setDisposable(source.subscribe(iter))
return iter
}
override func run(_ sources: SequenceGenerator) -> Disposable {
let triggerSubscription = self.handler.subscribe(self.notifier.asObserver())
let superSubscription = super.run(sources)
return Disposables.create(superSubscription, triggerSubscription)
}
}
final private class RetryWhenSequence<Sequence: Swift.Sequence, TriggerObservable: ObservableType, Error>: Producer<Sequence.Element.Element> where Sequence.Element: ObservableType {
typealias Element = Sequence.Element.Element
private let sources: Sequence
fileprivate let notificationHandler: (Observable<Error>) -> TriggerObservable
init(sources: Sequence, notificationHandler: @escaping (Observable<Error>) -> TriggerObservable) {
self.sources = sources
self.notificationHandler = notificationHandler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = RetryWhenSequenceSink<Sequence, Observer, TriggerObservable, Error>(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run((self.sources.makeIterator(), nil))
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,139 @@
//
// Sample.swift
// RxSwift
//
// Created by Krunoslav Zaher on 5/1/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Samples the source observable sequence using a sampler observable sequence producing sampling ticks.
Upon each sampling tick, the latest element (if any) in the source sequence during the last sampling interval is sent to the resulting sequence.
**In case there were no new elements between sampler ticks, you may provide a default value to be emitted, instead
to the resulting sequence otherwise no element is sent.**
- seealso: [sample operator on reactivex.io](http://reactivex.io/documentation/operators/sample.html)
- parameter sampler: Sampling tick sequence.
- parameter defaultValue: a value to return if there are no new elements between sampler ticks
- returns: Sampled observable sequence.
*/
public func sample<Source: ObservableType>(_ sampler: Source, defaultValue: Element? = nil)
-> Observable<Element> {
return Sample(source: self.asObservable(), sampler: sampler.asObservable(), defaultValue: defaultValue)
}
}
final private class SamplerSink<Observer: ObserverType, SampleType>
: ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias Element = SampleType
typealias Parent = SampleSequenceSink<Observer, SampleType>
private let parent: Parent
var lock: RecursiveLock {
self.parent.lock
}
init(parent: Parent) {
self.parent = parent
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next, .completed:
if let element = parent.element ?? self.parent.defaultValue {
self.parent.element = nil
self.parent.forwardOn(.next(element))
}
if self.parent.atEnd {
self.parent.forwardOn(.completed)
self.parent.dispose()
}
case .error(let e):
self.parent.forwardOn(.error(e))
self.parent.dispose()
}
}
}
final private class SampleSequenceSink<Observer: ObserverType, SampleType>
: Sink<Observer>
, ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias Element = Observer.Element
typealias Parent = Sample<Element, SampleType>
fileprivate let parent: Parent
fileprivate let defaultValue: Element?
let lock = RecursiveLock()
// state
fileprivate var element = nil as Element?
fileprivate var atEnd = false
private let sourceSubscription = SingleAssignmentDisposable()
init(parent: Parent, observer: Observer, cancel: Cancelable, defaultValue: Element? = nil) {
self.parent = parent
self.defaultValue = defaultValue
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
self.sourceSubscription.setDisposable(self.parent.source.subscribe(self))
let samplerSubscription = self.parent.sampler.subscribe(SamplerSink(parent: self))
return Disposables.create(sourceSubscription, samplerSubscription)
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next(let element):
self.element = element
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
self.atEnd = true
self.sourceSubscription.dispose()
}
}
}
final private class Sample<Element, SampleType>: Producer<Element> {
fileprivate let source: Observable<Element>
fileprivate let sampler: Observable<SampleType>
fileprivate let defaultValue: Element?
init(source: Observable<Element>, sampler: Observable<SampleType>, defaultValue: Element? = nil) {
self.source = source
self.sampler = sampler
self.defaultValue = defaultValue
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = SampleSequenceSink(parent: self, observer: observer, cancel: cancel, defaultValue: self.defaultValue)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,100 @@
//
// Scan.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/14/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Applies an accumulator function over an observable sequence and returns each intermediate result. The specified seed value is used as the initial accumulator value.
For aggregation behavior with no intermediate results, see `reduce`.
- seealso: [scan operator on reactivex.io](http://reactivex.io/documentation/operators/scan.html)
- parameter seed: The initial accumulator value.
- parameter accumulator: An accumulator function to be invoked on each element.
- returns: An observable sequence containing the accumulated values.
*/
public func scan<A>(into seed: A, accumulator: @escaping (inout A, Element) throws -> Void)
-> Observable<A> {
Scan(source: self.asObservable(), seed: seed, accumulator: accumulator)
}
/**
Applies an accumulator function over an observable sequence and returns each intermediate result. The specified seed value is used as the initial accumulator value.
For aggregation behavior with no intermediate results, see `reduce`.
- seealso: [scan operator on reactivex.io](http://reactivex.io/documentation/operators/scan.html)
- parameter seed: The initial accumulator value.
- parameter accumulator: An accumulator function to be invoked on each element.
- returns: An observable sequence containing the accumulated values.
*/
public func scan<A>(_ seed: A, accumulator: @escaping (A, Element) throws -> A)
-> Observable<A> {
return Scan(source: self.asObservable(), seed: seed) { acc, element in
let currentAcc = acc
acc = try accumulator(currentAcc, element)
}
}
}
final private class ScanSink<Element, Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Accumulate = Observer.Element
typealias Parent = Scan<Element, Accumulate>
private let parent: Parent
private var accumulate: Accumulate
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.accumulate = parent.seed
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let element):
do {
try self.parent.accumulator(&self.accumulate, element)
self.forwardOn(.next(self.accumulate))
}
catch let error {
self.forwardOn(.error(error))
self.dispose()
}
case .error(let error):
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class Scan<Element, Accumulate>: Producer<Accumulate> {
typealias Accumulator = (inout Accumulate, Element) throws -> Void
private let source: Observable<Element>
fileprivate let seed: Accumulate
fileprivate let accumulator: Accumulator
init(source: Observable<Element>, seed: Accumulate, accumulator: @escaping Accumulator) {
self.source = source
self.seed = seed
self.accumulator = accumulator
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Accumulate {
let sink = ScanSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,89 @@
//
// Sequence.swift
// RxSwift
//
// Created by Krunoslav Zaher on 11/14/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
// MARK: of
/**
This method creates a new Observable instance with a variable number of elements.
- seealso: [from operator on reactivex.io](http://reactivex.io/documentation/operators/from.html)
- parameter elements: Elements to generate.
- parameter scheduler: Scheduler to send elements on. If `nil`, elements are sent immediately on subscription.
- returns: The observable sequence whose elements are pulled from the given arguments.
*/
public static func of(_ elements: Element ..., scheduler: ImmediateSchedulerType = CurrentThreadScheduler.instance) -> Observable<Element> {
ObservableSequence(elements: elements, scheduler: scheduler)
}
}
extension ObservableType {
/**
Converts an array to an observable sequence.
- seealso: [from operator on reactivex.io](http://reactivex.io/documentation/operators/from.html)
- returns: The observable sequence whose elements are pulled from the given enumerable sequence.
*/
public static func from(_ array: [Element], scheduler: ImmediateSchedulerType = CurrentThreadScheduler.instance) -> Observable<Element> {
ObservableSequence(elements: array, scheduler: scheduler)
}
/**
Converts a sequence to an observable sequence.
- seealso: [from operator on reactivex.io](http://reactivex.io/documentation/operators/from.html)
- returns: The observable sequence whose elements are pulled from the given enumerable sequence.
*/
public static func from<Sequence: Swift.Sequence>(_ sequence: Sequence, scheduler: ImmediateSchedulerType = CurrentThreadScheduler.instance) -> Observable<Element> where Sequence.Element == Element {
ObservableSequence(elements: sequence, scheduler: scheduler)
}
}
final private class ObservableSequenceSink<Sequence: Swift.Sequence, Observer: ObserverType>: Sink<Observer> where Sequence.Element == Observer.Element {
typealias Parent = ObservableSequence<Sequence>
private let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
return self.parent.scheduler.scheduleRecursive(self.parent.elements.makeIterator()) { iterator, recurse in
var mutableIterator = iterator
if let next = mutableIterator.next() {
self.forwardOn(.next(next))
recurse(mutableIterator)
}
else {
self.forwardOn(.completed)
self.dispose()
}
}
}
}
final private class ObservableSequence<Sequence: Swift.Sequence>: Producer<Sequence.Element> {
fileprivate let elements: Sequence
fileprivate let scheduler: ImmediateSchedulerType
init(elements: Sequence, scheduler: ImmediateSchedulerType) {
self.elements = elements
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = ObservableSequenceSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,443 @@
//
// ShareReplayScope.swift
// RxSwift
//
// Created by Krunoslav Zaher on 5/28/17.
// Copyright © 2017 Krunoslav Zaher. All rights reserved.
//
/// Subject lifetime scope
public enum SubjectLifetimeScope {
/**
**Each connection will have it's own subject instance to store replay events.**
**Connections will be isolated from each another.**
Configures the underlying implementation to behave equivalent to.
```
source.multicast(makeSubject: { MySubject() }).refCount()
```
**This is the recommended default.**
This has the following consequences:
* `retry` or `concat` operators will function as expected because terminating the sequence will clear internal state.
* Each connection to source observable sequence will use it's own subject.
* When the number of subscribers drops from 1 to 0 and connection to source sequence is disposed, subject will be cleared.
```
let xs = Observable.deferred { () -> Observable<TimeInterval> in
print("Performing work ...")
return Observable.just(Date().timeIntervalSince1970)
}
.share(replay: 1, scope: .whileConnected)
_ = xs.subscribe(onNext: { print("next \($0)") }, onCompleted: { print("completed\n") })
_ = xs.subscribe(onNext: { print("next \($0)") }, onCompleted: { print("completed\n") })
_ = xs.subscribe(onNext: { print("next \($0)") }, onCompleted: { print("completed\n") })
```
Notice how time interval is different and `Performing work ...` is printed each time)
```
Performing work ...
next 1495998900.82141
completed
Performing work ...
next 1495998900.82359
completed
Performing work ...
next 1495998900.82444
completed
```
*/
case whileConnected
/**
**One subject will store replay events for all connections to source.**
**Connections won't be isolated from each another.**
Configures the underlying implementation behave equivalent to.
```
source.multicast(MySubject()).refCount()
```
This has the following consequences:
* Using `retry` or `concat` operators after this operator usually isn't advised.
* Each connection to source observable sequence will share the same subject.
* After number of subscribers drops from 1 to 0 and connection to source observable sequence is dispose, this operator will
continue holding a reference to the same subject.
If at some later moment a new observer initiates a new connection to source it can potentially receive
some of the stale events received during previous connection.
* After source sequence terminates any new observer will always immediately receive replayed elements and terminal event.
No new subscriptions to source observable sequence will be attempted.
```
let xs = Observable.deferred { () -> Observable<TimeInterval> in
print("Performing work ...")
return Observable.just(Date().timeIntervalSince1970)
}
.share(replay: 1, scope: .forever)
_ = xs.subscribe(onNext: { print("next \($0)") }, onCompleted: { print("completed\n") })
_ = xs.subscribe(onNext: { print("next \($0)") }, onCompleted: { print("completed\n") })
_ = xs.subscribe(onNext: { print("next \($0)") }, onCompleted: { print("completed\n") })
```
Notice how time interval is the same, replayed, and `Performing work ...` is printed only once
```
Performing work ...
next 1495999013.76356
completed
next 1495999013.76356
completed
next 1495999013.76356
completed
```
*/
case forever
}
extension ObservableType {
/**
Returns an observable sequence that **shares a single subscription to the underlying sequence**, and immediately upon subscription replays elements in buffer.
This operator is equivalent to:
* `.whileConnected`
```
// Each connection will have it's own subject instance to store replay events.
// Connections will be isolated from each another.
source.multicast(makeSubject: { Replay.create(bufferSize: replay) }).refCount()
```
* `.forever`
```
// One subject will store replay events for all connections to source.
// Connections won't be isolated from each another.
source.multicast(Replay.create(bufferSize: replay)).refCount()
```
It uses optimized versions of the operators for most common operations.
- parameter replay: Maximum element count of the replay buffer.
- parameter scope: Lifetime scope of sharing subject. For more information see `SubjectLifetimeScope` enum.
- seealso: [shareReplay operator on reactivex.io](http://reactivex.io/documentation/operators/replay.html)
- returns: An observable sequence that contains the elements of a sequence produced by multicasting the source sequence.
*/
public func share(replay: Int = 0, scope: SubjectLifetimeScope = .whileConnected)
-> Observable<Element> {
switch scope {
case .forever:
switch replay {
case 0: return self.multicast(PublishSubject()).refCount()
default: return self.multicast(ReplaySubject.create(bufferSize: replay)).refCount()
}
case .whileConnected:
switch replay {
case 0: return ShareWhileConnected(source: self.asObservable())
case 1: return ShareReplay1WhileConnected(source: self.asObservable())
default: return self.multicast(makeSubject: { ReplaySubject.create(bufferSize: replay) }).refCount()
}
}
}
}
private final class ShareReplay1WhileConnectedConnection<Element>
: ObserverType
, SynchronizedUnsubscribeType {
typealias Observers = AnyObserver<Element>.s
typealias DisposeKey = Observers.KeyType
typealias Parent = ShareReplay1WhileConnected<Element>
private let parent: Parent
private let subscription = SingleAssignmentDisposable()
private let lock: RecursiveLock
private var disposed: Bool = false
fileprivate var observers = Observers()
private var element: Element?
init(parent: Parent, lock: RecursiveLock) {
self.parent = parent
self.lock = lock
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
}
final func on(_ event: Event<Element>) {
let observers = self.lock.performLocked { self.synchronized_on(event) }
dispatch(observers, event)
}
final private func synchronized_on(_ event: Event<Element>) -> Observers {
if self.disposed {
return Observers()
}
switch event {
case .next(let element):
self.element = element
return self.observers
case .error, .completed:
let observers = self.observers
self.synchronized_dispose()
return observers
}
}
final func connect() {
self.subscription.setDisposable(self.parent.source.subscribe(self))
}
final func synchronized_subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
self.lock.performLocked {
if let element = self.element {
observer.on(.next(element))
}
let disposeKey = self.observers.insert(observer.on)
return SubscriptionDisposable(owner: self, key: disposeKey)
}
}
final private func synchronized_dispose() {
self.disposed = true
if self.parent.connection === self {
self.parent.connection = nil
}
self.observers = Observers()
}
final func synchronizedUnsubscribe(_ disposeKey: DisposeKey) {
if self.lock.performLocked({ self.synchronized_unsubscribe(disposeKey) }) {
self.subscription.dispose()
}
}
@inline(__always)
final private func synchronized_unsubscribe(_ disposeKey: DisposeKey) -> Bool {
// if already unsubscribed, just return
if self.observers.removeKey(disposeKey) == nil {
return false
}
if self.observers.count == 0 {
self.synchronized_dispose()
return true
}
return false
}
#if TRACE_RESOURCES
deinit {
_ = Resources.decrementTotal()
}
#endif
}
// optimized version of share replay for most common case
final private class ShareReplay1WhileConnected<Element>
: Observable<Element> {
fileprivate typealias Connection = ShareReplay1WhileConnectedConnection<Element>
fileprivate let source: Observable<Element>
private let lock = RecursiveLock()
fileprivate var connection: Connection?
init(source: Observable<Element>) {
self.source = source
}
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
self.lock.lock()
let connection = self.synchronized_subscribe(observer)
let count = connection.observers.count
let disposable = connection.synchronized_subscribe(observer)
self.lock.unlock()
if count == 0 {
connection.connect()
}
return disposable
}
@inline(__always)
private func synchronized_subscribe<Observer: ObserverType>(_ observer: Observer) -> Connection where Observer.Element == Element {
let connection: Connection
if let existingConnection = self.connection {
connection = existingConnection
}
else {
connection = ShareReplay1WhileConnectedConnection<Element>(
parent: self,
lock: self.lock)
self.connection = connection
}
return connection
}
}
private final class ShareWhileConnectedConnection<Element>
: ObserverType
, SynchronizedUnsubscribeType {
typealias Observers = AnyObserver<Element>.s
typealias DisposeKey = Observers.KeyType
typealias Parent = ShareWhileConnected<Element>
private let parent: Parent
private let subscription = SingleAssignmentDisposable()
private let lock: RecursiveLock
private var disposed: Bool = false
fileprivate var observers = Observers()
init(parent: Parent, lock: RecursiveLock) {
self.parent = parent
self.lock = lock
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
}
final func on(_ event: Event<Element>) {
let observers = self.lock.performLocked { self.synchronized_on(event) }
dispatch(observers, event)
}
final private func synchronized_on(_ event: Event<Element>) -> Observers {
if self.disposed {
return Observers()
}
switch event {
case .next:
return self.observers
case .error, .completed:
let observers = self.observers
self.synchronized_dispose()
return observers
}
}
final func connect() {
self.subscription.setDisposable(self.parent.source.subscribe(self))
}
final func synchronized_subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
self.lock.performLocked {
let disposeKey = self.observers.insert(observer.on)
return SubscriptionDisposable(owner: self, key: disposeKey)
}
}
final private func synchronized_dispose() {
self.disposed = true
if self.parent.connection === self {
self.parent.connection = nil
}
self.observers = Observers()
}
final func synchronizedUnsubscribe(_ disposeKey: DisposeKey) {
if self.lock.performLocked({ self.synchronized_unsubscribe(disposeKey) }) {
self.subscription.dispose()
}
}
@inline(__always)
final private func synchronized_unsubscribe(_ disposeKey: DisposeKey) -> Bool {
// if already unsubscribed, just return
if self.observers.removeKey(disposeKey) == nil {
return false
}
if self.observers.count == 0 {
self.synchronized_dispose()
return true
}
return false
}
#if TRACE_RESOURCES
deinit {
_ = Resources.decrementTotal()
}
#endif
}
// optimized version of share replay for most common case
final private class ShareWhileConnected<Element>
: Observable<Element> {
fileprivate typealias Connection = ShareWhileConnectedConnection<Element>
fileprivate let source: Observable<Element>
private let lock = RecursiveLock()
fileprivate var connection: Connection?
init(source: Observable<Element>) {
self.source = source
}
override func subscribe<Observer: ObserverType>(_ observer: Observer) -> Disposable where Observer.Element == Element {
self.lock.lock()
let connection = self.synchronized_subscribe(observer)
let count = connection.observers.count
let disposable = connection.synchronized_subscribe(observer)
self.lock.unlock()
if count == 0 {
connection.connect()
}
return disposable
}
@inline(__always)
private func synchronized_subscribe<Observer: ObserverType>(_ observer: Observer) -> Connection where Observer.Element == Element {
let connection: Connection
if let existingConnection = self.connection {
connection = existingConnection
}
else {
connection = ShareWhileConnectedConnection<Element>(
parent: self,
lock: self.lock)
self.connection = connection
}
return connection
}
}

View File

@@ -0,0 +1,104 @@
//
// SingleAsync.swift
// RxSwift
//
// Created by Junior B. on 09/11/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
The single operator is similar to first, but throws a `RxError.noElements` or `RxError.moreThanOneElement`
if the source Observable does not emit exactly one element before successfully completing.
- seealso: [single operator on reactivex.io](http://reactivex.io/documentation/operators/first.html)
- returns: An observable sequence that emits a single element or throws an exception if more (or none) of them are emitted.
*/
public func single()
-> Observable<Element> {
SingleAsync(source: self.asObservable())
}
/**
The single operator is similar to first, but throws a `RxError.NoElements` or `RxError.MoreThanOneElement`
if the source Observable does not emit exactly one element before successfully completing.
- seealso: [single operator on reactivex.io](http://reactivex.io/documentation/operators/first.html)
- parameter predicate: A function to test each source element for a condition.
- returns: An observable sequence that emits a single element or throws an exception if more (or none) of them are emitted.
*/
public func single(_ predicate: @escaping (Element) throws -> Bool)
-> Observable<Element> {
SingleAsync(source: self.asObservable(), predicate: predicate)
}
}
private final class SingleAsyncSink<Observer: ObserverType> : Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = SingleAsync<Element>
private let parent: Parent
private var seenValue: Bool = false
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
do {
let forward = try self.parent.predicate?(value) ?? true
if !forward {
return
}
}
catch let error {
self.forwardOn(.error(error as Swift.Error))
self.dispose()
return
}
if self.seenValue {
self.forwardOn(.error(RxError.moreThanOneElement))
self.dispose()
return
}
self.seenValue = true
self.forwardOn(.next(value))
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
if self.seenValue {
self.forwardOn(.completed)
} else {
self.forwardOn(.error(RxError.noElements))
}
self.dispose()
}
}
}
final class SingleAsync<Element>: Producer<Element> {
typealias Predicate = (Element) throws -> Bool
private let source: Observable<Element>
fileprivate let predicate: Predicate?
init(source: Observable<Element>, predicate: Predicate? = nil) {
self.source = source
self.predicate = predicate
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = SingleAsyncSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,75 @@
//
// Sink.swift
// RxSwift
//
// Created by Krunoslav Zaher on 2/19/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
class Sink<Observer: ObserverType>: Disposable {
fileprivate let observer: Observer
fileprivate let cancel: Cancelable
private let disposed = AtomicInt(0)
#if DEBUG
private let synchronizationTracker = SynchronizationTracker()
#endif
init(observer: Observer, cancel: Cancelable) {
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
self.observer = observer
self.cancel = cancel
}
final func forwardOn(_ event: Event<Observer.Element>) {
#if DEBUG
self.synchronizationTracker.register(synchronizationErrorMessage: .default)
defer { self.synchronizationTracker.unregister() }
#endif
if isFlagSet(self.disposed, 1) {
return
}
self.observer.on(event)
}
final func forwarder() -> SinkForward<Observer> {
SinkForward(forward: self)
}
final var isDisposed: Bool {
isFlagSet(self.disposed, 1)
}
func dispose() {
fetchOr(self.disposed, 1)
self.cancel.dispose()
}
deinit {
#if TRACE_RESOURCES
_ = Resources.decrementTotal()
#endif
}
}
final class SinkForward<Observer: ObserverType>: ObserverType {
typealias Element = Observer.Element
private let forward: Sink<Observer>
init(forward: Sink<Observer>) {
self.forward = forward
}
final func on(_ event: Event<Element>) {
switch event {
case .next:
self.forward.observer.on(event)
case .error, .completed:
self.forward.observer.on(event)
self.forward.cancel.dispose()
}
}
}

View File

@@ -0,0 +1,158 @@
//
// Skip.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/25/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Bypasses a specified number of elements in an observable sequence and then returns the remaining elements.
- seealso: [skip operator on reactivex.io](http://reactivex.io/documentation/operators/skip.html)
- parameter count: The number of elements to skip before returning the remaining elements.
- returns: An observable sequence that contains the elements that occur after the specified index in the input sequence.
*/
public func skip(_ count: Int)
-> Observable<Element> {
SkipCount(source: self.asObservable(), count: count)
}
}
extension ObservableType {
/**
Skips elements for the specified duration from the start of the observable source sequence, using the specified scheduler to run timers.
- seealso: [skip operator on reactivex.io](http://reactivex.io/documentation/operators/skip.html)
- parameter duration: Duration for skipping elements from the start of the sequence.
- parameter scheduler: Scheduler to run the timer on.
- returns: An observable sequence with the elements skipped during the specified duration from the start of the source sequence.
*/
public func skip(_ duration: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
SkipTime(source: self.asObservable(), duration: duration, scheduler: scheduler)
}
}
// count version
final private class SkipCountSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = SkipCount<Element>
let parent: Parent
var remaining: Int
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.remaining = parent.count
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
if self.remaining <= 0 {
self.forwardOn(.next(value))
}
else {
self.remaining -= 1
}
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
self.forwardOn(event)
self.dispose()
}
}
}
final private class SkipCount<Element>: Producer<Element> {
let source: Observable<Element>
let count: Int
init(source: Observable<Element>, count: Int) {
self.source = source
self.count = count
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = SkipCountSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}
// time version
final private class SkipTimeSink<Element, Observer: ObserverType>: Sink<Observer>, ObserverType where Observer.Element == Element {
typealias Parent = SkipTime<Element>
let parent: Parent
// state
var open = false
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
if self.open {
self.forwardOn(.next(value))
}
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
self.forwardOn(event)
self.dispose()
}
}
func tick() {
self.open = true
}
func run() -> Disposable {
let disposeTimer = self.parent.scheduler.scheduleRelative((), dueTime: self.parent.duration) { _ in
self.tick()
return Disposables.create()
}
let disposeSubscription = self.parent.source.subscribe(self)
return Disposables.create(disposeTimer, disposeSubscription)
}
}
final private class SkipTime<Element>: Producer<Element> {
let source: Observable<Element>
let duration: RxTimeInterval
let scheduler: SchedulerType
init(source: Observable<Element>, duration: RxTimeInterval, scheduler: SchedulerType) {
self.source = source
self.scheduler = scheduler
self.duration = duration
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = SkipTimeSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,152 @@
//
// SkipUntil.swift
// RxSwift
//
// Created by Yury Korolev on 10/3/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns the elements from the source observable sequence that are emitted after the other observable sequence produces an element.
- seealso: [skipUntil operator on reactivex.io](http://reactivex.io/documentation/operators/skipuntil.html)
- parameter other: Observable sequence that starts propagation of elements of the source sequence.
- returns: An observable sequence containing the elements of the source sequence that are emitted after the other sequence emits an item.
*/
public func skip<Source: ObservableType>(until other: Source)
-> Observable<Element> {
SkipUntil(source: self.asObservable(), other: other.asObservable())
}
/**
Returns the elements from the source observable sequence that are emitted after the other observable sequence produces an element.
- seealso: [skipUntil operator on reactivex.io](http://reactivex.io/documentation/operators/skipuntil.html)
- parameter other: Observable sequence that starts propagation of elements of the source sequence.
- returns: An observable sequence containing the elements of the source sequence that are emitted after the other sequence emits an item.
*/
@available(*, deprecated, renamed: "skip(until:)")
public func skipUntil<Source: ObservableType>(_ other: Source)
-> Observable<Element> {
skip(until: other)
}
}
final private class SkipUntilSinkOther<Other, Observer: ObserverType>
: ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias Parent = SkipUntilSink<Other, Observer>
typealias Element = Other
private let parent: Parent
var lock: RecursiveLock {
self.parent.lock
}
let subscription = SingleAssignmentDisposable()
init(parent: Parent) {
self.parent = parent
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next:
self.parent.forwardElements = true
self.subscription.dispose()
case .error(let e):
self.parent.forwardOn(.error(e))
self.parent.dispose()
case .completed:
self.subscription.dispose()
}
}
#if TRACE_RESOURCES
deinit {
_ = Resources.decrementTotal()
}
#endif
}
final private class SkipUntilSink<Other, Observer: ObserverType>
: Sink<Observer>
, ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias Element = Observer.Element
typealias Parent = SkipUntil<Element, Other>
let lock = RecursiveLock()
private let parent: Parent
fileprivate var forwardElements = false
private let sourceSubscription = SingleAssignmentDisposable()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next:
if self.forwardElements {
self.forwardOn(event)
}
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
if self.forwardElements {
self.forwardOn(event)
}
self.dispose()
}
}
func run() -> Disposable {
let sourceSubscription = self.parent.source.subscribe(self)
let otherObserver = SkipUntilSinkOther(parent: self)
let otherSubscription = self.parent.other.subscribe(otherObserver)
self.sourceSubscription.setDisposable(sourceSubscription)
otherObserver.subscription.setDisposable(otherSubscription)
return Disposables.create(sourceSubscription, otherObserver.subscription)
}
}
final private class SkipUntil<Element, Other>: Producer<Element> {
fileprivate let source: Observable<Element>
fileprivate let other: Observable<Other>
init(source: Observable<Element>, other: Observable<Other>) {
self.source = source
self.other = other
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = SkipUntilSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,87 @@
//
// SkipWhile.swift
// RxSwift
//
// Created by Yury Korolev on 10/9/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Bypasses elements in an observable sequence as long as a specified condition is true and then returns the remaining elements.
- seealso: [skipWhile operator on reactivex.io](http://reactivex.io/documentation/operators/skipwhile.html)
- parameter predicate: A function to test each element for a condition.
- returns: An observable sequence that contains the elements from the input sequence starting at the first element in the linear series that does not pass the test specified by predicate.
*/
public func skip(while predicate: @escaping (Element) throws -> Bool) -> Observable<Element> {
SkipWhile(source: self.asObservable(), predicate: predicate)
}
/**
Bypasses elements in an observable sequence as long as a specified condition is true and then returns the remaining elements.
- seealso: [skipWhile operator on reactivex.io](http://reactivex.io/documentation/operators/skipwhile.html)
- parameter predicate: A function to test each element for a condition.
- returns: An observable sequence that contains the elements from the input sequence starting at the first element in the linear series that does not pass the test specified by predicate.
*/
@available(*, deprecated, renamed: "skip(while:)")
public func skipWhile(_ predicate: @escaping (Element) throws -> Bool) -> Observable<Element> {
SkipWhile(source: self.asObservable(), predicate: predicate)
}
}
final private class SkipWhileSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = SkipWhile<Element>
private let parent: Parent
private var running = false
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
if !self.running {
do {
self.running = try !self.parent.predicate(value)
} catch let e {
self.forwardOn(.error(e))
self.dispose()
return
}
}
if self.running {
self.forwardOn(.next(value))
}
case .error, .completed:
self.forwardOn(event)
self.dispose()
}
}
}
final private class SkipWhile<Element>: Producer<Element> {
typealias Predicate = (Element) throws -> Bool
private let source: Observable<Element>
fileprivate let predicate: Predicate
init(source: Observable<Element>, predicate: @escaping Predicate) {
self.source = source
self.predicate = predicate
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = SkipWhileSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,42 @@
//
// StartWith.swift
// RxSwift
//
// Created by Krunoslav Zaher on 4/6/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Prepends a sequence of values to an observable sequence.
- seealso: [startWith operator on reactivex.io](http://reactivex.io/documentation/operators/startwith.html)
- parameter elements: Elements to prepend to the specified sequence.
- returns: The source sequence prepended with the specified values.
*/
public func startWith(_ elements: Element ...)
-> Observable<Element> {
return StartWith(source: self.asObservable(), elements: elements)
}
}
final private class StartWith<Element>: Producer<Element> {
let elements: [Element]
let source: Observable<Element>
init(source: Observable<Element>, elements: [Element]) {
self.source = source
self.elements = elements
super.init()
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
for e in self.elements {
observer.on(.next(e))
}
return (sink: Disposables.create(), subscription: self.source.subscribe(observer))
}
}

View File

@@ -0,0 +1,103 @@
//
// SubscribeOn.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/14/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Wraps the source sequence in order to run its subscription and unsubscription logic on the specified
scheduler.
This operation is not commonly used.
This only performs the side-effects of subscription and unsubscription on the specified scheduler.
In order to invoke observer callbacks on a `scheduler`, use `observeOn`.
- seealso: [subscribeOn operator on reactivex.io](http://reactivex.io/documentation/operators/subscribeon.html)
- parameter scheduler: Scheduler to perform subscription and unsubscription actions on.
- returns: The source sequence whose subscriptions and unsubscriptions happen on the specified scheduler.
*/
public func subscribe(on scheduler: ImmediateSchedulerType)
-> Observable<Element> {
SubscribeOn(source: self, scheduler: scheduler)
}
/**
Wraps the source sequence in order to run its subscription and unsubscription logic on the specified
scheduler.
This operation is not commonly used.
This only performs the side-effects of subscription and unsubscription on the specified scheduler.
In order to invoke observer callbacks on a `scheduler`, use `observeOn`.
- seealso: [subscribeOn operator on reactivex.io](http://reactivex.io/documentation/operators/subscribeon.html)
- parameter scheduler: Scheduler to perform subscription and unsubscription actions on.
- returns: The source sequence whose subscriptions and unsubscriptions happen on the specified scheduler.
*/
@available(*, deprecated, renamed: "subscribe(on:)")
public func subscribeOn(_ scheduler: ImmediateSchedulerType)
-> Observable<Element> {
subscribe(on: scheduler)
}
}
final private class SubscribeOnSink<Ob: ObservableType, Observer: ObserverType>: Sink<Observer>, ObserverType where Ob.Element == Observer.Element {
typealias Element = Observer.Element
typealias Parent = SubscribeOn<Ob>
let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
self.forwardOn(event)
if event.isStopEvent {
self.dispose()
}
}
func run() -> Disposable {
let disposeEverything = SerialDisposable()
let cancelSchedule = SingleAssignmentDisposable()
disposeEverything.disposable = cancelSchedule
let disposeSchedule = self.parent.scheduler.schedule(()) { _ -> Disposable in
let subscription = self.parent.source.subscribe(self)
disposeEverything.disposable = ScheduledDisposable(scheduler: self.parent.scheduler, disposable: subscription)
return Disposables.create()
}
cancelSchedule.setDisposable(disposeSchedule)
return disposeEverything
}
}
final private class SubscribeOn<Ob: ObservableType>: Producer<Ob.Element> {
let source: Ob
let scheduler: ImmediateSchedulerType
init(source: Ob, scheduler: ImmediateSchedulerType) {
self.source = source
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Ob.Element {
let sink = SubscribeOnSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,251 @@
//
// Switch.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/12/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Projects each element of an observable sequence into a new sequence of observable sequences and then
transforms an observable sequence of observable sequences into an observable sequence producing values only from the most recent observable sequence.
It is a combination of `map` + `switchLatest` operator
- seealso: [flatMapLatest operator on reactivex.io](http://reactivex.io/documentation/operators/flatmap.html)
- parameter selector: A transform function to apply to each element.
- returns: An observable sequence whose elements are the result of invoking the transform function on each element of source producing an
Observable of Observable sequences and that at any point in time produces the elements of the most recent inner observable sequence that has been received.
*/
public func flatMapLatest<Source: ObservableConvertibleType>(_ selector: @escaping (Element) throws -> Source)
-> Observable<Source.Element> {
return FlatMapLatest(source: self.asObservable(), selector: selector)
}
/**
Projects each element of an observable sequence into a new sequence of observable sequences and then
transforms an observable sequence of observable sequences into an observable sequence producing values only from the most recent observable sequence.
It is a combination of `map` + `switchLatest` operator
- seealso: [flatMapLatest operator on reactivex.io](http://reactivex.io/documentation/operators/flatmap.html)
- parameter selector: A transform function to apply to each element.
- returns: An observable sequence whose elements are the result of invoking the transform function on each element of source producing an
Observable of Observable sequences and that at any point in time produces the elements of the most recent inner observable sequence that has been received.
*/
public func flatMapLatest<Source: InfallibleType>(_ selector: @escaping (Element) throws -> Source)
-> Infallible<Source.Element> {
return Infallible(flatMapLatest(selector))
}
}
extension ObservableType where Element: ObservableConvertibleType {
/**
Transforms an observable sequence of observable sequences into an observable sequence
producing values only from the most recent observable sequence.
Each time a new inner observable sequence is received, unsubscribe from the
previous inner observable sequence.
- seealso: [switch operator on reactivex.io](http://reactivex.io/documentation/operators/switch.html)
- returns: The observable sequence that at any point in time produces the elements of the most recent inner observable sequence that has been received.
*/
public func switchLatest() -> Observable<Element.Element> {
Switch(source: self.asObservable())
}
}
private class SwitchSink<SourceType, Source: ObservableConvertibleType, Observer: ObserverType>
: Sink<Observer>
, ObserverType where Source.Element == Observer.Element {
typealias Element = SourceType
private let subscriptions: SingleAssignmentDisposable = SingleAssignmentDisposable()
private let innerSubscription: SerialDisposable = SerialDisposable()
let lock = RecursiveLock()
// state
fileprivate var stopped = false
fileprivate var latest = 0
fileprivate var hasLatest = false
override init(observer: Observer, cancel: Cancelable) {
super.init(observer: observer, cancel: cancel)
}
func run(_ source: Observable<SourceType>) -> Disposable {
let subscription = source.subscribe(self)
self.subscriptions.setDisposable(subscription)
return Disposables.create(subscriptions, innerSubscription)
}
func performMap(_ element: SourceType) throws -> Source {
rxAbstractMethod()
}
@inline(__always)
final private func nextElementArrived(element: Element) -> (Int, Observable<Source.Element>)? {
self.lock.lock(); defer { self.lock.unlock() }
do {
let observable = try self.performMap(element).asObservable()
self.hasLatest = true
self.latest = self.latest &+ 1
return (self.latest, observable)
}
catch let error {
self.forwardOn(.error(error))
self.dispose()
}
return nil
}
func on(_ event: Event<Element>) {
switch event {
case .next(let element):
if let (latest, observable) = self.nextElementArrived(element: element) {
let d = SingleAssignmentDisposable()
self.innerSubscription.disposable = d
let observer = SwitchSinkIter(parent: self, id: latest, this: d)
let disposable = observable.subscribe(observer)
d.setDisposable(disposable)
}
case .error(let error):
self.lock.lock(); defer { self.lock.unlock() }
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.lock.lock(); defer { self.lock.unlock() }
self.stopped = true
self.subscriptions.dispose()
if !self.hasLatest {
self.forwardOn(.completed)
self.dispose()
}
}
}
}
final private class SwitchSinkIter<SourceType, Source: ObservableConvertibleType, Observer: ObserverType>
: ObserverType
, LockOwnerType
, SynchronizedOnType where Source.Element == Observer.Element {
typealias Element = Source.Element
typealias Parent = SwitchSink<SourceType, Source, Observer>
private let parent: Parent
private let id: Int
private let this: Disposable
var lock: RecursiveLock {
self.parent.lock
}
init(parent: Parent, id: Int, this: Disposable) {
self.parent = parent
self.id = id
self.this = this
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next: break
case .error, .completed:
self.this.dispose()
}
if self.parent.latest != self.id {
return
}
switch event {
case .next:
self.parent.forwardOn(event)
case .error:
self.parent.forwardOn(event)
self.parent.dispose()
case .completed:
self.parent.hasLatest = false
if self.parent.stopped {
self.parent.forwardOn(event)
self.parent.dispose()
}
}
}
}
// MARK: Specializations
final private class SwitchIdentitySink<Source: ObservableConvertibleType, Observer: ObserverType>: SwitchSink<Source, Source, Observer>
where Observer.Element == Source.Element {
override init(observer: Observer, cancel: Cancelable) {
super.init(observer: observer, cancel: cancel)
}
override func performMap(_ element: Source) throws -> Source {
element
}
}
final private class MapSwitchSink<SourceType, Source: ObservableConvertibleType, Observer: ObserverType>: SwitchSink<SourceType, Source, Observer> where Observer.Element == Source.Element {
typealias Selector = (SourceType) throws -> Source
private let selector: Selector
init(selector: @escaping Selector, observer: Observer, cancel: Cancelable) {
self.selector = selector
super.init(observer: observer, cancel: cancel)
}
override func performMap(_ element: SourceType) throws -> Source {
try self.selector(element)
}
}
// MARK: Producers
final private class Switch<Source: ObservableConvertibleType>: Producer<Source.Element> {
private let source: Observable<Source>
init(source: Observable<Source>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Source.Element {
let sink = SwitchIdentitySink<Source, Observer>(observer: observer, cancel: cancel)
let subscription = sink.run(self.source)
return (sink: sink, subscription: subscription)
}
}
final private class FlatMapLatest<SourceType, Source: ObservableConvertibleType>: Producer<Source.Element> {
typealias Selector = (SourceType) throws -> Source
private let source: Observable<SourceType>
private let selector: Selector
init(source: Observable<SourceType>, selector: @escaping Selector) {
self.source = source
self.selector = selector
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Source.Element {
let sink = MapSwitchSink<SourceType, Source, Observer>(selector: self.selector, observer: observer, cancel: cancel)
let subscription = sink.run(self.source)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,104 @@
//
// SwitchIfEmpty.swift
// RxSwift
//
// Created by sergdort on 23/12/2016.
// Copyright © 2016 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns the elements of the specified sequence or `switchTo` sequence if the sequence is empty.
- seealso: [DefaultIfEmpty operator on reactivex.io](http://reactivex.io/documentation/operators/defaultifempty.html)
- parameter other: Observable sequence being returned when source sequence is empty.
- returns: Observable sequence that contains elements from switchTo sequence if source is empty, otherwise returns source sequence elements.
*/
public func ifEmpty(switchTo other: Observable<Element>) -> Observable<Element> {
SwitchIfEmpty(source: self.asObservable(), ifEmpty: other)
}
}
final private class SwitchIfEmpty<Element>: Producer<Element> {
private let source: Observable<Element>
private let ifEmpty: Observable<Element>
init(source: Observable<Element>, ifEmpty: Observable<Element>) {
self.source = source
self.ifEmpty = ifEmpty
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = SwitchIfEmptySink(ifEmpty: self.ifEmpty,
observer: observer,
cancel: cancel)
let subscription = sink.run(self.source.asObservable())
return (sink: sink, subscription: subscription)
}
}
final private class SwitchIfEmptySink<Observer: ObserverType>: Sink<Observer>
, ObserverType {
typealias Element = Observer.Element
private let ifEmpty: Observable<Element>
private var isEmpty = true
private let ifEmptySubscription = SingleAssignmentDisposable()
init(ifEmpty: Observable<Element>, observer: Observer, cancel: Cancelable) {
self.ifEmpty = ifEmpty
super.init(observer: observer, cancel: cancel)
}
func run(_ source: Observable<Observer.Element>) -> Disposable {
let subscription = source.subscribe(self)
return Disposables.create(subscription, ifEmptySubscription)
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.isEmpty = false
self.forwardOn(event)
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
guard self.isEmpty else {
self.forwardOn(.completed)
self.dispose()
return
}
let ifEmptySink = SwitchIfEmptySinkIter(parent: self)
self.ifEmptySubscription.setDisposable(self.ifEmpty.subscribe(ifEmptySink))
}
}
}
final private class SwitchIfEmptySinkIter<Observer: ObserverType>
: ObserverType {
typealias Element = Observer.Element
typealias Parent = SwitchIfEmptySink<Observer>
private let parent: Parent
init(parent: Parent) {
self.parent = parent
}
func on(_ event: Event<Element>) {
switch event {
case .next:
self.parent.forwardOn(event)
case .error:
self.parent.forwardOn(event)
self.parent.dispose()
case .completed:
self.parent.forwardOn(event)
self.parent.dispose()
}
}
}

View File

@@ -0,0 +1,193 @@
//
// Take.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/12/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns a specified number of contiguous elements from the start of an observable sequence.
- seealso: [take operator on reactivex.io](http://reactivex.io/documentation/operators/take.html)
- parameter count: The number of elements to return.
- returns: An observable sequence that contains the specified number of elements from the start of the input sequence.
*/
public func take(_ count: Int)
-> Observable<Element> {
if count == 0 {
return Observable.empty()
}
else {
return TakeCount(source: self.asObservable(), count: count)
}
}
}
extension ObservableType {
/**
Takes elements for the specified duration from the start of the observable source sequence, using the specified scheduler to run timers.
- seealso: [take operator on reactivex.io](http://reactivex.io/documentation/operators/take.html)
- parameter duration: Duration for taking elements from the start of the sequence.
- parameter scheduler: Scheduler to run the timer on.
- returns: An observable sequence with the elements taken during the specified duration from the start of the source sequence.
*/
public func take(for duration: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
TakeTime(source: self.asObservable(), duration: duration, scheduler: scheduler)
}
/**
Takes elements for the specified duration from the start of the observable source sequence, using the specified scheduler to run timers.
- seealso: [take operator on reactivex.io](http://reactivex.io/documentation/operators/take.html)
- parameter duration: Duration for taking elements from the start of the sequence.
- parameter scheduler: Scheduler to run the timer on.
- returns: An observable sequence with the elements taken during the specified duration from the start of the source sequence.
*/
@available(*, deprecated, renamed: "take(for:scheduler:)")
public func take(_ duration: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
take(for: duration, scheduler: scheduler)
}
}
// count version
final private class TakeCountSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = TakeCount<Element>
private let parent: Parent
private var remaining: Int
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.remaining = parent.count
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
if self.remaining > 0 {
self.remaining -= 1
self.forwardOn(.next(value))
if self.remaining == 0 {
self.forwardOn(.completed)
self.dispose()
}
}
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
self.forwardOn(event)
self.dispose()
}
}
}
final private class TakeCount<Element>: Producer<Element> {
private let source: Observable<Element>
fileprivate let count: Int
init(source: Observable<Element>, count: Int) {
if count < 0 {
rxFatalError("count can't be negative")
}
self.source = source
self.count = count
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = TakeCountSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}
// time version
final private class TakeTimeSink<Element, Observer: ObserverType>
: Sink<Observer>
, LockOwnerType
, ObserverType
, SynchronizedOnType where Observer.Element == Element {
typealias Parent = TakeTime<Element>
private let parent: Parent
let lock = RecursiveLock()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next(let value):
self.forwardOn(.next(value))
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
self.forwardOn(event)
self.dispose()
}
}
func tick() {
self.lock.performLocked {
self.forwardOn(.completed)
self.dispose()
}
}
func run() -> Disposable {
let disposeTimer = self.parent.scheduler.scheduleRelative((), dueTime: self.parent.duration) { _ in
self.tick()
return Disposables.create()
}
let disposeSubscription = self.parent.source.subscribe(self)
return Disposables.create(disposeTimer, disposeSubscription)
}
}
final private class TakeTime<Element>: Producer<Element> {
typealias TimeInterval = RxTimeInterval
fileprivate let source: Observable<Element>
fileprivate let duration: TimeInterval
fileprivate let scheduler: SchedulerType
init(source: Observable<Element>, duration: TimeInterval, scheduler: SchedulerType) {
self.source = source
self.scheduler = scheduler
self.duration = duration
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = TakeTimeSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,78 @@
//
// TakeLast.swift
// RxSwift
//
// Created by Tomi Koskinen on 25/10/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns a specified number of contiguous elements from the end of an observable sequence.
This operator accumulates a buffer with a length enough to store elements count elements. Upon completion of the source sequence, this buffer is drained on the result sequence. This causes the elements to be delayed.
- seealso: [takeLast operator on reactivex.io](http://reactivex.io/documentation/operators/takelast.html)
- parameter count: Number of elements to take from the end of the source sequence.
- returns: An observable sequence containing the specified number of elements from the end of the source sequence.
*/
public func takeLast(_ count: Int)
-> Observable<Element> {
TakeLast(source: self.asObservable(), count: count)
}
}
final private class TakeLastSink<Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = TakeLast<Element>
private let parent: Parent
private var elements: Queue<Element>
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
self.elements = Queue<Element>(capacity: parent.count + 1)
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
self.elements.enqueue(value)
if self.elements.count > self.parent.count {
_ = self.elements.dequeue()
}
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
for e in self.elements {
self.forwardOn(.next(e))
}
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class TakeLast<Element>: Producer<Element> {
private let source: Observable<Element>
fileprivate let count: Int
init(source: Observable<Element>, count: Int) {
if count < 0 {
rxFatalError("count can't be negative")
}
self.source = source
self.count = count
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = TakeLastSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,285 @@
//
// TakeWithPredicate.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/7/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Returns the elements from the source observable sequence until the other observable sequence produces an element.
- seealso: [takeUntil operator on reactivex.io](http://reactivex.io/documentation/operators/takeuntil.html)
- parameter other: Observable sequence that terminates propagation of elements of the source sequence.
- returns: An observable sequence containing the elements of the source sequence up to the point the other sequence interrupted further propagation.
*/
public func take<Source: ObservableType>(until other: Source)
-> Observable<Element> {
TakeUntil(source: self.asObservable(), other: other.asObservable())
}
/**
Returns elements from an observable sequence until the specified condition is true.
- seealso: [takeUntil operator on reactivex.io](http://reactivex.io/documentation/operators/takeuntil.html)
- parameter predicate: A function to test each element for a condition.
- parameter behavior: Whether or not to include the last element matching the predicate. Defaults to `exclusive`.
- returns: An observable sequence that contains the elements from the input sequence that occur before the element at which the test passes.
*/
public func take(until predicate: @escaping (Element) throws -> Bool,
behavior: TakeBehavior = .exclusive)
-> Observable<Element> {
TakeUntilPredicate(source: self.asObservable(),
behavior: behavior,
predicate: predicate)
}
/**
Returns elements from an observable sequence as long as a specified condition is true.
- seealso: [takeWhile operator on reactivex.io](http://reactivex.io/documentation/operators/takewhile.html)
- parameter predicate: A function to test each element for a condition.
- returns: An observable sequence that contains the elements from the input sequence that occur before the element at which the test no longer passes.
*/
public func take(while predicate: @escaping (Element) throws -> Bool,
behavior: TakeBehavior = .exclusive)
-> Observable<Element> {
take(until: { try !predicate($0) }, behavior: behavior)
}
/**
Returns the elements from the source observable sequence until the other observable sequence produces an element.
- seealso: [takeUntil operator on reactivex.io](http://reactivex.io/documentation/operators/takeuntil.html)
- parameter other: Observable sequence that terminates propagation of elements of the source sequence.
- returns: An observable sequence containing the elements of the source sequence up to the point the other sequence interrupted further propagation.
*/
@available(*, deprecated, renamed: "take(until:)")
public func takeUntil<Source: ObservableType>(_ other: Source)
-> Observable<Element> {
take(until: other)
}
/**
Returns elements from an observable sequence until the specified condition is true.
- seealso: [takeUntil operator on reactivex.io](http://reactivex.io/documentation/operators/takeuntil.html)
- parameter behavior: Whether or not to include the last element matching the predicate.
- parameter predicate: A function to test each element for a condition.
- returns: An observable sequence that contains the elements from the input sequence that occur before the element at which the test passes.
*/
@available(*, deprecated, renamed: "take(until:behavior:)")
public func takeUntil(_ behavior: TakeBehavior,
predicate: @escaping (Element) throws -> Bool)
-> Observable<Element> {
take(until: predicate, behavior: behavior)
}
/**
Returns elements from an observable sequence as long as a specified condition is true.
- seealso: [takeWhile operator on reactivex.io](http://reactivex.io/documentation/operators/takewhile.html)
- parameter predicate: A function to test each element for a condition.
- returns: An observable sequence that contains the elements from the input sequence that occur before the element at which the test no longer passes.
*/
@available(*, deprecated, renamed: "take(while:)")
public func takeWhile(_ predicate: @escaping (Element) throws -> Bool)
-> Observable<Element> {
take(until: { try !predicate($0) }, behavior: .exclusive)
}
}
/// Behaviors for the take operator family.
public enum TakeBehavior {
/// Include the last element matching the predicate.
case inclusive
/// Exclude the last element matching the predicate.
case exclusive
}
// MARK: - TakeUntil Observable
final private class TakeUntilSinkOther<Other, Observer: ObserverType>
: ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias Parent = TakeUntilSink<Other, Observer>
typealias Element = Other
private let parent: Parent
var lock: RecursiveLock {
self.parent.lock
}
fileprivate let subscription = SingleAssignmentDisposable()
init(parent: Parent) {
self.parent = parent
#if TRACE_RESOURCES
_ = Resources.incrementTotal()
#endif
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next:
self.parent.forwardOn(.completed)
self.parent.dispose()
case .error(let e):
self.parent.forwardOn(.error(e))
self.parent.dispose()
case .completed:
self.subscription.dispose()
}
}
#if TRACE_RESOURCES
deinit {
_ = Resources.decrementTotal()
}
#endif
}
final private class TakeUntilSink<Other, Observer: ObserverType>
: Sink<Observer>
, LockOwnerType
, ObserverType
, SynchronizedOnType {
typealias Element = Observer.Element
typealias Parent = TakeUntil<Element, Other>
private let parent: Parent
let lock = RecursiveLock()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next:
self.forwardOn(event)
case .error:
self.forwardOn(event)
self.dispose()
case .completed:
self.forwardOn(event)
self.dispose()
}
}
func run() -> Disposable {
let otherObserver = TakeUntilSinkOther(parent: self)
let otherSubscription = self.parent.other.subscribe(otherObserver)
otherObserver.subscription.setDisposable(otherSubscription)
let sourceSubscription = self.parent.source.subscribe(self)
return Disposables.create(sourceSubscription, otherObserver.subscription)
}
}
final private class TakeUntil<Element, Other>: Producer<Element> {
fileprivate let source: Observable<Element>
fileprivate let other: Observable<Other>
init(source: Observable<Element>, other: Observable<Other>) {
self.source = source
self.other = other
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = TakeUntilSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
// MARK: - TakeUntil Predicate
final private class TakeUntilPredicateSink<Observer: ObserverType>
: Sink<Observer>, ObserverType {
typealias Element = Observer.Element
typealias Parent = TakeUntilPredicate<Element>
private let parent: Parent
private var running = true
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<Element>) {
switch event {
case .next(let value):
if !self.running {
return
}
do {
self.running = try !self.parent.predicate(value)
} catch let e {
self.forwardOn(.error(e))
self.dispose()
return
}
if self.running {
self.forwardOn(.next(value))
} else {
if self.parent.behavior == .inclusive {
self.forwardOn(.next(value))
}
self.forwardOn(.completed)
self.dispose()
}
case .error, .completed:
self.forwardOn(event)
self.dispose()
}
}
}
final private class TakeUntilPredicate<Element>: Producer<Element> {
typealias Predicate = (Element) throws -> Bool
private let source: Observable<Element>
fileprivate let predicate: Predicate
fileprivate let behavior: TakeBehavior
init(source: Observable<Element>,
behavior: TakeBehavior,
predicate: @escaping Predicate) {
self.source = source
self.behavior = behavior
self.predicate = predicate
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = TakeUntilPredicateSink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,160 @@
//
// Throttle.swift
// RxSwift
//
// Created by Krunoslav Zaher on 3/22/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
import Foundation
extension ObservableType {
/**
Returns an Observable that emits the first and the latest item emitted by the source Observable during sequential time windows of a specified duration.
This operator makes sure that no two elements are emitted in less then dueTime.
- seealso: [debounce operator on reactivex.io](http://reactivex.io/documentation/operators/debounce.html)
- parameter dueTime: Throttling duration for each element.
- parameter latest: Should latest element received in a dueTime wide time window since last element emission be emitted.
- parameter scheduler: Scheduler to run the throttle timers on.
- returns: The throttled sequence.
*/
public func throttle(_ dueTime: RxTimeInterval, latest: Bool = true, scheduler: SchedulerType)
-> Observable<Element> {
Throttle(source: self.asObservable(), dueTime: dueTime, latest: latest, scheduler: scheduler)
}
}
final private class ThrottleSink<Observer: ObserverType>
: Sink<Observer>
, ObserverType
, LockOwnerType
, SynchronizedOnType {
typealias Element = Observer.Element
typealias ParentType = Throttle<Element>
private let parent: ParentType
let lock = RecursiveLock()
// state
private var lastUnsentElement: Element?
private var lastSentTime: Date?
private var completed: Bool = false
let cancellable = SerialDisposable()
init(parent: ParentType, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
let subscription = self.parent.source.subscribe(self)
return Disposables.create(subscription, cancellable)
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
switch event {
case .next(let element):
let now = self.parent.scheduler.now
let reducedScheduledTime: RxTimeInterval
if let lastSendingTime = self.lastSentTime {
reducedScheduledTime = self.parent.dueTime.reduceWithSpanBetween(earlierDate: lastSendingTime, laterDate: now)
}
else {
reducedScheduledTime = .nanoseconds(0)
}
if reducedScheduledTime.isNow {
self.sendNow(element: element)
return
}
if !self.parent.latest {
return
}
let isThereAlreadyInFlightRequest = self.lastUnsentElement != nil
self.lastUnsentElement = element
if isThereAlreadyInFlightRequest {
return
}
let scheduler = self.parent.scheduler
let d = SingleAssignmentDisposable()
self.cancellable.disposable = d
d.setDisposable(scheduler.scheduleRelative(0, dueTime: reducedScheduledTime, action: self.propagate))
case .error:
self.lastUnsentElement = nil
self.forwardOn(event)
self.dispose()
case .completed:
if self.lastUnsentElement != nil {
self.completed = true
}
else {
self.forwardOn(.completed)
self.dispose()
}
}
}
private func sendNow(element: Element) {
self.lastUnsentElement = nil
self.forwardOn(.next(element))
// in case element processing takes a while, this should give some more room
self.lastSentTime = self.parent.scheduler.now
}
func propagate(_: Int) -> Disposable {
self.lock.performLocked {
if let lastUnsentElement = self.lastUnsentElement {
self.sendNow(element: lastUnsentElement)
}
if self.completed {
self.forwardOn(.completed)
self.dispose()
}
}
return Disposables.create()
}
}
final private class Throttle<Element>: Producer<Element> {
fileprivate let source: Observable<Element>
fileprivate let dueTime: RxTimeInterval
fileprivate let latest: Bool
fileprivate let scheduler: SchedulerType
init(source: Observable<Element>, dueTime: RxTimeInterval, latest: Bool, scheduler: SchedulerType) {
self.source = source
self.dueTime = dueTime
self.latest = latest
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = ThrottleSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,151 @@
//
// Timeout.swift
// RxSwift
//
// Created by Tomi Koskinen on 13/11/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Applies a timeout policy for each element in the observable sequence. If the next element isn't received within the specified timeout duration starting from its predecessor, a TimeoutError is propagated to the observer.
- seealso: [timeout operator on reactivex.io](http://reactivex.io/documentation/operators/timeout.html)
- parameter dueTime: Maximum duration between values before a timeout occurs.
- parameter scheduler: Scheduler to run the timeout timer on.
- returns: An observable sequence with a `RxError.timeout` in case of a timeout.
*/
public func timeout(_ dueTime: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
return Timeout(source: self.asObservable(), dueTime: dueTime, other: Observable.error(RxError.timeout), scheduler: scheduler)
}
/**
Applies a timeout policy for each element in the observable sequence, using the specified scheduler to run timeout timers. If the next element isn't received within the specified timeout duration starting from its predecessor, the other observable sequence is used to produce future messages from that point on.
- seealso: [timeout operator on reactivex.io](http://reactivex.io/documentation/operators/timeout.html)
- parameter dueTime: Maximum duration between values before a timeout occurs.
- parameter other: Sequence to return in case of a timeout.
- parameter scheduler: Scheduler to run the timeout timer on.
- returns: The source sequence switching to the other sequence in case of a timeout.
*/
public func timeout<Source: ObservableConvertibleType>(_ dueTime: RxTimeInterval, other: Source, scheduler: SchedulerType)
-> Observable<Element> where Element == Source.Element {
return Timeout(source: self.asObservable(), dueTime: dueTime, other: other.asObservable(), scheduler: scheduler)
}
}
final private class TimeoutSink<Observer: ObserverType>: Sink<Observer>, LockOwnerType, ObserverType {
typealias Element = Observer.Element
typealias Parent = Timeout<Element>
private let parent: Parent
let lock = RecursiveLock()
private let timerD = SerialDisposable()
private let subscription = SerialDisposable()
private var id = 0
private var switched = false
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
let original = SingleAssignmentDisposable()
self.subscription.disposable = original
self.createTimeoutTimer()
original.setDisposable(self.parent.source.subscribe(self))
return Disposables.create(subscription, timerD)
}
func on(_ event: Event<Element>) {
switch event {
case .next:
var onNextWins = false
self.lock.performLocked {
onNextWins = !self.switched
if onNextWins {
self.id = self.id &+ 1
}
}
if onNextWins {
self.forwardOn(event)
self.createTimeoutTimer()
}
case .error, .completed:
var onEventWins = false
self.lock.performLocked {
onEventWins = !self.switched
if onEventWins {
self.id = self.id &+ 1
}
}
if onEventWins {
self.forwardOn(event)
self.dispose()
}
}
}
private func createTimeoutTimer() {
if self.timerD.isDisposed {
return
}
let nextTimer = SingleAssignmentDisposable()
self.timerD.disposable = nextTimer
let disposeSchedule = self.parent.scheduler.scheduleRelative(self.id, dueTime: self.parent.dueTime) { state in
var timerWins = false
self.lock.performLocked {
self.switched = (state == self.id)
timerWins = self.switched
}
if timerWins {
self.subscription.disposable = self.parent.other.subscribe(self.forwarder())
}
return Disposables.create()
}
nextTimer.setDisposable(disposeSchedule)
}
}
final private class Timeout<Element>: Producer<Element> {
fileprivate let source: Observable<Element>
fileprivate let dueTime: RxTimeInterval
fileprivate let other: Observable<Element>
fileprivate let scheduler: SchedulerType
init(source: Observable<Element>, dueTime: RxTimeInterval, other: Observable<Element>, scheduler: SchedulerType) {
self.source = source
self.dueTime = dueTime
self.other = other
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = TimeoutSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,117 @@
//
// Timer.swift
// RxSwift
//
// Created by Krunoslav Zaher on 6/7/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType where Element: RxAbstractInteger {
/**
Returns an observable sequence that produces a value after each period, using the specified scheduler to run timers and to send out observer messages.
- seealso: [interval operator on reactivex.io](http://reactivex.io/documentation/operators/interval.html)
- parameter period: Period for producing the values in the resulting sequence.
- parameter scheduler: Scheduler to run the timer on.
- returns: An observable sequence that produces a value after each period.
*/
public static func interval(_ period: RxTimeInterval, scheduler: SchedulerType)
-> Observable<Element> {
return Timer(
dueTime: period,
period: period,
scheduler: scheduler
)
}
}
extension ObservableType where Element: RxAbstractInteger {
/**
Returns an observable sequence that periodically produces a value after the specified initial relative due time has elapsed, using the specified scheduler to run timers.
- seealso: [timer operator on reactivex.io](http://reactivex.io/documentation/operators/timer.html)
- parameter dueTime: Relative time at which to produce the first value.
- parameter period: Period to produce subsequent values.
- parameter scheduler: Scheduler to run timers on.
- returns: An observable sequence that produces a value after due time has elapsed and then each period.
*/
public static func timer(_ dueTime: RxTimeInterval, period: RxTimeInterval? = nil, scheduler: SchedulerType)
-> Observable<Element> {
return Timer(
dueTime: dueTime,
period: period,
scheduler: scheduler
)
}
}
import Foundation
final private class TimerSink<Observer: ObserverType> : Sink<Observer> where Observer.Element : RxAbstractInteger {
typealias Parent = Timer<Observer.Element>
private let parent: Parent
private let lock = RecursiveLock()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
return self.parent.scheduler.schedulePeriodic(0 as Observer.Element, startAfter: self.parent.dueTime, period: self.parent.period!) { state in
self.lock.performLocked {
self.forwardOn(.next(state))
return state &+ 1
}
}
}
}
final private class TimerOneOffSink<Observer: ObserverType>: Sink<Observer> where Observer.Element: RxAbstractInteger {
typealias Parent = Timer<Observer.Element>
private let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
return self.parent.scheduler.scheduleRelative(self, dueTime: self.parent.dueTime) { [unowned self] _ -> Disposable in
self.forwardOn(.next(0))
self.forwardOn(.completed)
self.dispose()
return Disposables.create()
}
}
}
final private class Timer<Element: RxAbstractInteger>: Producer<Element> {
fileprivate let scheduler: SchedulerType
fileprivate let dueTime: RxTimeInterval
fileprivate let period: RxTimeInterval?
init(dueTime: RxTimeInterval, period: RxTimeInterval?, scheduler: SchedulerType) {
self.scheduler = scheduler
self.dueTime = dueTime
self.period = period
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
if self.period != nil {
let sink = TimerSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
else {
let sink = TimerOneOffSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}
}

View File

@@ -0,0 +1,64 @@
//
// ToArray.swift
// RxSwift
//
// Created by Junior B. on 20/10/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Converts an Observable into a Single that emits the whole sequence as a single array and then terminates.
For aggregation behavior see `reduce`.
- seealso: [toArray operator on reactivex.io](http://reactivex.io/documentation/operators/to.html)
- returns: A Single sequence containing all the emitted elements as array.
*/
public func toArray()
-> Single<[Element]> {
PrimitiveSequence(raw: ToArray(source: self.asObservable()))
}
}
final private class ToArraySink<SourceType, Observer: ObserverType>: Sink<Observer>, ObserverType where Observer.Element == [SourceType] {
typealias Parent = ToArray<SourceType>
let parent: Parent
var list = [SourceType]()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func on(_ event: Event<SourceType>) {
switch event {
case .next(let value):
self.list.append(value)
case .error(let e):
self.forwardOn(.error(e))
self.dispose()
case .completed:
self.forwardOn(.next(self.list))
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class ToArray<SourceType>: Producer<[SourceType]> {
let source: Observable<SourceType>
init(source: Observable<SourceType>) {
self.source = source
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == [SourceType] {
let sink = ToArraySink(parent: self, observer: observer, cancel: cancel)
let subscription = self.source.subscribe(sink)
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,90 @@
//
// Using.swift
// RxSwift
//
// Created by Yury Korolev on 10/15/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Constructs an observable sequence that depends on a resource object, whose lifetime is tied to the resulting observable sequence's lifetime.
- seealso: [using operator on reactivex.io](http://reactivex.io/documentation/operators/using.html)
- parameter resourceFactory: Factory function to obtain a resource object.
- parameter observableFactory: Factory function to obtain an observable sequence that depends on the obtained resource.
- returns: An observable sequence whose lifetime controls the lifetime of the dependent resource object.
*/
public static func using<Resource: Disposable>(_ resourceFactory: @escaping () throws -> Resource, observableFactory: @escaping (Resource) throws -> Observable<Element>) -> Observable<Element> {
Using(resourceFactory: resourceFactory, observableFactory: observableFactory)
}
}
final private class UsingSink<ResourceType: Disposable, Observer: ObserverType>: Sink<Observer>, ObserverType {
typealias SourceType = Observer.Element
typealias Parent = Using<SourceType, ResourceType>
private let parent: Parent
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
var disposable = Disposables.create()
do {
let resource = try self.parent.resourceFactory()
disposable = resource
let source = try self.parent.observableFactory(resource)
return Disposables.create(
source.subscribe(self),
disposable
)
} catch let error {
return Disposables.create(
Observable.error(error).subscribe(self),
disposable
)
}
}
func on(_ event: Event<SourceType>) {
switch event {
case let .next(value):
self.forwardOn(.next(value))
case let .error(error):
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.forwardOn(.completed)
self.dispose()
}
}
}
final private class Using<SourceType, ResourceType: Disposable>: Producer<SourceType> {
typealias Element = SourceType
typealias ResourceFactory = () throws -> ResourceType
typealias ObservableFactory = (ResourceType) throws -> Observable<SourceType>
fileprivate let resourceFactory: ResourceFactory
fileprivate let observableFactory: ObservableFactory
init(resourceFactory: @escaping ResourceFactory, observableFactory: @escaping ObservableFactory) {
self.resourceFactory = resourceFactory
self.observableFactory = observableFactory
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Element {
let sink = UsingSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

View File

@@ -0,0 +1,168 @@
//
// Window.swift
// RxSwift
//
// Created by Junior B. on 29/10/15.
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
//
extension ObservableType {
/**
Projects each element of an observable sequence into a window that is completed when either its full or a given amount of time has elapsed.
- seealso: [window operator on reactivex.io](http://reactivex.io/documentation/operators/window.html)
- parameter timeSpan: Maximum time length of a window.
- parameter count: Maximum element count of a window.
- parameter scheduler: Scheduler to run windowing timers on.
- returns: An observable sequence of windows (instances of `Observable`).
*/
public func window(timeSpan: RxTimeInterval, count: Int, scheduler: SchedulerType)
-> Observable<Observable<Element>> {
return WindowTimeCount(source: self.asObservable(), timeSpan: timeSpan, count: count, scheduler: scheduler)
}
}
final private class WindowTimeCountSink<Element, Observer: ObserverType>
: Sink<Observer>
, ObserverType
, LockOwnerType
, SynchronizedOnType where Observer.Element == Observable<Element> {
typealias Parent = WindowTimeCount<Element>
private let parent: Parent
let lock = RecursiveLock()
private var subject = PublishSubject<Element>()
private var count = 0
private var windowId = 0
private let timerD = SerialDisposable()
private let refCountDisposable: RefCountDisposable
private let groupDisposable = CompositeDisposable()
init(parent: Parent, observer: Observer, cancel: Cancelable) {
self.parent = parent
_ = self.groupDisposable.insert(self.timerD)
self.refCountDisposable = RefCountDisposable(disposable: self.groupDisposable)
super.init(observer: observer, cancel: cancel)
}
func run() -> Disposable {
self.forwardOn(.next(AddRef(source: self.subject, refCount: self.refCountDisposable).asObservable()))
self.createTimer(self.windowId)
_ = self.groupDisposable.insert(self.parent.source.subscribe(self))
return self.refCountDisposable
}
func startNewWindowAndCompleteCurrentOne() {
self.subject.on(.completed)
self.subject = PublishSubject<Element>()
self.forwardOn(.next(AddRef(source: self.subject, refCount: self.refCountDisposable).asObservable()))
}
func on(_ event: Event<Element>) {
self.synchronizedOn(event)
}
func synchronized_on(_ event: Event<Element>) {
var newWindow = false
var newId = 0
switch event {
case .next(let element):
self.subject.on(.next(element))
do {
_ = try incrementChecked(&self.count)
} catch let e {
self.subject.on(.error(e as Swift.Error))
self.dispose()
}
if self.count == self.parent.count {
newWindow = true
self.count = 0
self.windowId += 1
newId = self.windowId
self.startNewWindowAndCompleteCurrentOne()
}
case .error(let error):
self.subject.on(.error(error))
self.forwardOn(.error(error))
self.dispose()
case .completed:
self.subject.on(.completed)
self.forwardOn(.completed)
self.dispose()
}
if newWindow {
self.createTimer(newId)
}
}
func createTimer(_ windowId: Int) {
if self.timerD.isDisposed {
return
}
if self.windowId != windowId {
return
}
let nextTimer = SingleAssignmentDisposable()
self.timerD.disposable = nextTimer
let scheduledRelative = self.parent.scheduler.scheduleRelative(windowId, dueTime: self.parent.timeSpan) { previousWindowId in
var newId = 0
self.lock.performLocked {
if previousWindowId != self.windowId {
return
}
self.count = 0
self.windowId = self.windowId &+ 1
newId = self.windowId
self.startNewWindowAndCompleteCurrentOne()
}
self.createTimer(newId)
return Disposables.create()
}
nextTimer.setDisposable(scheduledRelative)
}
}
final private class WindowTimeCount<Element>: Producer<Observable<Element>> {
fileprivate let timeSpan: RxTimeInterval
fileprivate let count: Int
fileprivate let scheduler: SchedulerType
fileprivate let source: Observable<Element>
init(source: Observable<Element>, timeSpan: RxTimeInterval, count: Int, scheduler: SchedulerType) {
self.source = source
self.timeSpan = timeSpan
self.count = count
self.scheduler = scheduler
}
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == Observable<Element> {
let sink = WindowTimeCountSink(parent: self, observer: observer, cancel: cancel)
let subscription = sink.run()
return (sink: sink, subscription: subscription)
}
}

Some files were not shown because too many files have changed in this diff Show More