ru.tinkoff.kora.http.client.ok.OkHttpRequestBody Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of http-client-ok Show documentation

Kora http-client-ok module

The newest version!

package ru.tinkoff.kora.http.client.ok;

import jakarta.annotation.Nonnull;
import jakarta.annotation.Nullable;
import okhttp3.MediaType;
import okhttp3.RequestBody;
import okio.BufferedSink;
import ru.tinkoff.kora.http.client.common.HttpClientConnectionException;
import ru.tinkoff.kora.http.client.common.HttpClientEncoderException;
import ru.tinkoff.kora.http.common.body.HttpBodyOutput;

import java.io.IOException;
import java.util.Objects;

public final class OkHttpRequestBody extends RequestBody {
    private final HttpBodyOutput bodyOutput;

    public OkHttpRequestBody(HttpBodyOutput bodyOutput) {
        this.bodyOutput = Objects.requireNonNull(bodyOutput);
    }

    @Nullable
    @Override
    public MediaType contentType() {
        var contentType = bodyOutput.contentType();
        if (contentType == null) {
            return null;
        }
        return MediaType.get(contentType);
    }

    @Override
    public long contentLength() {
        return bodyOutput.contentLength();
    }

    @Override
    public void writeTo(@Nonnull BufferedSink bufferedSink) {
        try {
            bodyOutput.write(bufferedSink.outputStream());
        } catch (IOException e) {
            throw new HttpClientConnectionException(e);
        } catch (Exception e) {
            throw new HttpClientEncoderException(e);
        }
    }
}