blob: 7aa31bf4ded8599fae0bc02ae7dfd4fb24185cdf (
plain)
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
|
package org.onap.portalng.bff.config;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.annotation.Configuration;
import org.springframework.stereotype.Component;
import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
import org.springframework.web.reactive.function.client.ExchangeStrategies;
import org.springframework.web.reactive.function.client.WebClient;
@Configuration
public class WebClientConfig {
@Component
public static class WebClientBeanPostProcessor implements BeanPostProcessor {
private final ExchangeStrategies exchangeStrategies;
private final ExchangeFilterFunction idTokenExchangeFilterFunction;
private final ExchangeFilterFunction errorHandlingExchangeFilterFunction;
private final ExchangeFilterFunction logResponseExchangeFilterFunction;
public WebClientBeanPostProcessor(
ExchangeStrategies exchangeStrategies,
@Qualifier(BeansConfig.ID_TOKEN_EXCHANGE_FILTER_FUNCTION)
ExchangeFilterFunction idTokenExchangeFilterFunction,
@Qualifier(BeansConfig.ERROR_HANDLING_EXCHANGE_FILTER_FUNCTION)
ExchangeFilterFunction errorHandlingExchangeFilterFunction,
@Qualifier(BeansConfig.LOG_RESPONSE_EXCHANGE_FILTER_FUNCTION)
ExchangeFilterFunction logResponseExchangeFilterFunction) {
this.exchangeStrategies = exchangeStrategies;
this.idTokenExchangeFilterFunction = idTokenExchangeFilterFunction;
this.errorHandlingExchangeFilterFunction = errorHandlingExchangeFilterFunction;
this.logResponseExchangeFilterFunction = logResponseExchangeFilterFunction;
}
@Override
public Object postProcessBeforeInitialization(Object bean, String beanName)
throws BeansException {
return bean;
}
@Override
public Object postProcessAfterInitialization(Object bean, String beanName)
throws BeansException {
if (bean instanceof WebClient.Builder builder) {
return builder
.exchangeStrategies(exchangeStrategies)
.filter(idTokenExchangeFilterFunction)
.filter(errorHandlingExchangeFilterFunction)
.filter(logResponseExchangeFilterFunction);
}
return bean;
}
}
}
|