如何用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应用”的内容就介绍到这里了，感谢大家的阅读。如果想了解更多行业相关的知识可以关注亿速云网站，小编将为大家输出更多高质量的实用文章！