1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
/*
* ============LICENSE_START=======================================================
* Copyright (C) 2024 Nordix Foundation.
* ================================================================================
* 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.
*
* SPDX-License-Identifier: Apache-2.0
* ============LICENSE_END=========================================================
*/
package org.onap.cps.ncmp.api.impl.config;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import io.netty.channel.ChannelOption;
import io.netty.handler.timeout.ReadTimeoutHandler;
import io.netty.handler.timeout.WriteTimeoutHandler;
import io.netty.resolver.DefaultAddressResolverGroup;
import java.time.Duration;
import java.util.concurrent.TimeUnit;
import lombok.RequiredArgsConstructor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.client.reactive.ReactorClientHttpConnector;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.netty.http.client.HttpClient;
import reactor.netty.resources.ConnectionProvider;
/**
* Configures and creates WebClient beans for various DMI services including data, model, and health check services.
* The configuration utilizes Netty-based HttpClient with custom connection settings, read and write timeouts,
* and initializes WebClient with these settings to ensure optimal performance and resource management.
*/
@Configuration
@RequiredArgsConstructor
public class DmiWebClientConfiguration {
private final HttpClientConfiguration httpClientConfiguration;
private static final Duration DEFAULT_RESPONSE_TIMEOUT = Duration.ofSeconds(30);
/**
* Configures and creates a WebClient bean for DMI data services.
*
* @return a WebClient instance configured for data services.
*/
@Bean
public WebClient dataServicesWebClient() {
final HttpClientConfiguration.DataServices dataServiceConfig = httpClientConfiguration.getDataServices();
final ConnectionProvider dataServicesConnectionProvider
= getConnectionProvider(dataServiceConfig.getConnectionProviderName(),
dataServiceConfig.getMaximumConnectionsTotal(), dataServiceConfig.getPendingAcquireMaxCount());
final HttpClient dataServicesHttpClient = createHttpClient(dataServiceConfig, dataServicesConnectionProvider);
return buildAndGetWebClient(dataServicesHttpClient, dataServiceConfig.getMaximumInMemorySizeInMegabytes());
}
/**
* Configures and creates a WebClient bean for DMI model services.
*
* @return a WebClient instance configured for model services.
*/
@Bean
public WebClient modelServicesWebClient() {
final HttpClientConfiguration.ModelServices modelServiceConfig = httpClientConfiguration.getModelServices();
final ConnectionProvider modelServicesConnectionProvider
= getConnectionProvider(modelServiceConfig.getConnectionProviderName(),
modelServiceConfig.getMaximumConnectionsTotal(),
modelServiceConfig.getPendingAcquireMaxCount());
final HttpClient modelServicesHttpClient
= createHttpClient(modelServiceConfig, modelServicesConnectionProvider);
return buildAndGetWebClient(modelServicesHttpClient, modelServiceConfig.getMaximumInMemorySizeInMegabytes());
}
/**
* Configures and creates a WebClient bean for DMI health check services.
*
* @return a WebClient instance configured for health check services.
*/
@Bean
public WebClient healthChecksWebClient() {
final HttpClientConfiguration.HealthCheckServices healthCheckServiceConfig
= httpClientConfiguration.getHealthCheckServices();
final ConnectionProvider healthChecksConnectionProvider
= getConnectionProvider(healthCheckServiceConfig.getConnectionProviderName(),
healthCheckServiceConfig.getMaximumConnectionsTotal(),
healthCheckServiceConfig.getPendingAcquireMaxCount());
final HttpClient healthChecksHttpClient
= createHttpClient(healthCheckServiceConfig, healthChecksConnectionProvider);
return buildAndGetWebClient(healthChecksHttpClient,
healthCheckServiceConfig.getMaximumInMemorySizeInMegabytes());
}
/**
* Provides a WebClient.Builder bean for creating WebClient instances.
*
* @return a WebClient.Builder instance.
*/
@Bean
public WebClient.Builder webClientBuilder() {
return WebClient.builder();
}
private static HttpClient createHttpClient(final HttpClientConfiguration.ServiceConfig serviceConfig,
final ConnectionProvider connectionProvider) {
return HttpClient.create(connectionProvider)
.responseTimeout(DEFAULT_RESPONSE_TIMEOUT)
.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, serviceConfig.getConnectionTimeoutInSeconds() * 1000)
.doOnConnected(connection -> connection.addHandlerLast(new ReadTimeoutHandler(
serviceConfig.getReadTimeoutInSeconds(), TimeUnit.SECONDS)).addHandlerLast(
new WriteTimeoutHandler(serviceConfig.getWriteTimeoutInSeconds(), TimeUnit.SECONDS)))
.resolver(DefaultAddressResolverGroup.INSTANCE)
.compress(true);
}
@SuppressFBWarnings("BC_UNCONFIRMED_CAST_OF_RETURN_VALUE")
private static ConnectionProvider getConnectionProvider(final String connectionProviderName,
final int maximumConnectionsTotal,
final int pendingAcquireMaxCount) {
return ConnectionProvider.builder(connectionProviderName)
.maxConnections(maximumConnectionsTotal)
.pendingAcquireMaxCount(pendingAcquireMaxCount)
.build();
}
private WebClient buildAndGetWebClient(final HttpClient httpClient,
final int maximumInMemorySizeInMegabytes) {
return webClientBuilder()
.defaultHeaders(header -> header.set(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE))
.defaultHeaders(header -> header.set(HttpHeaders.ACCEPT, MediaType.APPLICATION_JSON_VALUE))
.clientConnector(new ReactorClientHttpConnector(httpClient))
.codecs(configurer -> configurer.defaultCodecs()
.maxInMemorySize(maximumInMemorySizeInMegabytes * 1024 * 1024)).build();
}
}
|
