org.http4k.azure.AzureHttpClient.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http4k-azure Show documentation
Http4k Azure integration tooling
The newest version!
package org.http4k.azure

import com.azure.core.http.HttpClient
import com.azure.core.http.HttpHeader
import com.azure.core.http.HttpHeaders
import com.azure.core.http.HttpRequest
import com.azure.core.http.HttpResponse
import org.http4k.core.HttpHandler
import org.http4k.core.Method
import org.http4k.core.Request
import org.http4k.core.Response
import reactor.core.publisher.Flux
import reactor.core.publisher.Mono
import java.io.InputStream
import java.io.PipedInputStream
import java.io.PipedOutputStream
import java.nio.ByteBuffer
import java.nio.charset.Charset
import java.util.concurrent.CountDownLatch

/**
 * Pluggable Http client adapter for Azure SDK.
 */
class AzureHttpClient(private val http: HttpHandler) : HttpClient {
    override fun send(request: HttpRequest) = http(request.toHttp4k()).fromHttp4k(request)
}

internal fun Flux.toInputStream(): InputStream {
    val pipedInputStream = PipedInputStream()
    val pipedOutputStream = PipedOutputStream(pipedInputStream)

    val latch = CountDownLatch(1)

    doOnNext { byteBuffer ->
        val bytes = ByteArray(byteBuffer.remaining())
        byteBuffer.get(bytes)
        pipedOutputStream.write(bytes)
    }.doOnComplete {
        try {
            pipedOutputStream.close()
        } finally {
            latch.countDown()
        }
    }.doOnError {
        try {
            pipedOutputStream.close()
        } finally {
            latch.countDown()
        }
    }.subscribe()

    latch.await()

    return pipedInputStream
}

private fun Response.fromHttp4k(request: HttpRequest): Mono = Mono.just(
    object : HttpResponse(request) {
        override fun getStatusCode() = [email protected]

        @Deprecated("Deprecated in Java")
        override fun getHeaderValue(p0: String) = [email protected](p0) ?: ""

        override fun getHeaders() = HttpHeaders(
            [email protected] { it.first }.map { (k, v) -> HttpHeader(k, v.map { it.second ?: "" }) }
        )

        override fun getBody() = [email protected]()

        fun InputStream.inputStreamToFlux() = Flux.generate {
            val buffer = ByteArray(4096)
            when (val bytesRead = read(buffer)) {
                -1 -> it.complete()
                else -> it.next(ByteBuffer.wrap(buffer, 0, bytesRead))
            }
        }.doFinally {
            close()
        }

        override fun getBodyAsByteArray() = Mono.just([email protected]())

        override fun getBodyAsString() = Mono.just([email protected]())

        override fun getBodyAsString(p0: Charset) = Mono.just([email protected](p0).readText())
    })

private fun HttpRequest.toHttp4k() = Request(Method.valueOf(httpMethod.name), url.toExternalForm())
        .headers(headers.flatMap { h -> h.values.map { h.name to it } })
        .body(body.toInputStream())