reactivefeign.webclient.client5.HttpClient5WebReactiveFeign Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2018 The Feign Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package reactivefeign.webclient.client5;

import org.springframework.http.client.reactive.ClientHttpConnector;
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.web.reactive.function.client.WebClientRequestException;
import reactivefeign.ReactiveOptions;
import reactivefeign.client.ReactiveHttpRequest;
import reactivefeign.client.ReadTimeoutException;
import reactivefeign.webclient.CoreWebBuilder;
import reactivefeign.webclient.WebClientFeignCustomizer;

import java.util.function.BiFunction;

import static reactivefeign.webclient.client5.HttpClient5HttpConnectorBuilder.buildHttp5ClientHttpConnector;


/**
 * {@link WebClient} based implementation of reactive Feign
 *
 * @author Sergii Karpenko
 */
public class HttpClient5WebReactiveFeign {

    public static  Builder builder() {
        return builder(WebClient.builder());
    }

    public static  Builder builder(WebClient.Builder webClientBuilder) {
        return new Builder<>(webClientBuilder);
    }

    public static  Builder builder(WebClient.Builder webClientBuilder,
                                         WebClientFeignCustomizer webClientCustomizer) {
        return new Builder<>(webClientBuilder, webClientCustomizer);
    }

    public static class Builder extends CoreWebBuilder {

        private HttpClient5ReactiveOptions reactiveOptions;

        protected Builder(WebClient.Builder webClientBuilder) {
            super(webClientBuilder);
        }

        protected Builder(WebClient.Builder webClientBuilder, WebClientFeignCustomizer webClientCustomizer) {
            super(webClientBuilder, webClientCustomizer);
        }

        @Override
        public Builder options(ReactiveOptions options) {
            this.reactiveOptions = (HttpClient5ReactiveOptions)options;
            return this;
        }

        @Override
        public BiFunction errorMapper(){
            return (request, throwable) -> {
                if(throwable instanceof WebClientRequestException
                   && throwable.getCause() instanceof java.util.concurrent.TimeoutException){
                    return new ReadTimeoutException(throwable, request);
                }
                return null;
            };
        }

        @Override
        protected ClientHttpConnector clientConnector() {

            return buildHttp5ClientHttpConnector(reactiveOptions);
        }

    }

}