use socketioclientoptions for engine

This commit is contained in:
Erik 2015-10-21 15:07:00 -04:00
parent 0dd3def2e4
commit 9abf2e77e0
4 changed files with 30 additions and 14 deletions

View File

@ -15,7 +15,7 @@ class SocketEngineTest: XCTestCase {
override func setUp() {
super.setUp()
client = SocketIOClient(socketURL: "")
engine = SocketEngine(client: client, opts: nil)
engine = SocketEngine(client: client, options: nil)
client.setTestable()
}

View File

@ -42,7 +42,6 @@ public final class SocketEngine: NSObject, SocketEngineSpec, WebSocketDelegate {
private let handleQueue = dispatch_queue_create("com.socketio.engineHandleQueue", DISPATCH_QUEUE_SERIAL)
private let logType = "SocketEngine"
private let parseQueue = dispatch_queue_create("com.socketio.engineParseQueue", DISPATCH_QUEUE_SERIAL)
private let session: NSURLSession!
private let workQueue = NSOperationQueue()
private var closed = false
@ -63,6 +62,7 @@ public final class SocketEngine: NSObject, SocketEngineSpec, WebSocketDelegate {
private var postWait = [String]()
private var probing = false
private var probeWait = ProbeWaitQueue()
private var session: NSURLSession!
private var waitingForPoll = false
private var waitingForPost = false
private var websocketConnected = false
@ -71,16 +71,33 @@ public final class SocketEngine: NSObject, SocketEngineSpec, WebSocketDelegate {
private(set) var polling = true
private(set) var websocket = false
public init(client: SocketEngineClient, opts: NSDictionary?) {
init(client: SocketEngineClient, options: Set<SocketIOClientOption>) {
self.client = client
for option in options {
switch option {
case .SessionDelegate(let delegate):
session = NSURLSession(configuration: .defaultSessionConfiguration(),
delegate: opts?["sessionDelegate"] as? NSURLSessionDelegate,
delegate: delegate,
delegateQueue: workQueue)
forceWebsockets = opts?["forceWebsockets"] as? Bool ?? false
forcePolling = opts?["forcePolling"] as? Bool ?? false
cookies = opts?["cookies"] as? [NSHTTPCookie]
socketPath = opts?["path"] as? String ?? ""
extraHeaders = opts?["extraHeaders"] as? [String: String]
case .ForcePolling(let force):
forcePolling = force
case .ForceWebsockets(let force):
forceWebsockets = force
case .Cookies(let cookies):
self.cookies = cookies
case .Path(let path):
socketPath = path
case .ExtraHeaders(let headers):
extraHeaders = headers
default:
continue
}
}
}
public convenience init(client: SocketEngineClient, options: NSDictionary?) {
self.init(client: client,
options: SocketIOClientOption.NSDictionaryToSocketOptionsSet(options ?? [:]))
}
deinit {

View File

@ -33,7 +33,7 @@ import Foundation
var urlPolling: String {get}
var urlWebSocket: String {get}
init(client: SocketEngineClient, opts: NSDictionary?)
init(client: SocketEngineClient, options: NSDictionary?)
func close()
func open(opts: [String: AnyObject]?)

View File

@ -111,8 +111,7 @@ public final class SocketIOClient: NSObject, SocketEngineClient {
private func addEngine() -> SocketEngine {
Logger.log("Adding engine", type: logType)
let newEngine = SocketEngine(client: self, opts:
SocketIOClientOption.SocketOptionsSetToNSDictionary(options ?? []))
let newEngine = SocketEngine(client: self, options: options ?? [])
engine = newEngine
return newEngine