1let url = URL(string: "wss://example.com/socket")!
2let webSocketTask = URLSession.shared.webSocketTask(with: url)
3webSocketTask.resume()
4
5// Sending a message
6let message = URLSessionWebSocketTask.Message.string("Hello, server!")
7webSocketTask.send(message) { error in
8    if let error = error {
9        print("Send error: \(error)")
10    }
11}
12
13// Receiving messages
14func receiveMessage() {
15    webSocketTask.receive { result in
16        switch result {
17        case .success(let message):
18            switch message {
19            case .string(let text):
20                print("Received string: \(text)")
21            case .data(let data):
22                print("Received data: \(data)")
23            @unknown default:
24                break
25            }
26            receiveMessage() // Recursively listen
27        case .failure(let error):
28            print("Receive error: \(error)")
29        }
30    }
31}
32receiveMessage()
33
34// Ping/Pong
35webSocketTask.sendPing { error in
36    if let error = error {
37        print("Ping error: \(error)")
38    }
39}
40
41// Close the connection
42webSocketTask.cancel(with: .goingAway, reason: nil)
43

1import Network
2
3let endpoint = NWEndpoint.url(URL(string: "wss://example.com/socket")!)
4let parameters = NWParameters.tls
5parameters.defaultProtocolStack.applicationProtocols.insert(NWProtocolWebSocket.Options(), at: 0)
6
7let connection = NWConnection(to: endpoint, using: parameters)
8connection.stateUpdateHandler = { state in
9    switch state {
10    case .ready:
11        print("WebSocket ready")
12    case .failed(let error):
13        print("Connection failed: \(error)")
14    default:
15        break
16    }
17}
18connection.start(queue: .main)
19

1struct ChatMessage: Codable {
2    let sender: String
3    let content: String
4}
5
6let chatMessage = ChatMessage(sender: "Alice", content: "Hi Bob!")
7let encoder = JSONEncoder()
8if let data = try? encoder.encode(chatMessage),
9   let jsonString = String(data: data, encoding: .utf8) {
10    let message = URLSessionWebSocketTask.Message.string(jsonString)
11    webSocketTask.send(message) { error in
12        if let error = error {
13            print("Send error: \(error)")
14        }
15    }
16}
17
18// Receiving would use the receiveMessage() function from earlier, then decode:
19if case .string(let jsonString) = message {
20    if let data = jsonString.data(using: .utf8),
21       let receivedMessage = try? JSONDecoder().decode(ChatMessage.self, from: data) {
22        print("Chat from \(receivedMessage.sender): \(receivedMessage.content)")
23    }
24}
25

1func reconnectWebSocket(after delay: TimeInterval = 2.0) {
2    DispatchQueue.main.asyncAfter(deadline: .now() + delay) {
3        webSocketTask = URLSession.shared.webSocketTask(with: url)
4        webSocketTask.resume()
5        receiveMessage()
6    }
7}
8
9webSocketTask.receive { result in
10    switch result {
11    case .failure(_):
12        reconnectWebSocket()
13    default:
14        break
15    }
16}
17