如何用websocket開發(fā)Swift應(yīng)用

本篇內(nèi)容介紹了“如何用websocket開發(fā)Swift應(yīng)用”的有關(guān)知識(shí)，在實(shí)際案例的操作過(guò)程中，不少人都會(huì)遇到這樣的困境，接下來(lái)就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

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

1、用URLSession創(chuàng)建WebSocket

借助于URLSessionWebSocketTask，用URLSession創(chuàng)建WebSocket變得非常簡(jiǎn)單了。下面說(shuō)明一個(gè)WebSocket連接的5個(gè)核心操作。

第1個(gè)，打開連接：

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

第2個(gè)，發(fā)送字符串或數(shù)據(jù)

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

第3個(gè)，接收響應(yīng)結(jié)果消息

webSocketTask.receive{result in  ...}

第4個(gè)，斷開連接

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

第5個(gè)，我們可以利用ping/pong機(jī)制來(lái)保證連接保持激活：

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

2、基于SwiftUI + WebSocket實(shí)現(xiàn)比特幣實(shí)時(shí)行情界面

首先到這個(gè)網(wǎng)站生成可以訪問(wèn)比特幣行情的API KEY：

下面的SwiftUI視圖包含一個(gè)SF符號(hào)圖片以及一個(gè)顯示比特幣實(shí)時(shí)價(jià)格的文本框：

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類發(fā)布 —— 可以在下面的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來(lái)替換XYZ。

用于Publisher上的操作符非常重要。我們使用了Debounce來(lái)降低實(shí)時(shí)更新的頻率，還使用了removeDuplicates來(lái)剔除重復(fù)的值。

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

構(gòu)造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")
            }
        }
    }
}

下面的結(jié)構(gòu)模型用來(lái)解碼API的響應(yīng)結(jié)果：

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
    }
}

現(xiàn)在使用WatchOS仿真器運(yùn)行應(yīng)用，看起來(lái)是這樣：

“如何用websocket開發(fā)Swift應(yīng)用”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！