為了提高代碼的聚合, 用swift 寫的一個事件訂閱分發(fā)器.
鏈接: https://github.com/Xianlau/SwiftObservation
- 支持cocoapods
pod 'SwiftObservation'
用法案例:
import UIKit
import SwiftObservation
class ViewController: UIViewController {
let disposebag: DisposeBag = DisposeBag()
override func viewDidLoad() {
super.viewDidLoad()
//名稱事件有更新
ObserverManager.nameEvent.subscribe { (name) in
print(name)
}.dispose(by: disposebag)
//年齡事件有更新
ObserverManager.ageEvent.subscribe { (age) in
print(age)
}.dispose(by: disposebag)
//更改名稱
ObserverManager.publishNameEvent(value: "liuweixiang")
//更改年齡
ObserverManager.publishAgeEvent(value: 18)
}
}
被觀察的對象: ObserverManager
import Foundation
import SwiftObservation
class ObserverManager {
static let shared = ObserverManager.init()
///名稱subject 用來發(fā)布事件
private let nameSubject = BehaviorSubject<String>(value: "zhangsan") {
print("我被銷毀啦")
}
///年齡subject 用來發(fā)布事件
private let ageSubject = BehaviorSubject<UInt>(value: 15)
///名稱獲取
public static var name : String {
return shared.nameSubject.value
}
///年齡獲取
public static var age : UInt {
return shared.ageSubject.value
}
///名稱可監(jiān)聽實例 用來給外界監(jiān)聽
public static var nameEvent: Observable<String> {
return shared.nameSubject.asObservable()
}
///年齡可監(jiān)聽實例 用來給外界監(jiān)聽
public static var ageEvent: Observable<UInt> {
return shared.ageSubject.asObservable()
}
///發(fā)布名稱改變的事件
public static func publishNameEvent(value: String) {
if value != shared.nameSubject.value {
shared.nameSubject.publish(value)
}
}
///發(fā)布年齡改變的事件
public static func publishAgeEvent(value: UInt) {
if value != shared.ageSubject.value {
shared.ageSubject.publish(value)
}
}
}
整個觀察器的邏輯
- 負責分布事件的Subject
import Foundation
/// 只發(fā)布內容，但不緩存內容
public class PublishSubject<Value>: Observable<Value> {
///轉成可觀察對象
public func asObservable() -> Observable<Value> {
return self
}
///發(fā)布新事件
public func publish(_ value: Value) {
notifyObserverArr(value)
}
}
/// 發(fā)布內容，同時也緩存最新內容
public class BehaviorSubject<Value>: PublishSubject<Value> {
private var _value: Value
public var value: Value {
return _value
}
///初始化, onDispose: 銷毀的時候執(zhí)行的閉包
public init(value: Value, onDispose: @escaping () -> Void = {}) {
_value = value
super.init(onDispose)
}
///發(fā)布新事件
public override func publish(_ value: Value) {
_value = value
super.publish(_value)
}
}
- 可觀察對象實例
/*
*一個可以觀察的對象
*
*/
public class Observable<Value> {
///觀察者, 就是一個閉包, 事件發(fā)布的時候觸發(fā)
public typealias Observer = (Value) -> Void
//序列的迭代器
private var uniqueId = (0...).makeIterator()
//遞歸鎖
private let lock = NSRecursiveLock()
//銷毀的時候執(zhí)行的閉包
private var onDispose: () -> Void
//觀察者數(shù)組
fileprivate var observerArr: [Int : (Observer, DispatchQueue)] = [:]
// onDispose: 銷毀的時候執(zhí)行的閉包
init(_ onDispose: @escaping () -> Void = {}) {
self.onDispose = onDispose
}
/*
* 事件的訂閱函數(shù), 返回一個可以銷毀的對象
* - observer: 觀察者, 就是一個閉包, 事件發(fā)布的時候觸發(fā)
* - queue: 在指定的線程里面執(zhí)行閉包
*/
public func subscribe(observer: @escaping Observer, on queue: DispatchQueue = .main) -> Disposable {
lock.lock()
defer {
lock.unlock()
}
let id = uniqueId.next()!
observerArr[id] = (observer, queue)
//訂閱執(zhí)行完后, 返回一個銷毀的對象, 用來銷毀自身的觀察者, 以及執(zhí)行銷毀后的閉包
let disposable = Disposable.init { [weak self] in
self?.observerArr[id] = nil
self?.onDispose()
}
return disposable
}
}
extension Observable {
//有新的事件, 通知所有觀察者
func notifyObserverArr(_ value: Value) {
observerArr.forEach {
let observer = $0.value.0
let queue = $0.value.1
if isCurrentQueue(queue) {
observer(value)
} else {
queue.async { observer(value) }
}
}
}
//判斷是否當前線程
private func isCurrentQueue(_ queue: DispatchQueue) -> Bool {
let key = DispatchSpecificKey<UInt32>()
queue.setSpecific(key: key, value: arc4random())
defer { queue.setSpecific(key: key, value: nil) }
return DispatchQueue.getSpecific(key: key) != nil
}
}
- 提供銷毀資源的實例
import Foundation
/*
*一個用于提供銷毀資源的實例.
*把自身添加到DisposeBag里, DisposeBag在銷毀的時候, 會順便把此對象銷毀.
*/
public final class Disposable {
///銷毀后執(zhí)行的閉包
let dispose: () -> Void
public init(_ dispose: @escaping () -> Void) {
self.dispose = dispose
}
deinit {
dispose()
}
///把自身添加到DisposeBag里, DisposeBag在銷毀的時候, 會順便把此對象銷毀
public func dispose(by disposeBag: DisposeBag) {
disposeBag.add(self)
}
}
/*
*管理'disposableArr'的銷毀, 自身銷毀后, 所有的'disposableArr'也會銷毀
*
*/
public final class DisposeBag {
private let lock: NSRecursiveLock = NSRecursiveLock()
private var disposableArr: [Disposable] = [Disposable]()
private var isDisposed = false
public init() {
}
deinit {
dispose()
}
/// 添加一個需要銷毀的資源
public func add(_ disposable: Disposable) {
lock.lock()
defer { lock.unlock() }
if isDisposed { return }
disposableArr.append(disposable)
}
//移除所有Disposable
private func removeDisposableArr() -> [Disposable] {
lock.lock()
defer { lock.unlock() }
let disposableARR = self.disposableArr
self.disposableArr.removeAll(keepingCapacity: false)
self.isDisposed = true
return disposableARR
}
//銷毀所有的Disable對象
private func dispose() {
let disposableARR = removeDisposableArr()
//執(zhí)行每個Disable銷毀前的閉包
for disposable in disposableARR {
disposable.dispose()
}
}
}
