iOS разработка на Swift

class WebSocketManager: ObservableObject {
    @Published var messages: [Message] = []
    private var webSocket: URLSessionWebSocketTask?
    
    func connect(with code: String) {
        let url = URL(string: "wss://\(code).otvet-dz.online/ws")!
        webSocket = URLSession.shared.webSocketTask(with: url)
        webSocket?.resume()
        receiveMessage()
    }
    
    private func receiveMessage() {
        webSocket?.receive { [weak self] result in
            switch result {
            case .success(let message):
                self?.handleMessage(message)
                self?.receiveMessage()
            case .failure(let error):
                print("Error: \(error)")
            }
        }
    }
}