Skip to content

Commit

Permalink
feat: using customizer to increase read and connect timeout
Browse files Browse the repository at this point in the history
  • Loading branch information
rajadilipkolli committed Apr 7, 2024
1 parent 5339177 commit bf51f8b
Showing 1 changed file with 8 additions and 11 deletions.
Original file line number Diff line number Diff line change
@@ -1,23 +1,20 @@
package com.learning.ai.llmragwithspringai.config;

import java.time.Duration;
import org.springframework.boot.web.client.ClientHttpRequestFactories;
import org.springframework.boot.web.client.ClientHttpRequestFactorySettings;
import org.springframework.boot.web.client.RestClientCustomizer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.client.JdkClientHttpRequestFactory;
import org.springframework.web.client.RestClient;

@Configuration(proxyBeanMethods = false)
public class RestClientBuilderConfig {

@Bean
RestClient.Builder restClientBuilder(JdkClientHttpRequestFactory jdkClientHttpRequestFactory) {
return RestClient.builder().requestFactory(jdkClientHttpRequestFactory);
}

@Bean
JdkClientHttpRequestFactory jdkClientHttpRequestFactory() {
JdkClientHttpRequestFactory jdkClientHttpRequestFactory = new JdkClientHttpRequestFactory();
jdkClientHttpRequestFactory.setReadTimeout(Duration.ofMinutes(5));
return jdkClientHttpRequestFactory;
public RestClientCustomizer restClientCustomizer() {
return restClientBuilder -> restClientBuilder.requestFactory(
ClientHttpRequestFactories.get(ClientHttpRequestFactorySettings.DEFAULTS
.withConnectTimeout(Duration.ofSeconds(60))
.withReadTimeout(Duration.ofMinutes(5))));
}
}

0 comments on commit bf51f8b

Please sign in to comment.