websocket and user center

2018-09-03 14:43:21 +08:00
parent e64e011d8f
commit 3a3e6eeee9
92 changed files with 11993 additions and 2376 deletions
@@ -0,0 +1,145 @@
+//
+//  SocketAckEmitter.swift
+//  Socket.IO-Client-Swift
+//
+//  Created by Erik Little on 9/16/15.
+//
+//  Permission is hereby granted, free of charge, to any person obtaining a copy
+//  of this software and associated documentation files (the "Software"), to deal
+//  in the Software without restriction, including without limitation the rights
+//  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+//  copies of the Software, and to permit persons to whom the Software is
+//  furnished to do so, subject to the following conditions:
+//
+//  The above copyright notice and this permission notice shall be included in
+//  all copies or substantial portions of the Software.
+//
+//  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+//  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+//  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+//  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+//  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+//  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+//  THE SOFTWARE.
+
+import Dispatch
+import Foundation
+
+/// A class that represents a waiting ack call.
+///
+/// **NOTE**: You should not store this beyond the life of the event handler.
+public final class SocketAckEmitter : NSObject {
+    private unowned let socket: SocketIOClient
+    private let ackNum: Int
+
+    /// A view into this emitter where emits do not check for binary data.
+    ///
+    /// Usage:
+    ///
+    /// ```swift
+    /// ack.rawEmitView.with(myObject)
+    /// ```
+    ///
+    /// **NOTE**: It is not safe to hold on to this view beyond the life of the socket.
+    @objc
+    public private(set) lazy var rawEmitView = SocketRawAckView(socket: socket, ackNum: ackNum)
+
+    // MARK: Properties
+
+    /// If true, this handler is expecting to be acked. Call `with(_: SocketData...)` to ack.
+    public var expected: Bool {
+        return ackNum != -1
+    }
+
+    // MARK: Initializers
+
+    /// Creates a new `SocketAckEmitter`.
+    ///
+    /// - parameter socket: The socket for this emitter.
+    /// - parameter ackNum: The ack number for this emitter.
+    public init(socket: SocketIOClient, ackNum: Int) {
+        self.socket = socket
+        self.ackNum = ackNum
+    }
+
+    // MARK: Methods
+
+    /// Call to ack receiving this event.
+    ///
+    /// If an error occurs trying to transform `items` into their socket representation, a `SocketClientEvent.error`
+    /// will be emitted. The structure of the error data is `[ackNum, items, theError]`
+    ///
+    /// - parameter items: A variable number of items to send when acking.
+    public func with(_ items: SocketData...) {
+        guard ackNum != -1 else { return }
+
+        do {
+            socket.emitAck(ackNum, with: try items.map({ try $0.socketRepresentation() }))
+        } catch {
+            socket.handleClientEvent(.error, data: [ackNum, items, error])
+        }
+    }
+
+    /// Call to ack receiving this event.
+    ///
+    /// - parameter items: An array of items to send when acking. Use `[]` to send nothing.
+    @objc
+    public func with(_ items: [Any]) {
+        guard ackNum != -1 else { return }
+
+        socket.emitAck(ackNum, with: items)
+    }
+
+}
+
+/// A class that represents an emit that will request an ack that has not yet been sent.
+/// Call `timingOut(after:callback:)` to complete the emit
+/// Example:
+///
+/// ```swift
+/// socket.emitWithAck("myEvent").timingOut(after: 1) {data in
+///     ...
+/// }
+/// ```
+public final class OnAckCallback : NSObject {
+    private let ackNumber: Int
+    private let binary: Bool
+    private let items: [Any]
+
+    private weak var socket: SocketIOClient?
+
+    init(ackNumber: Int, items: [Any], socket: SocketIOClient, binary: Bool = true) {
+        self.ackNumber = ackNumber
+        self.items = items
+        self.socket = socket
+        self.binary = binary
+    }
+
+    deinit {
+        DefaultSocketLogger.Logger.log("OnAckCallback for \(ackNumber) being released", type: "OnAckCallback")
+    }
+
+    // MARK: Methods
+
+    /// Completes an emitWithAck. If this isn't called, the emit never happens.
+    ///
+    /// - parameter seconds: The number of seconds before this emit times out if an ack hasn't been received.
+    /// - parameter callback: The callback called when an ack is received, or when a timeout happens.
+    ///                       To check for timeout, use `SocketAckStatus`'s `noAck` case.
+    @objc
+    public func timingOut(after seconds: Double, callback: @escaping AckCallback) {
+        guard let socket = self.socket, ackNumber != -1 else { return }
+
+        socket.ackHandlers.addAck(ackNumber, callback: callback)
+        socket.emit(items, ack: ackNumber, binary: binary)
+
+        guard seconds != 0 else { return }
+
+        socket.manager?.handleQueue.asyncAfter(deadline: DispatchTime.now() + seconds) {[weak socket] in
+            guard let socket = socket else { return }
+
+            socket.ackHandlers.timeoutAck(self.ackNumber)
+        }
+    }
+
+}
@@ -0,0 +1,77 @@
+//
+//  SocketAckManager.swift
+//  Socket.IO-Client-Swift
+//
+//  Created by Erik Little on 4/3/15.
+//
+//  Permission is hereby granted, free of charge, to any person obtaining a copy
+//  of this software and associated documentation files (the "Software"), to deal
+//  in the Software without restriction, including without limitation the rights
+//  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+//  copies of the Software, and to permit persons to whom the Software is
+//  furnished to do so, subject to the following conditions:
+//
+//  The above copyright notice and this permission notice shall be included in
+//  all copies or substantial portions of the Software.
+//
+//  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+//  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+//  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+//  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+//  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+//  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+//  THE SOFTWARE.
+
+import Dispatch
+import Foundation
+
+/// The status of an ack.
+public enum SocketAckStatus : String {
+    // MARK: Cases
+
+    /// The ack timed out.
+    case noAck = "NO ACK"
+}
+
+private struct SocketAck : Hashable {
+    let ack: Int
+    var callback: AckCallback!
+    var hashValue: Int {
+        return ack.hashValue
+    }
+
+    init(ack: Int) {
+        self.ack = ack
+    }
+
+    init(ack: Int, callback: @escaping AckCallback) {
+        self.ack = ack
+        self.callback = callback
+    }
+
+    fileprivate static func <(lhs: SocketAck, rhs: SocketAck) -> Bool {
+        return lhs.ack < rhs.ack
+    }
+
+    fileprivate static func ==(lhs: SocketAck, rhs: SocketAck) -> Bool {
+        return lhs.ack == rhs.ack
+    }
+}
+
+class SocketAckManager {
+    private var acks = Set<SocketAck>(minimumCapacity: 1)
+
+    func addAck(_ ack: Int, callback: @escaping AckCallback) {
+        acks.insert(SocketAck(ack: ack, callback: callback))
+    }
+
+    /// Should be called on handle queue
+    func executeAck(_ ack: Int, with items: [Any]) {
+        acks.remove(SocketAck(ack: ack))?.callback(items)
+    }
+
+    /// Should be called on handle queue
+    func timeoutAck(_ ack: Int) {
+       acks.remove(SocketAck(ack: ack))?.callback?([SocketAckStatus.noAck.rawValue])
+    }
+}