com.fireflysource.net.http.client.impl.AsyncHttpClientRequestBuilder.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of firefly-net Show documentation
There is a newer version: 5.0.2
package com.fireflysource.net.http.client.impl

import com.fireflysource.net.http.client.*
import com.fireflysource.net.http.client.impl.HttpProtocolNegotiator.addHttp2UpgradeHeader
import com.fireflysource.net.http.client.impl.content.provider.ByteBufferContentProvider
import com.fireflysource.net.http.client.impl.content.provider.MultiPartContentProvider
import com.fireflysource.net.http.client.impl.content.provider.StringContentProvider
import com.fireflysource.net.http.common.model.*
import java.nio.ByteBuffer
import java.nio.charset.Charset
import java.nio.charset.StandardCharsets
import java.util.concurrent.CompletableFuture
import java.util.function.Supplier

class AsyncHttpClientRequestBuilder(
    private val connectionManager: HttpClientConnectionManager,
    method: String,
    uri: HttpURI,
    httpVersion: HttpVersion
) : HttpClientRequestBuilder {

    private val multiPartContentProvider: MultiPartContentProvider by lazy { MultiPartContentProvider() }

    val httpRequest: AsyncHttpClientRequest = AsyncHttpClientRequest()

    init {
        httpRequest.method = method
        httpRequest.uri = uri
        httpRequest.httpVersion = httpVersion
    }

    override fun cookies(cookies: MutableList?): HttpClientRequestBuilder {
        httpRequest.cookies = cookies
        return this
    }

    override fun put(name: String, list: MutableList): HttpClientRequestBuilder {
        httpRequest.httpFields.put(name, list)
        return this
    }

    override fun put(header: HttpHeader, value: String): HttpClientRequestBuilder {
        httpRequest.httpFields.put(header, value)
        return this
    }

    override fun put(name: String, value: String): HttpClientRequestBuilder {
        httpRequest.httpFields.put(name, value)
        return this
    }

    override fun put(field: HttpField): HttpClientRequestBuilder {
        httpRequest.httpFields.put(field)
        return this
    }

    override fun addAll(fields: HttpFields): HttpClientRequestBuilder {
        httpRequest.httpFields.addAll(fields)
        return this
    }

    override fun add(field: HttpField): HttpClientRequestBuilder {
        httpRequest.httpFields.add(field)
        return this
    }

    override fun addCsv(header: HttpHeader, vararg values: String): HttpClientRequestBuilder {
        httpRequest.httpFields.addCSV(header, *values)
        return this
    }

    override fun addCsv(header: String, vararg values: String): HttpClientRequestBuilder {
        httpRequest.httpFields.addCSV(header, *values)
        return this
    }

    override fun trailerSupplier(trailerSupplier: Supplier?): HttpClientRequestBuilder {
        httpRequest.trailerSupplier = trailerSupplier
        return this
    }

    override fun body(content: String): HttpClientRequestBuilder = body(content, StandardCharsets.UTF_8)

    override fun body(content: String, charset: Charset): HttpClientRequestBuilder =
        contentProvider(StringContentProvider(content, charset))

    override fun body(buffer: ByteBuffer): HttpClientRequestBuilder = contentProvider(ByteBufferContentProvider(buffer))

    override fun contentProvider(contentProvider: HttpClientContentProvider?): HttpClientRequestBuilder {
        httpRequest.contentProvider = contentProvider
        return this
    }

    override fun addPart(
        name: String,
        content: HttpClientContentProvider,
        fields: HttpFields?
    ): HttpClientRequestBuilder {
        contentProvider(multiPartContentProvider)
        multiPartContentProvider.addPart(name, content, fields)
        return this
    }

    override fun addFilePart(
        name: String,
        fileName: String,
        content: HttpClientContentProvider,
        fields: HttpFields?
    ): HttpClientRequestBuilder {
        contentProvider(multiPartContentProvider)
        multiPartContentProvider.addFilePart(name, fileName, content, fields)
        return this
    }

    override fun addFormInput(name: String, value: String): HttpClientRequestBuilder {
        httpRequest.formInputs.add(name, value)
        return this
    }

    override fun addFormInputs(name: String, values: MutableList): HttpClientRequestBuilder {
        httpRequest.formInputs.addValues(name, values)
        return this
    }

    override fun putFormInput(name: String, value: String): HttpClientRequestBuilder {
        httpRequest.formInputs.put(name, value)
        return this
    }

    override fun putFormInputs(name: String, values: MutableList): HttpClientRequestBuilder {
        httpRequest.formInputs.putValues(name, values)
        return this
    }

    override fun removeFormInput(name: String): HttpClientRequestBuilder {
        httpRequest.formInputs.remove(name)
        return this
    }

    override fun addQueryString(name: String, value: String): HttpClientRequestBuilder {
        httpRequest.queryStrings.add(name, value)
        return this
    }

    override fun addQueryStrings(name: String, values: MutableList): HttpClientRequestBuilder {
        httpRequest.queryStrings.addValues(name, values)
        return this
    }

    override fun putQueryString(name: String, value: String): HttpClientRequestBuilder {
        httpRequest.queryStrings.put(name, value)
        return this
    }

    override fun putQueryStrings(name: String, values: MutableList): HttpClientRequestBuilder {
        httpRequest.queryStrings[name] = values
        return this
    }

    override fun removeQueryString(name: String): HttpClientRequestBuilder {
        httpRequest.queryStrings.remove(name)
        return this
    }

    override fun contentHandler(contentHandler: HttpClientContentHandler?): HttpClientRequestBuilder {
        httpRequest.contentHandler = contentHandler
        return this
    }

    override fun http2Settings(http2Settings: Map?): HttpClientRequestBuilder {
        httpRequest.http2Settings = http2Settings
        return this
    }

    override fun upgradeHttp2(): HttpClientRequestBuilder {
        addHttp2UpgradeHeader(httpRequest)
        return this
    }

    override fun submit(): CompletableFuture {
        return connectionManager.send(httpRequest)
    }

    override fun getHttpClientRequest(): HttpClientRequest = httpRequest

}