fix

ExpediaGroup · Aug 27, 2024 · 0fae1eb · 0fae1eb
1 parent 6208344
commit 0fae1eb
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/core/src/main/kotlin/com/expediagroup/sdk/core/client/Environment.kt b/core/src/main/kotlin/com/expediagroup/sdk/core/client/Environment.kt
@@ -20,10 +20,9 @@ import com.expediagroup.sdk.core.model.Properties
 import com.expediagroup.sdk.core.model.TransactionId
 import io.ktor.client.request.HttpRequestBuilder
 import io.ktor.http.HttpHeaders
-import java.util.UUID
 
 interface EnvironmentProvider {
-    fun HttpRequestBuilder.appendHeaders(transactionId: TransactionId = TransactionId())
+    fun HttpRequestBuilder.appendHeaders(extraHeaders: Map<String, String> = mapOf(HeaderKey.TRANSACTION_ID to TransactionId().dequeue().toString()))
 }
 
 class DefaultEnvironmentProvider(
@@ -36,11 +35,11 @@ class DefaultEnvironmentProvider(
     private val userAgent = "expediagroup-sdk-java-$namespace/${properties["sdk-version"]!!} (Java $javaVersion; $operatingSystemName $operatingSystemVersion)"
 
     @Suppress("MemberVisibilityCanBePrivate")
-    override fun HttpRequestBuilder.appendHeaders(transactionId: TransactionId) {
+    override fun HttpRequestBuilder.appendHeaders(extraHeaders: Map<String, String>) {
         with(headers) {
             append(HttpHeaders.UserAgent, userAgent)
             append(HeaderKey.X_SDK_TITLE, properties["sdk-title"]!!)
-            append(HeaderKey.TRANSACTION_ID, transactionId.dequeue().toString())
+            extraHeaders.forEach { (key, value) -> append(key, value) }
         }
     }
 }