如何用websocket開發Swift應用

本篇內容介紹了“如何用websocket開發Swift應用”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！

先看一下我們最終的成果：

1、用URLSession創建WebSocket

借助于URLSessionWebSocketTask，用URLSession創建WebSocket變得非常簡單了。下面說明一個WebSocket連接的5個核心操作。

第1個，打開連接：

let urlSession = URLSession(configuration: .default)
let webSocketTask = urlSession.webSocketTask(with: "wss://ws.finnhub.io?token=XYZ")
webSocketTask.resume()

第2個，發送字符串或數據

webSocketTask.send(.string("Hello")){error in ...}

第3個，接收響應結果消息

webSocketTask.receive{result in  ...}

第4個，斷開連接

webSocketTask.cancel(with: .goingAway, reason: nil)

第5個，我們可以利用ping/pong機制來保證連接保持激活：

webSocketTask?.sendPing { (error) in ... }

2、基于SwiftUI + WebSocket實現比特幣實時行情界面

首先到這個網站生成可以訪問比特幣行情的API KEY：

下面的SwiftUI視圖包含一個SF符號圖片以及一個顯示比特幣實時價格的文本框：

import SwiftUI
import Combine
import Foundation

struct ContentView: View {
    
    @ObservedObject var service = WebSocketService()
    
    var body: some View {
        
        VStack{
            
            Image(systemName: "bitcoinsign.circle.fill")
                .font(.system(size: 150))
                .foregroundColor(Color(red: 247 / 255, green: 142 / 255, blue: 26 / 255))
                .padding()
            
            Text("USD")
                .font(.largeTitle)
                .padding()
            
            Text(service.priceResult)
                .font(.system(size: 60))
            
        }.onAppear {
            self.service.connect()
        }
    }
}

priceResult需要從ObservableObject類發布 —— 可以在下面的WebSocketService.swift 中看到：

class WebSocketService : ObservableObject {
  
    private let urlSession = URLSession(configuration: .default)
    private var webSocketTask: URLSessionWebSocketTask?
    
    private let baseURL = URL(string: "wss://ws.finnhub.io?token=XYZ")!
    
    let didChange = PassthroughSubject<Void, Never>()
    @Published var price: String = ""
    
    private var cancellable: AnyCancellable? = nil
    
    var priceResult: String = "" {
        didSet {
            didChange.send()
        }
    }

    init() {
        cancellable = AnyCancellable($price
            .debounce(for: 0.5, scheduler: DispatchQueue.main)
            .removeDuplicates()
            .assign(to: \.priceResult, on: self))
        
    }
}

在上面的代碼中，我們定義了一些屬性并在init方法中建立了訂閱。你需要使用自己的API KEY來替換XYZ。

用于Publisher上的操作符非常重要。我們使用了Debounce來降低實時更新的頻率，還使用了removeDuplicates來剔除重復的值。

要手動將變化發送到SwiftUI視圖，當priceResult屬性被更新時可以觸發didChange.send()方法。

構造WebSocket URLSession的WebSocketService類的其他代碼如下：

class WebSocketService : ObservableObject {

func connect() {
        
        stop()
        webSocketTask = urlSession.webSocketTask(with: baseURL)
        webSocketTask?.resume()
        
        sendMessage()
        receiveMessage()
    }

    func stop() {
        webSocketTask?.cancel(with: .goingAway, reason: nil)
    }
    
    private func sendMessage()
    {
        let string = "{\"type\":\"subscribe\",\"symbol\":\"BINANCE:BTCUSDT\"}"
        
        let message = URLSessionWebSocketTask.Message.string(string)
        webSocketTask?.send(message) { error in
            if let error = error {
                print("WebSocket couldn’t send message because: \(error)")
            }
        }
    }
    
    private func receiveMessage() {
        webSocketTask?.receive {[weak self] result in
            
            switch result {
            case .failure(let error):
                print("Error in receiving message: \(error)")
            case .success(.string(let str)):
                
                do {
                    let decoder = JSONDecoder()
                    let result = try decoder.decode(APIResponse.self, from: Data(str.utf8))
                    DispatchQueue.main.async{
                        self?.price = "\(result.data[0].p)"
                    }
                    
                } catch  {
                    print("error is \(error.localizedDescription)")
                }
                
                self?.receiveMessage()
                
            default:
                print("default")
            }
        }
    }
}

下面的結構模型用來解碼API的響應結果：

struct APIResponse: Codable {
    var data: [PriceData]
    var type : String
    
    private enum CodingKeys: String, CodingKey {
        case data, type
    }
}
struct PriceData: Codable{
    
    public var p: Float
    
    private enum CodingKeys: String, CodingKey {
        case p
    }
}

現在使用WatchOS仿真器運行應用，看起來是這樣：

“如何用websocket開發Swift應用”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！