swift 写的一个事件订阅分发器

为了提高代码的聚合, 用swift 写的一个事件订阅分发器.

链接: 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 用来发布事件
    private let nameSubject = BehaviorSubject<String>(value: "zhangsan") {
        print("我被销毁啦")
    }
    ///年龄subject 用来发布事件
    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
    }
    
    ///名称可监听实例 用来给外界监听
    public static var nameEvent: Observable<String> {
        return shared.nameSubject.asObservable()
    }
    
    ///年龄可监听实例 用来给外界监听
    public static var ageEvent: Observable<UInt> {
        return shared.ageSubject.asObservable()
    }

    
    ///发布名称改变的事件
    public static func publishNameEvent(value: String) {
        if value != shared.nameSubject.value {
            shared.nameSubject.publish(value)
        }
    }
    
    ///发布年龄改变的事件
    public static func publishAgeEvent(value: UInt) {
        if value != shared.ageSubject.value {
            shared.ageSubject.publish(value)
        }
    }
}

整个观察器的逻辑

    1. 负责分布事件的Subject
import Foundation

/// 只发布内容,但不缓存内容
public class PublishSubject<Value>: Observable<Value> {
    
    ///转成可观察对象
    public func asObservable() -> Observable<Value> {
        return self
    }
    
    ///发布新事件
    public func publish(_ value: Value) {
        notifyObserverArr(value)
    }
}

/// 发布内容,同时也缓存最新内容
public class BehaviorSubject<Value>: PublishSubject<Value> {
    
    private var _value: Value
    
    public var value: Value {
        return _value
    }
    
    
    ///初始化, onDispose: 销毁的时候执行的闭包
    public init(value: Value, onDispose: @escaping () -> Void = {}) {
        _value = value
        super.init(onDispose)
    }
    
    ///发布新事件
    public override func publish(_ value: Value) {
        _value = value
        super.publish(_value)
    }
}

    1. 可观察对象实例

/*
 *一个可以观察的对象
 *
 */
public class Observable<Value> {
    
    ///观察者, 就是一个闭包, 事件发布的时候触发
    public typealias Observer = (Value) -> Void
    
    //序列的迭代器
    private var uniqueId = (0...).makeIterator()
    //递归锁
    private let lock = NSRecursiveLock()
    //销毁的时候执行的闭包
    private var onDispose: () -> Void
    
    //观察者数组
    fileprivate var observerArr: [Int : (Observer, DispatchQueue)] = [:]
    
    
    // onDispose: 销毁的时候执行的闭包
    init(_ onDispose: @escaping () -> Void = {}) {
        self.onDispose = onDispose
    }
    
    /*
     *  事件的订阅函数, 返回一个可以销毁的对象
     *  - observer: 观察者, 就是一个闭包, 事件发布的时候触发
     *  - queue: 在指定的线程里面执行闭包
     */
    public func subscribe(observer: @escaping Observer, on queue: DispatchQueue = .main) -> Disposable {
        lock.lock()
        defer {
            lock.unlock()
        }
        
        let id = uniqueId.next()!
        observerArr[id] = (observer, queue)
        
        //订阅执行完后, 返回一个销毁的对象, 用来销毁自身的观察者, 以及执行销毁后的闭包
        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
    }
}

    1. 提供销毁资源的实例

import Foundation

/*
 *一个用于提供销毁资源的实例.
 *把自身添加到DisposeBag里, DisposeBag在销毁的时候, 会顺便把此对象销毁.
 */
public final class Disposable {
    
    ///销毁后执行的闭包
    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()
        //执行每个Disable销毁前的闭包
        for disposable in disposableARR {
            disposable.dispose()
        }
    }
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,117评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,328评论 1 293
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,839评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,007评论 0 206
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,384评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,629评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,880评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,593评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,313评论 1 243
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,575评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,066评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,392评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,052评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,082评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,844评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,662评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,575评论 2 270

推荐阅读更多精彩内容