Merge pull request #1 from TelenorNorway/feature/header-support

feat: Support header and multiple headers
TelenorNorway · Mar 22, 2024 · 5052b8d · 5052b8d
2 parents 23512b0 + 8cb9d00
commit 5052b8d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/main/kotlin/no/telenor/kt/restclient/util/util.kt b/src/main/kotlin/no/telenor/kt/restclient/util/util.kt
@@ -3,6 +3,7 @@ package no.telenor.kt.restclient.util
 import org.springframework.http.MediaType
 import org.springframework.http.ResponseEntity
 import org.springframework.web.client.RestClient
+import org.springframework.web.client.RestClient.RequestBodySpec
 import org.springframework.web.client.RestClient.RequestHeadersSpec
 import org.springframework.web.client.RestClient.RequestHeadersUriSpec
 import org.springframework.web.client.RestClientResponseException
@@ -13,10 +14,16 @@ import java.net.URI
 // --- request body utility---
 
 @Suppress("UNNECESSARY_NOT_NULL_ASSERTION")
-fun RestClient.RequestBodySpec.json(value: Any) = this
+fun RequestBodySpec.json(value: Any) = this
 	.body(value)
 	.contentType(MediaType.APPLICATION_JSON)!!
 
+fun RequestHeadersSpec<*>.header(header: String, value: Any) : RequestHeadersSpec<*> = this
+	.header(header, value.toString())
+
+fun RequestHeadersSpec<*>.headers(vararg headersWithValue: Pair<String, Any>) : RequestHeadersSpec<*> = this
+	.headers { httpHeaders -> headersWithValue.forEach { httpHeaders.add(it.first, it.second.toString()) }}
+
 // --- response body utility ---
 
 inline fun <reified T : Any> RequestHeadersSpec<*>.retrieveJsonOrNull(): T? = this