[kotlin-client][jvm-spring-webclient] Extract data from PartConfig fo…

…r multipart/form-data requests (#19811)
OpenAPITools · Oct 10, 2024 · d14eab8 · d14eab8
1 parent 627c0f4
commit d14eab8
Show file tree

Hide file tree

Showing 4 changed files with 76 additions and 8 deletions.
diff --git a/...sources/kotlin-client/libraries/jvm-spring-webclient/infrastructure/ApiClient.kt.mustache b/...sources/kotlin-client/libraries/jvm-spring-webclient/infrastructure/ApiClient.kt.mustache
@@ -6,6 +6,7 @@ import org.springframework.http.HttpMethod
 import org.springframework.http.MediaType
 import org.springframework.web.reactive.function.client.WebClient
 import org.springframework.http.ResponseEntity
+import org.springframework.http.client.MultipartBodyBuilder
 import org.springframework.util.LinkedMultiValueMap
 import reactor.core.publisher.Mono
 
@@ -47,8 +48,24 @@ open class ApiClient(protected val client: WebClient) {
     private fun <I> WebClient.RequestBodySpec.headers(requestConfig: RequestConfig<I>) =
         apply { requestConfig.headers.forEach { (name, value) -> header(name, value) } }
 
-    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>) =
-        apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>): WebClient.RequestBodySpec {
+        when {
+            requestConfig.headers[HttpHeaders.CONTENT_TYPE] == MediaType.MULTIPART_FORM_DATA_VALUE -> {
+                val builder = MultipartBodyBuilder()
+                (requestConfig.body as Map<String, PartConfig<*>>).forEach { (name, part) ->
+                    if (part.body != null) {
+                        val partBuilder = builder.part(name, part.body)
+                        val partHeaders = part.headers
+                        partHeaders.forEach { partBuilder.header(it.key, it.value) }
+                    }
+                }
+                return apply { bodyValue(builder.build()) }
+            }
+            else -> {
+                return apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+            }
+        }
+    }
 }
 
 inline fun <reified T: Any> parseDateToQueryString(value : T): String {

diff --git a/...vm-spring-3-webclient/src/main/kotlin/org/openapitools/client/infrastructure/ApiClient.kt b/...vm-spring-3-webclient/src/main/kotlin/org/openapitools/client/infrastructure/ApiClient.kt
@@ -6,6 +6,7 @@ import org.springframework.http.HttpMethod
 import org.springframework.http.MediaType
 import org.springframework.web.reactive.function.client.WebClient
 import org.springframework.http.ResponseEntity
+import org.springframework.http.client.MultipartBodyBuilder
 import org.springframework.util.LinkedMultiValueMap
 import reactor.core.publisher.Mono
 
@@ -47,8 +48,24 @@ open class ApiClient(protected val client: WebClient) {
     private fun <I> WebClient.RequestBodySpec.headers(requestConfig: RequestConfig<I>) =
         apply { requestConfig.headers.forEach { (name, value) -> header(name, value) } }
 
-    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>) =
-        apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>): WebClient.RequestBodySpec {
+        when {
+            requestConfig.headers[HttpHeaders.CONTENT_TYPE] == MediaType.MULTIPART_FORM_DATA_VALUE -> {
+                val builder = MultipartBodyBuilder()
+                (requestConfig.body as Map<String, PartConfig<*>>).forEach { (name, part) ->
+                    if (part.body != null) {
+                        val partBuilder = builder.part(name, part.body)
+                        val partHeaders = part.headers
+                        partHeaders.forEach { partBuilder.header(it.key, it.value) }
+                    }
+                }
+                return apply { bodyValue(builder.build()) }
+            }
+            else -> {
+                return apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+            }
+        }
+    }
 }
 
 inline fun <reified T: Any> parseDateToQueryString(value : T): String {

diff --git a/...vm-spring-2-webclient/src/main/kotlin/org/openapitools/client/infrastructure/ApiClient.kt b/...vm-spring-2-webclient/src/main/kotlin/org/openapitools/client/infrastructure/ApiClient.kt
@@ -6,6 +6,7 @@ import org.springframework.http.HttpMethod
 import org.springframework.http.MediaType
 import org.springframework.web.reactive.function.client.WebClient
 import org.springframework.http.ResponseEntity
+import org.springframework.http.client.MultipartBodyBuilder
 import org.springframework.util.LinkedMultiValueMap
 import reactor.core.publisher.Mono
 
@@ -47,8 +48,24 @@ open class ApiClient(protected val client: WebClient) {
     private fun <I> WebClient.RequestBodySpec.headers(requestConfig: RequestConfig<I>) =
         apply { requestConfig.headers.forEach { (name, value) -> header(name, value) } }
 
-    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>) =
-        apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>): WebClient.RequestBodySpec {
+        when {
+            requestConfig.headers[HttpHeaders.CONTENT_TYPE] == MediaType.MULTIPART_FORM_DATA_VALUE -> {
+                val builder = MultipartBodyBuilder()
+                (requestConfig.body as Map<String, PartConfig<*>>).forEach { (name, part) ->
+                    if (part.body != null) {
+                        val partBuilder = builder.part(name, part.body)
+                        val partHeaders = part.headers
+                        partHeaders.forEach { partBuilder.header(it.key, it.value) }
+                    }
+                }
+                return apply { bodyValue(builder.build()) }
+            }
+            else -> {
+                return apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+            }
+        }
+    }
 }
 
 inline fun <reified T: Any> parseDateToQueryString(value : T): String {

diff --git a/...vm-spring-3-webclient/src/main/kotlin/org/openapitools/client/infrastructure/ApiClient.kt b/...vm-spring-3-webclient/src/main/kotlin/org/openapitools/client/infrastructure/ApiClient.kt
@@ -6,6 +6,7 @@ import org.springframework.http.HttpMethod
 import org.springframework.http.MediaType
 import org.springframework.web.reactive.function.client.WebClient
 import org.springframework.http.ResponseEntity
+import org.springframework.http.client.MultipartBodyBuilder
 import org.springframework.util.LinkedMultiValueMap
 import reactor.core.publisher.Mono
 
@@ -47,8 +48,24 @@ open class ApiClient(protected val client: WebClient) {
     private fun <I> WebClient.RequestBodySpec.headers(requestConfig: RequestConfig<I>) =
         apply { requestConfig.headers.forEach { (name, value) -> header(name, value) } }
 
-    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>) =
-        apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+    private fun <I : Any> WebClient.RequestBodySpec.body(requestConfig: RequestConfig<I>): WebClient.RequestBodySpec {
+        when {
+            requestConfig.headers[HttpHeaders.CONTENT_TYPE] == MediaType.MULTIPART_FORM_DATA_VALUE -> {
+                val builder = MultipartBodyBuilder()
+                (requestConfig.body as Map<String, PartConfig<*>>).forEach { (name, part) ->
+                    if (part.body != null) {
+                        val partBuilder = builder.part(name, part.body)
+                        val partHeaders = part.headers
+                        partHeaders.forEach { partBuilder.header(it.key, it.value) }
+                    }
+                }
+                return apply { bodyValue(builder.build()) }
+            }
+            else -> {
+                return apply { if (requestConfig.body != null) bodyValue(requestConfig.body) }
+            }
+        }
+    }
 }
 
 inline fun <reified T: Any> parseDateToQueryString(value : T): String {