This commit is contained in:
Erik 2015-03-14 14:11:32 -04:00
parent 2cc68aead6
commit c177671112

View File

@ -29,6 +29,7 @@ public typealias AckCallback = (NSArray?) -> Void
@objc public class SocketAckHandler { @objc public class SocketAckHandler {
let ackNum:Int! let ackNum:Int!
let event:String! let event:String!
var acked = false
var callback:AckCallback? var callback:AckCallback?
init(event:String, ackNum:Int = 0) { init(event:String, ackNum:Int = 0) {
@ -36,12 +37,24 @@ public typealias AckCallback = (NSArray?) -> Void
self.event = event self.event = event
} }
func onAck(callback:AckCallback) { public func onAck(timeout:UInt64, withCallback callback:AckCallback) {
self.callback = callback self.callback = callback
let time = dispatch_time(DISPATCH_TIME_NOW, Int64(timeout * NSEC_PER_SEC))
dispatch_after(time, dispatch_get_main_queue()) {[weak self] in
if self == nil {
return
}
if !self!.acked {
self?.executeAck(["No ACK"])
}
}
} }
func executeAck(data:NSArray?) { func executeAck(data:NSArray?) {
dispatch_async(dispatch_get_main_queue()) {[cb = self.callback] in dispatch_async(dispatch_get_main_queue()) {[weak self, cb = self.callback] in
self?.acked = true
cb?(data) cb?(data)
return return
} }