WebSocket 双端实践（iOS/ Golang）

一、使用Starscream（swift）完成客户端长链需求

首先附上Starscream：GitHub地址

第一步：将Starsream导入到项目。
打开Podfile，加上：

pod ‘Starscream’, '~> 4.0.0’复制代码
接着pod install。

第二步：实现WebSocket能力。
导入头文件，import Starscream

初始化WebSocket，把一些请求头包装一下（与服务端对好）

private func initWebSocket() {
// 包装请求头
var request = URLRequest(url: URL(string: “ws://127.0.0.1:8000/chat”)!)
request.timeoutInterval = 5 // Sets the timeout for the connection
request.setValue(“some message”, forHTTPHeaderField: “Qi-WebSocket-Header”)
request.setValue(“some message”, forHTTPHeaderField: “Qi-WebSocket-Protocol”)
request.setValue(“0.0.1”, forHTTPHeaderField: “Qi-WebSocket-Version”)
request.setValue(“some message”, forHTTPHeaderField: “Qi-WebSocket-Protocol-2”)
socketManager = WebSocket(request: request)
socketManager?.delegate = self
}复制代码
同时，我用三个Button的点击事件，分别模拟了connect（连接）、write（通信）、disconnect（断开）。

// Mark - Actions
// 连接
@objc func connetButtonClicked() {
    socketManager?.connect()
}
// 通信
@objc func sendButtonClicked() {
    socketManager?.write(string: "some message.")
}
// 断开
@objc func closeButtonCliked() {
    socketManager?.disconnect()
}复制代码

第三步：实现WebSocket回调方法（接收服务端消息）
遵守并实现WebSocketDelegate。

extension ViewController: WebSocketDelegate {
    // 通信（与服务端协商好）
    func didReceive(event: WebSocketEvent, client: WebSocket) {
        switch event {
        case .connected(let headers):
            isConnected = true
            print("websocket is connected: \(headers)")
        case .disconnected(let reason, let code):
            isConnected = false
            print("websocket is disconnected: \(reason) with code: \(code)")