DisplayLink.swift

DisplayLink.swift

import Nucleus

let display = DisplayLink()

public class DisplayLink {
  
  private init () {
    link = CADisplayLink(target: self, selector: "callClosures")
    link.addToRunLoop(NSRunLoop.mainRunLoop(), forMode: NSRunLoopCommonModes)
  }
  
  public func link (id: String, _ closure: VoidFunc) {
    closures[id] = closure
  }
  
  public func unlink (id: String) {
    closures[id] = nil
  }

  private let link: CADisplayLink!
  private var closures = [String:VoidFunc]()
  
  @objc private func callClosures () {
    for (_, closure) in closures { closure() }
  }
}