74 lines
2.4 KiB
Swift
74 lines
2.4 KiB
Swift
//
|
|
// ScenicQueueSocketClient.swift
|
|
// suixinkan
|
|
//
|
|
|
|
import Foundation
|
|
|
|
/// 排队 WebSocket 客户端，订阅打卡点队列刷新与叫号事件。
|
|
final class ScenicQueueSocketClient {
|
|
var onMessage: ((ScenicQueueSocketMessage) -> Void)?
|
|
|
|
private let environment: APIEnvironment
|
|
private let session: URLSession
|
|
private let decoder = JSONDecoder()
|
|
private var task: URLSessionWebSocketTask?
|
|
private var connectedSpotId: Int64?
|
|
|
|
init(environment: APIEnvironment = .current, session: URLSession = .shared) {
|
|
self.environment = environment
|
|
self.session = session
|
|
}
|
|
|
|
/// 建立连接并订阅指定打卡点。
|
|
func connect(socketToken: String, scenicId: Int64, scenicSpotId: Int64) {
|
|
guard !socketToken.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty else { return }
|
|
disconnect()
|
|
connectedSpotId = scenicSpotId
|
|
let webSocketTask = session.webSocketTask(with: environment.webSocketURL)
|
|
task = webSocketTask
|
|
webSocketTask.resume()
|
|
sendSubscribePayloads(scenicSpotId: scenicSpotId)
|
|
receiveLoop()
|
|
}
|
|
|
|
/// 断开连接。
|
|
func disconnect() {
|
|
task?.cancel(with: .normalClosure, reason: nil)
|
|
task = nil
|
|
connectedSpotId = nil
|
|
}
|
|
|
|
private func sendSubscribePayloads(scenicSpotId: Int64) {
|
|
[
|
|
#"{"type":306,"params":{"scenic_spot_id":\#(scenicSpotId)}}"#,
|
|
#"{"type":307,"params":{"scenic_spot_id":\#(scenicSpotId)}}"#,
|
|
].forEach { payload in
|
|
task?.send(.string(payload)) { _ in }
|
|
}
|
|
}
|
|
|
|
private func receiveLoop() {
|
|
task?.receive { [weak self] result in
|
|
guard let self else { return }
|
|
switch result {
|
|
case .success(let message):
|
|
if case .string(let text) = message {
|
|
self.handle(text)
|
|
}
|
|
self.receiveLoop()
|
|
case .failure:
|
|
self.task = nil
|
|
}
|
|
}
|
|
}
|
|
|
|
private func handle(_ text: String) {
|
|
let trimmed = text.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
guard trimmed.hasPrefix("{"), let data = trimmed.data(using: .utf8) else { return }
|
|
guard let message = try? decoder.decode(ScenicQueueSocketMessage.self, from: data) else { return }
|
|
guard message.isQueueEvent else { return }
|
|
onMessage?(message)
|
|
}
|
|
}
|