Publish messages and errors (Service)

Author: backslash-f

Sources/ObservableWebSocketClient/Network/ObservableWebSocketService.swift

@@ -7,12 +7,13 @@
 
 import Foundation
 
-@Observable public final class ObservableWebSocketService {
-    public var session = URLSession(configuration: .default)
+public final class ObservableWebSocketService: ObservableObject {
+
+    @Published public var message: URLSessionWebSocketTask.Message? = nil
 
-    public var message: URLSessionWebSocketTask.Message?
+    @Published public var error: ObservableWebSocketClientError? = nil
 
-    public var error: ObservableWebSocketClientError?
+    public var session = URLSession(configuration: .default)
 
     private let websocketURL: URL
 
@@ -28,28 +29,24 @@ import Foundation
 // MARK: - Private
 
 private extension ObservableWebSocketService {
-
+    
+    func initializeWebSocket() {
+        webSocketTask = session.webSocketTask(with: websocketURL)
+        webSocketTask?.resume()
+    }
+    
     func receiveMessage() {
         webSocketTask?.receive { result in
             switch result {
             case .success(let message):
                 self.message = message
                 // Listen for the next message.
                 self.receiveMessage()
-
+                
             case .failure(let error):
                 let codableError = CodableError(error)
                 self.error = .receivingMessage(codableError)
             }
         }
     }
 }
-
-// MARK: - Private
-
-private extension ObservableWebSocketService {
-    func initializeWebSocket() {
-        webSocketTask = session.webSocketTask(with: websocketURL)
-        webSocketTask?.resume()
-    }
-}