Merge pull request #1418 from DataDog/ncreated/REPLAY-1965/fix-sr-url-for-ap1-dc

REPLAY-1965 fix: SR data uploads to AP1 site

Author: ncreated

CHANGELOG.md

@@ -1,5 +1,6 @@
 # Unreleased
 
+- [BUGFIX] Fix session replay uploads to AP1 site. See [#1418][].
 - [BUGFIX] Allow instantiating custom instance of the SDK after default one. See [#1413][].
 - [BUGFIX] Do not propagate attributes from Errors and LongTasks to Views.
 - [IMPROVEMENT] Upgrade to PLCrashReporter 1.11.1.
@@ -496,6 +497,7 @@ Release `2.0` introduces breaking changes. Follow the [Migration Guide](MIGRATIO
 [#1355]: https://github.com/DataDog/dd-sdk-ios/pull/1355
 [#1410]: https://github.com/DataDog/dd-sdk-ios/pull/1410
 [#1413]: https://github.com/DataDog/dd-sdk-ios/pull/1413
+[#1418]: https://github.com/DataDog/dd-sdk-ios/pull/1418
 [#1419]: https://github.com/DataDog/dd-sdk-ios/pull/1419
 [@00fa9a]: https://github.com/00FA9A
 [@britton-earnin]: https://github.com/Britton-Earnin

DatadogRUM/Sources/Feature/RequestBuilder.swift

@@ -56,7 +56,7 @@ internal struct RequestBuilder: FeatureRequestBuilder {
         return builder.uploadRequest(with: data)
     }
 
-    func url(with context: DatadogContext) -> URL {
+    private func url(with context: DatadogContext) -> URL {
         customIntakeURL ?? context.site.endpoint.appendingPathComponent("api/v2/rum")
     }
 }

DatadogSessionReplay/Sources/Feature/RequestBuilder/RequestBuilder.swift

@@ -30,35 +30,15 @@ internal struct RequestBuilder: FeatureRequestBuilder {
         // error to let the core delete the batch:
         let records = try events.map { try EnrichedRecordJSON(jsonObjectData: $0.data) }
         let segment = try segmentBuilder.createSegmentJSON(from: records)
-        let url = customUploadURL ?? intakeURL(for: context.site)
 
-        return try createRequest(url: url, segment: segment, context: context)
+        return try createRequest(segment: segment, context: context)
     }
 
-    private func intakeURL(for site: DatadogSite) -> URL {
-        // swiftlint:disable force_unwrapping
-        switch site {
-        case .us1:
-            return URL(string: "https://session-replay.browser-intake-datadoghq.com/api/v2/replay")!
-        case .us3:
-            return URL(string: "https://session-replay.browser-intake-us3-datadoghq.com/api/v2/replay")!
-        case .us5:
-            return URL(string: "https://session-replay.browser-intake-us5-datadoghq.com/api/v2/replay")!
-        case .eu1:
-            return URL(string: "https://session-replay.browser-intake-datadoghq.eu/api/v2/replay")!
-        case .ap1:
-            return URL(string: "https://session-replay.browser-intake-datadoghq.eu/api/v2/replay")!
-        case .us1_fed:
-            return URL(string: "https://session-replay.browser-intake-ddog-gov.com/api/v2/replay")!
-        }
-        // swiftlint:enable force_unwrapping
-    }
-
-    private func createRequest(url: URL, segment: SegmentJSON, context: DatadogContext) throws -> URLRequest {
+    private func createRequest(segment: SegmentJSON, context: DatadogContext) throws -> URLRequest {
         var multipart = multipartBuilder
 
         let builder = URLRequestBuilder(
-            url: url,
+            url: url(with: context),
             queryItems: [],
             headers: [
                 .contentTypeHeader(contentType: .multipartFormData(boundary: multipart.boundary.uuidString)),
@@ -97,4 +77,8 @@ internal struct RequestBuilder: FeatureRequestBuilder {
         // Data is already compressed, so request building request w/o compression:
         return builder.uploadRequest(with: multipart.data, compress: false)
     }
+
+    private func url(with context: DatadogContext) -> URL {
+        customUploadURL ?? context.site.endpoint.appendingPathComponent("api/v2/replay")
+    }
 }

DatadogSessionReplay/Tests/Feature/RequestBuilder/RequestBuilderTests.swift

@@ -42,12 +42,12 @@ class RequestBuilderTests: XCTestCase {
         }
 
         // Then
-        XCTAssertEqual(try url(for: .us1), "https://session-replay.browser-intake-datadoghq.com/api/v2/replay")
-        XCTAssertEqual(try url(for: .us3), "https://session-replay.browser-intake-us3-datadoghq.com/api/v2/replay")
-        XCTAssertEqual(try url(for: .us5), "https://session-replay.browser-intake-us5-datadoghq.com/api/v2/replay")
-        XCTAssertEqual(try url(for: .eu1), "https://session-replay.browser-intake-datadoghq.eu/api/v2/replay")
-        XCTAssertEqual(try url(for: .ap1), "https://session-replay.browser-intake-datadoghq.eu/api/v2/replay")
-        XCTAssertEqual(try url(for: .us1_fed), "https://session-replay.browser-intake-ddog-gov.com/api/v2/replay")
+        XCTAssertEqual(try url(for: .us1), "https://browser-intake-datadoghq.com/api/v2/replay")
+        XCTAssertEqual(try url(for: .us3), "https://browser-intake-us3-datadoghq.com/api/v2/replay")
+        XCTAssertEqual(try url(for: .us5), "https://browser-intake-us5-datadoghq.com/api/v2/replay")
+        XCTAssertEqual(try url(for: .eu1), "https://browser-intake-datadoghq.eu/api/v2/replay")
+        XCTAssertEqual(try url(for: .ap1), "https://browser-intake-ap1-datadoghq.com/api/v2/replay")
+        XCTAssertEqual(try url(for: .us1_fed), "https://browser-intake-ddog-gov.com/api/v2/replay")
     }
 
     func testItSetsCustomIntakeURL() {