ru.tinkoff.kora.s3.client.minio.MinioS3ClientModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of s3-client-minio Show documentation
Kora s3-client-minio module
There is a newer version: 1.1.15
package ru.tinkoff.kora.s3.client.minio;

import io.minio.MinioAsyncClient;
import io.minio.MinioClient;
import io.minio.credentials.Provider;
import io.minio.credentials.StaticProvider;
import io.minio.http.HttpUtils;
import jakarta.annotation.Nullable;
import okhttp3.OkHttpClient;
import org.jetbrains.annotations.ApiStatus;
import ru.tinkoff.kora.common.DefaultComponent;
import ru.tinkoff.kora.common.Tag;
import ru.tinkoff.kora.config.common.Config;
import ru.tinkoff.kora.config.common.extractor.ConfigValueExtractor;
import ru.tinkoff.kora.s3.client.S3ClientModule;
import ru.tinkoff.kora.s3.client.S3Config;
import ru.tinkoff.kora.s3.client.S3KoraAsyncClient;
import ru.tinkoff.kora.s3.client.S3KoraClient;
import ru.tinkoff.kora.s3.client.telemetry.S3ClientTelemetryFactory;
import ru.tinkoff.kora.s3.client.telemetry.S3KoraClientTelemetryFactory;

import java.util.concurrent.TimeUnit;

@ApiStatus.Experimental
public interface MinioS3ClientModule extends S3ClientModule {

    @DefaultComponent
    default Provider minioCredentialProvider(S3Config s3Config) {
        return new StaticProvider(s3Config.accessKey(), s3Config.secretKey(), null);
    }

    default MinioS3ClientConfig minioS3ClientConfig(Config config, ConfigValueExtractor extractor) {
        var value = config.get("s3client.minio");
        return extractor.extract(value);
    }

    @Tag(MinioClient.class)
    @DefaultComponent
    default OkHttpClient minioOkHttpClient(@Nullable OkHttpClient okHttpClient,
                                           MinioS3ClientConfig minioS3ClientConfig) {
        long timeout = minioS3ClientConfig.requestTimeout().toMillis();
        return (okHttpClient == null)
            ? HttpUtils.newDefaultHttpClient(timeout, timeout, timeout)
            : okHttpClient.newBuilder().callTimeout(timeout, TimeUnit.MILLISECONDS).build();
    }

    default MinioClient minioClient(S3Config s3Config,
                                    MinioS3ClientConfig minioS3ClientConfig,
                                    Provider provider,
                                    S3ClientTelemetryFactory telemetryFactory,
                                    @Tag(MinioClient.class) OkHttpClient okHttpClient) {
        MinioClient builded = MinioClient.builder()
            .endpoint(s3Config.url())
            .region(s3Config.region())
            .credentialsProvider(provider)
            .httpClient(okHttpClient.newBuilder()
                .addInterceptor(new MinioS3ClientTelemetryInterceptor(telemetryFactory.get(s3Config.telemetry(), MinioClient.class), minioS3ClientConfig.addressStyle()))
                .build())
            .build();

        if (minioS3ClientConfig.addressStyle() == MinioS3ClientConfig.AddressStyle.PATH) {
            builded.disableVirtualStyleEndpoint();
        } else {
            builded.enableVirtualStyleEndpoint();
        }

        return builded;
    }

    default MinioAsyncClient minioAsyncClient(S3Config s3Config,
                                              MinioS3ClientConfig minioS3ClientConfig,
                                              Provider provider,
                                              S3ClientTelemetryFactory telemetryFactory,
                                              @Tag(MinioClient.class) OkHttpClient okHttpClient) {
        MinioAsyncClient builded = MinioAsyncClient.builder()
            .endpoint(s3Config.url())
            .region(s3Config.region())
            .credentialsProvider(provider)
            .httpClient(okHttpClient.newBuilder()
                .addInterceptor(new MinioS3ClientTelemetryInterceptor(telemetryFactory.get(s3Config.telemetry(), MinioAsyncClient.class), minioS3ClientConfig.addressStyle()))
                .build())
            .build();

        if (minioS3ClientConfig.addressStyle() == MinioS3ClientConfig.AddressStyle.PATH) {
            builded.disableVirtualStyleEndpoint();
        } else {
            builded.enableVirtualStyleEndpoint();
        }

        return builded;
    }

    default S3KoraClient MinioS3KoraClient(MinioClient minioClient,
                                           MinioS3ClientConfig minioS3ClientConfig,
                                           S3Config s3Config,
                                           S3KoraClientTelemetryFactory telemetryFactory) {
        var telemetry = telemetryFactory.get(s3Config.telemetry(), MinioClient.class);
        return new MinioS3KoraClient(minioClient, minioS3ClientConfig, telemetry);
    }

    default S3KoraAsyncClient MinioS3KoraAsyncClient(MinioAsyncClient minioAsyncClient,
                                                     MinioS3ClientConfig minioS3ClientConfig,
                                                     S3Config s3Config,
                                                     S3KoraClientTelemetryFactory telemetryFactory) {
        var telemetry = telemetryFactory.get(s3Config.telemetry(), MinioAsyncClient.class);
        return new MinioS3KoraAsyncClient(minioAsyncClient, minioS3ClientConfig, telemetry);
    }
}