org.eclipse.edc.http.client.EdcHttpClientImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http-lib Show documentation
edc :: http-lib
There is a newer version: 0.9.1
/*
 *  Copyright (c) 2022 Bayerische Motoren Werke Aktiengesellschaft (BMW AG)
 *
 *  This program and the accompanying materials are made available under the
 *  terms of the Apache License, Version 2.0 which is available at
 *  https://www.apache.org/licenses/LICENSE-2.0
 *
 *  SPDX-License-Identifier: Apache-2.0
 *
 *  Contributors:
 *       Bayerische Motoren Werke Aktiengesellschaft (BMW AG) - initial API and implementation
 *
 */

package org.eclipse.edc.http.client;

import dev.failsafe.RetryPolicy;
import okhttp3.HttpUrl;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.Response;
import okhttp3.dnsoverhttps.DnsOverHttps;
import org.eclipse.edc.http.spi.EdcHttpClient;
import org.eclipse.edc.http.spi.FallbackFactory;
import org.eclipse.edc.spi.monitor.Monitor;
import org.eclipse.edc.spi.result.Result;

import java.io.IOException;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.function.Function;

import static dev.failsafe.okhttp.FailsafeCall.with;
import static java.util.Collections.emptyList;
import static java.util.Objects.requireNonNull;

public class EdcHttpClientImpl implements EdcHttpClient {

    private final OkHttpClient okHttpClient;
    private final RetryPolicy retryPolicy;
    private final Monitor monitor;

    public EdcHttpClientImpl(OkHttpClient okHttpClient, RetryPolicy retryPolicy, Monitor monitor) {
        this.okHttpClient = okHttpClient;
        this.retryPolicy = retryPolicy;
        this.monitor = monitor;
    }

    @Override
    public Response execute(Request request) throws IOException {
        return execute(request, emptyList());
    }

    @Override
    public Response execute(Request request, List fallbacks) throws IOException {
        var call = okHttpClient.newCall(request);
        var builder = with(retryPolicy);
        fallbacks.stream().map(it -> it.create(request)).forEach(builder::compose);
        return builder.compose(call).execute();
    }

    @Override
    public  Result execute(Request request, Function> mappingFunction) {
        return execute(request, emptyList(), mappingFunction);
    }

    @Override
    public  Result execute(Request request, List fallbacks, Function> mappingFunction) {
        try (var response = execute(request, fallbacks)) {
            return mappingFunction.apply(response);
        } catch (Throwable e) {
            monitor.severe("HTTP client exception caught for request [%s, %s]".formatted(request.method(), request.url()), e);
            return Result.failure(e.getMessage());
        }
    }

    @Override
    public CompletableFuture executeAsync(Request request, List fallbacks) {
        var call = okHttpClient.newCall(request);
        var builder = with(retryPolicy);
        fallbacks.stream().map(it -> it.create(request)).forEach(builder::compose);

        return builder.compose(call)
                .executeAsync();
    }

    @Override
    public EdcHttpClient withDns(String dnsServer) {
        var url = requireNonNull(HttpUrl.get(dnsServer));

        var dns = new DnsOverHttps.Builder()
                .client(okHttpClient)
                .url(url)
                .includeIPv6(false)
                .build();

        return new EdcHttpClientImpl(okHttpClient.newBuilder().dns(dns).build(), retryPolicy, monitor);
    }

}