com.livk.commons.http.RestClientConfiguration Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of spring-extension-commons Show documentation
spring commons packages
The newest version!
/*
 * Copyright 2021-2024 spring-boot-extension the original author or 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
 *
 *       https://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 com.livk.commons.http;

import com.livk.auto.service.annotation.SpringAutoService;
import com.livk.commons.http.annotation.EnableRestClient;
import com.livk.commons.http.support.OkHttpClientHttpRequestFactory;
import okhttp3.ConnectionPool;
import okhttp3.OkHttpClient;
import org.springframework.boot.autoconfigure.AutoConfiguration;
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
import org.springframework.boot.web.client.RestClientCustomizer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestClient;

import java.util.concurrent.TimeUnit;

/**
 * 
 * RestClientConfiguration
 * 
 *
 * @author livk
 */
@AutoConfiguration
@SpringAutoService(EnableRestClient.class)
@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.SERVLET)
public class RestClientConfiguration {

	/**
	 * Rest client
	 * @param builder builder
	 * @return client
	 */
	@Bean
	public RestClient restClient(RestClient.Builder builder) {
		return builder.build();
	}

	/**
	 * The type Ok http client configuration.
	 */
	@Configuration(enforceUniqueMethods = false)
	@ConditionalOnClass(OkHttpClient.class)
	public static class OkHttpClientConfiguration {

		/**
		 * Rest template customizer rest template customizer.
		 * @return the rest template customizer
		 */
		@Bean
		@ConditionalOnMissingBean(OkHttpClient.class)
		public RestClientCustomizer restClientCustomizer() {
			ConnectionPool pool = new ConnectionPool(200, 300, TimeUnit.SECONDS);
			OkHttpClientHttpRequestFactory requestFactory = new OkHttpClientHttpRequestFactory().connectionPool(pool)
				.connectTimeout(3, TimeUnit.SECONDS)
				.readTimeout(3, TimeUnit.SECONDS)
				.writeTimeout(3, TimeUnit.SECONDS);
			return builder -> builder.requestFactory(requestFactory);
		}

		/**
		 * Rest template customizer rest template customizer.
		 * @param okHttpClient the ok http client
		 * @return the rest template customizer
		 */
		@Bean
		@ConditionalOnBean(OkHttpClient.class)
		public RestClientCustomizer restClientCustomizer(OkHttpClient okHttpClient) {
			return builder -> builder.requestFactory(new OkHttpClientHttpRequestFactory(okHttpClient));
		}

	}

}