fix(realtime): Set default heartbeat interval to 25s (#667)

* fix

* fix(realtime): Set default heartbeat interval to 25s

Author: grdsdev

Sources/Realtime/RealtimeClientV2.swift

@@ -16,7 +16,8 @@ import Helpers
 public typealias JSONObject = Helpers.JSONObject
 
 /// Factory function for returning a new WebSocket connection.
-typealias WebSocketTransport = @Sendable () async throws -> any WebSocket
+typealias WebSocketTransport = @Sendable (_ url: URL, _ headers: [String: String]) async throws ->
+  any WebSocket
 
 public final class RealtimeClientV2: Sendable {
   struct MutableState {
@@ -93,14 +94,11 @@ public final class RealtimeClientV2: Sendable {
     self.init(
       url: url,
       options: options,
-      wsTransport: {
+      wsTransport: { url, headers in
         let configuration = URLSessionConfiguration.default
-        configuration.httpAdditionalHeaders = options.headers.dictionary
+        configuration.httpAdditionalHeaders = headers
         return try await URLSessionWebSocket.connect(
-          to: Self.realtimeWebSocketURL(
-            baseURL: Self.realtimeBaseURL(url: url),
-            apikey: options.apikey
-          ),
+          to: url,
           configuration: configuration
         )
       },
@@ -172,7 +170,14 @@ public final class RealtimeClientV2: Sendable {
         status = .connecting
 
         do {
-          let conn = try await wsTransport()
+          let conn = try await wsTransport(
+            Self.realtimeWebSocketURL(
+              baseURL: Self.realtimeBaseURL(url: url),
+              apikey: options.apikey,
+              logLevel: options.logLevel
+            ),
+            options.headers.dictionary
+          )
           mutableState.withValue { $0.conn = conn }
           onConnected(reconnect: reconnect)
         } catch {
@@ -528,7 +533,7 @@ public final class RealtimeClientV2: Sendable {
     return url
   }
 
-  static func realtimeWebSocketURL(baseURL: URL, apikey: String?) -> URL {
+  static func realtimeWebSocketURL(baseURL: URL, apikey: String?, logLevel: LogLevel?) -> URL {
     guard var components = URLComponents(url: baseURL, resolvingAgainstBaseURL: false)
     else {
       return baseURL
@@ -540,6 +545,10 @@ public final class RealtimeClientV2: Sendable {
     }
     components.queryItems!.append(URLQueryItem(name: "vsn", value: "1.0.0"))
 
+    if let logLevel {
+      components.queryItems!.append(URLQueryItem(name: "log_level", value: logLevel.rawValue))
+    }
+
     components.path.append("/websocket")
     components.path = components.path.replacingOccurrences(of: "//", with: "/")
 

Sources/Realtime/Types.swift

@@ -21,11 +21,14 @@ public struct RealtimeClientOptions: Sendable {
   var timeoutInterval: TimeInterval
   var disconnectOnSessionLoss: Bool
   var connectOnSubscribe: Bool
+
+  /// Sets the log level for Realtime
+  var logLevel: LogLevel?
   var fetch: (@Sendable (_ request: URLRequest) async throws -> (Data, URLResponse))?
   package var accessToken: (@Sendable () async throws -> String?)?
   package var logger: (any SupabaseLogger)?
 
-  public static let defaultHeartbeatInterval: TimeInterval = 15
+  public static let defaultHeartbeatInterval: TimeInterval = 25
   public static let defaultReconnectDelay: TimeInterval = 7
   public static let defaultTimeoutInterval: TimeInterval = 10
   public static let defaultDisconnectOnSessionLoss = true
@@ -38,6 +41,7 @@ public struct RealtimeClientOptions: Sendable {
     timeoutInterval: TimeInterval = Self.defaultTimeoutInterval,
     disconnectOnSessionLoss: Bool = Self.defaultDisconnectOnSessionLoss,
     connectOnSubscribe: Bool = Self.defaultConnectOnSubscribe,
+    logLevel: LogLevel? = nil,
     fetch: (@Sendable (_ request: URLRequest) async throws -> (Data, URLResponse))? = nil,
     accessToken: (@Sendable () async throws -> String?)? = nil,
     logger: (any SupabaseLogger)? = nil
@@ -48,6 +52,7 @@ public struct RealtimeClientOptions: Sendable {
     self.timeoutInterval = timeoutInterval
     self.disconnectOnSessionLoss = disconnectOnSessionLoss
     self.connectOnSubscribe = connectOnSubscribe
+    self.logLevel = logLevel
     self.fetch = fetch
     self.accessToken = accessToken
     self.logger = logger
@@ -84,3 +89,8 @@ public enum RealtimeClientStatus: Sendable, CustomStringConvertible {
 extension HTTPField.Name {
   static let apiKey = Self("apiKey")!
 }
+
+/// Log level for Realtime.
+public enum LogLevel: String, Sendable {
+  case info, warn, error
+}

Sources/TestHelpers/MockExtensions.swift

@@ -12,7 +12,7 @@ import InlineSnapshotTesting
 extension Mock {
   package func snapshotRequest(
     message: @autoclosure () -> String = "",
-    record isRecording: Bool? = nil,
+    record isRecording: SnapshotTestingConfiguration.Record? = nil,
     timeout: TimeInterval = 5,
     syntaxDescriptor: InlineSnapshotSyntaxDescriptor = InlineSnapshotSyntaxDescriptor(),
     matches expected: (() -> String)? = nil,

Supabase.xcworkspace/xcshareddata/swiftpm/Package.resolved

@@ -14,7 +14,7 @@ import XCTest
 
 @available(macOS 13.0, iOS 16.0, watchOS 9.0, tvOS 16.0, *)
 final class RealtimeTests: XCTestCase {
-  let url = URL(string: "https://localhost:54321/realtime/v1")!
+  let url = URL(string: "http://localhost:54321/realtime/v1")!
   let apiKey = "anon.api.key"
 
   override func invokeTest() {
@@ -49,7 +49,7 @@ final class RealtimeTests: XCTestCase {
           "custom.access.token"
         }
       ),
-      wsTransport: { self.client },
+      wsTransport: { _, _ in self.client },
       http: http
     )
   }
@@ -60,6 +60,30 @@ final class RealtimeTests: XCTestCase {
     super.tearDown()
   }
 
+  func test_transport() async {
+    let client = RealtimeClientV2(
+      url: url,
+      options: RealtimeClientOptions(
+        headers: ["apikey": apiKey],
+        logLevel: .warn,
+        accessToken: {
+          "custom.access.token"
+        }
+      ),
+      wsTransport: { url, headers in
+        assertInlineSnapshot(of: url, as: .description) {
+          """
+          ws://localhost:54321/realtime/v1/websocket?apikey=anon.api.key&vsn=1.0.0&log_level=warn
+          """
+        }
+        return FakeWebSocket.fakes().0
+      },
+      http: http
+    )
+
+    await client.connect()
+  }
+
   func testBehavior() async throws {
     let channel = sut.channel("public:messages")
     var subscriptions: Set<ObservationToken> = []
@@ -352,7 +376,7 @@ final class RealtimeTests: XCTestCase {
     let request = await http.receivedRequests.last
     assertInlineSnapshot(of: request?.urlRequest, as: .raw(pretty: true)) {
       """
-      POST https://localhost:54321/realtime/v1/api/broadcast
+      POST http://localhost:54321/realtime/v1/api/broadcast
       Authorization: Bearer custom.access.token
       Content-Type: application/json
       apiKey: anon.api.key

Tests/RealtimeTests/RealtimeTests.swift

@@ -34,7 +34,7 @@ final class _PushTests: XCTestCase {
       options: RealtimeClientOptions(
         headers: ["apiKey": "apikey"]
       ),
-      wsTransport: { client },
+      wsTransport: { _, _ in client },
       http: HTTPClientMock()
     )
   }