|
|
|
|
@ -13,6 +13,7 @@
|
|
|
|
|
* 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.tencent.cloud.metadata.config;
|
|
|
|
|
@ -20,13 +21,18 @@ package com.tencent.cloud.metadata.config;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
|
|
|
|
|
import com.tencent.cloud.metadata.core.filter.gateway.Metadata2HeaderScgFilter;
|
|
|
|
|
import com.tencent.cloud.metadata.core.interceptor.Metadata2HeaderFeignInterceptor;
|
|
|
|
|
import com.tencent.cloud.metadata.core.interceptor.Metadata2HeaderRestTemplateInterceptor;
|
|
|
|
|
import com.tencent.cloud.common.constant.MetadataConstant;
|
|
|
|
|
import com.tencent.cloud.metadata.core.DecodeTransferMetadataReactiveFilter;
|
|
|
|
|
import com.tencent.cloud.metadata.core.DecodeTransferMetadataServletFilter;
|
|
|
|
|
import com.tencent.cloud.metadata.core.EncodeTransferMedataFeignInterceptor;
|
|
|
|
|
import com.tencent.cloud.metadata.core.EncodeTransferMedataRestTemplateInterceptor;
|
|
|
|
|
import com.tencent.cloud.metadata.core.EncodeTransferMedataScgFilter;
|
|
|
|
|
|
|
|
|
|
import org.springframework.beans.BeansException;
|
|
|
|
|
import org.springframework.beans.factory.config.BeanPostProcessor;
|
|
|
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
|
|
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
|
|
|
|
|
import org.springframework.boot.web.servlet.FilterRegistrationBean;
|
|
|
|
|
import org.springframework.cloud.gateway.filter.GlobalFilter;
|
|
|
|
|
import org.springframework.context.ApplicationContext;
|
|
|
|
|
import org.springframework.context.ApplicationContextAware;
|
|
|
|
|
@ -36,6 +42,11 @@ import org.springframework.http.client.ClientHttpRequestInterceptor;
|
|
|
|
|
import org.springframework.util.CollectionUtils;
|
|
|
|
|
import org.springframework.web.client.RestTemplate;
|
|
|
|
|
|
|
|
|
|
import static javax.servlet.DispatcherType.ASYNC;
|
|
|
|
|
import static javax.servlet.DispatcherType.ERROR;
|
|
|
|
|
import static javax.servlet.DispatcherType.FORWARD;
|
|
|
|
|
import static javax.servlet.DispatcherType.INCLUDE;
|
|
|
|
|
import static javax.servlet.DispatcherType.REQUEST;
|
|
|
|
|
/**
|
|
|
|
|
* Metadata transfer auto configuration.
|
|
|
|
|
*
|
|
|
|
|
@ -44,6 +55,46 @@ import org.springframework.web.client.RestTemplate;
|
|
|
|
|
@Configuration
|
|
|
|
|
public class MetadataTransferAutoConfiguration {
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Create when web application type is SERVLET.
|
|
|
|
|
*/
|
|
|
|
|
@Configuration
|
|
|
|
|
@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.SERVLET)
|
|
|
|
|
static class MetadataServletFilterConfig {
|
|
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
|
public FilterRegistrationBean<DecodeTransferMetadataServletFilter> metadataServletFilterRegistrationBean(
|
|
|
|
|
DecodeTransferMetadataServletFilter decodeTransferMetadataServletFilter) {
|
|
|
|
|
FilterRegistrationBean<DecodeTransferMetadataServletFilter> filterRegistrationBean = new FilterRegistrationBean<>(
|
|
|
|
|
decodeTransferMetadataServletFilter);
|
|
|
|
|
filterRegistrationBean.setDispatcherTypes(ASYNC, ERROR, FORWARD, INCLUDE,
|
|
|
|
|
REQUEST);
|
|
|
|
|
filterRegistrationBean
|
|
|
|
|
.setOrder(MetadataConstant.OrderConstant.WEB_FILTER_ORDER);
|
|
|
|
|
return filterRegistrationBean;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
|
public DecodeTransferMetadataServletFilter metadataServletFilter() {
|
|
|
|
|
return new DecodeTransferMetadataServletFilter();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Create when web application type is REACTIVE.
|
|
|
|
|
*/
|
|
|
|
|
@Configuration
|
|
|
|
|
@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.REACTIVE)
|
|
|
|
|
static class MetadataReactiveFilterConfig {
|
|
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
|
public DecodeTransferMetadataReactiveFilter metadataReactiveFilter() {
|
|
|
|
|
return new DecodeTransferMetadataReactiveFilter();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Create when gateway application is SCG.
|
|
|
|
|
*/
|
|
|
|
|
@ -52,8 +103,8 @@ public class MetadataTransferAutoConfiguration {
|
|
|
|
|
static class MetadataTransferScgFilterConfig {
|
|
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
|
public GlobalFilter metadata2HeaderScgFilter() {
|
|
|
|
|
return new Metadata2HeaderScgFilter();
|
|
|
|
|
public GlobalFilter encodeTransferMedataScgFilter() {
|
|
|
|
|
return new EncodeTransferMedataScgFilter();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
@ -66,8 +117,8 @@ public class MetadataTransferAutoConfiguration {
|
|
|
|
|
static class MetadataTransferFeignInterceptorConfig {
|
|
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
|
public Metadata2HeaderFeignInterceptor metadata2HeaderFeignInterceptor() {
|
|
|
|
|
return new Metadata2HeaderFeignInterceptor();
|
|
|
|
|
public EncodeTransferMedataFeignInterceptor encodeTransferMedataFeignInterceptor() {
|
|
|
|
|
return new EncodeTransferMedataFeignInterceptor();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
@ -82,13 +133,13 @@ public class MetadataTransferAutoConfiguration {
|
|
|
|
|
private ApplicationContext context;
|
|
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
|
public Metadata2HeaderRestTemplateInterceptor metadata2HeaderRestTemplateInterceptor() {
|
|
|
|
|
return new Metadata2HeaderRestTemplateInterceptor();
|
|
|
|
|
public EncodeTransferMedataRestTemplateInterceptor encodeTransferMedataRestTemplateInterceptor() {
|
|
|
|
|
return new EncodeTransferMedataRestTemplateInterceptor();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
|
BeanPostProcessor metadata2HeaderRestTemplatePostProcessor(
|
|
|
|
|
Metadata2HeaderRestTemplateInterceptor metadata2HeaderRestTemplateInterceptor) {
|
|
|
|
|
BeanPostProcessor encodeTransferMetadataRestTemplatePostProcessor(
|
|
|
|
|
EncodeTransferMedataRestTemplateInterceptor encodeTransferMedataRestTemplateInterceptor) {
|
|
|
|
|
// Coping with multiple bean injection scenarios
|
|
|
|
|
Map<String, RestTemplate> beans = this.context.getBeansOfType(RestTemplate.class);
|
|
|
|
|
// If the restTemplate has been created when the
|
|
|
|
|
@ -98,13 +149,15 @@ public class MetadataTransferAutoConfiguration {
|
|
|
|
|
for (RestTemplate restTemplate : beans.values()) {
|
|
|
|
|
List<ClientHttpRequestInterceptor> interceptors = restTemplate.getInterceptors();
|
|
|
|
|
// Avoid setting interceptor repeatedly.
|
|
|
|
|
if (null != interceptors && !interceptors.contains(metadata2HeaderRestTemplateInterceptor)) {
|
|
|
|
|
interceptors.add(metadata2HeaderRestTemplateInterceptor);
|
|
|
|
|
if (null != interceptors && !interceptors
|
|
|
|
|
.contains(encodeTransferMedataRestTemplateInterceptor)) {
|
|
|
|
|
interceptors.add(encodeTransferMedataRestTemplateInterceptor);
|
|
|
|
|
restTemplate.setInterceptors(interceptors);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return new Metadata2HeaderRestTemplatePostProcessor(metadata2HeaderRestTemplateInterceptor);
|
|
|
|
|
return new EncodeTransferMetadataRestTemplatePostProcessor(
|
|
|
|
|
encodeTransferMedataRestTemplateInterceptor);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
@ -112,13 +165,14 @@ public class MetadataTransferAutoConfiguration {
|
|
|
|
|
this.context = applicationContext;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static class Metadata2HeaderRestTemplatePostProcessor implements BeanPostProcessor {
|
|
|
|
|
public static class EncodeTransferMetadataRestTemplatePostProcessor
|
|
|
|
|
implements BeanPostProcessor {
|
|
|
|
|
|
|
|
|
|
private Metadata2HeaderRestTemplateInterceptor metadata2HeaderRestTemplateInterceptor;
|
|
|
|
|
private EncodeTransferMedataRestTemplateInterceptor encodeTransferMedataRestTemplateInterceptor;
|
|
|
|
|
|
|
|
|
|
Metadata2HeaderRestTemplatePostProcessor(
|
|
|
|
|
Metadata2HeaderRestTemplateInterceptor metadata2HeaderRestTemplateInterceptor) {
|
|
|
|
|
this.metadata2HeaderRestTemplateInterceptor = metadata2HeaderRestTemplateInterceptor;
|
|
|
|
|
EncodeTransferMetadataRestTemplatePostProcessor(
|
|
|
|
|
EncodeTransferMedataRestTemplateInterceptor encodeTransferMedataRestTemplateInterceptor) {
|
|
|
|
|
this.encodeTransferMedataRestTemplateInterceptor = encodeTransferMedataRestTemplateInterceptor;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
@ -132,8 +186,9 @@ public class MetadataTransferAutoConfiguration {
|
|
|
|
|
RestTemplate restTemplate = (RestTemplate) bean;
|
|
|
|
|
List<ClientHttpRequestInterceptor> interceptors = restTemplate.getInterceptors();
|
|
|
|
|
// Avoid setting interceptor repeatedly.
|
|
|
|
|
if (null != interceptors && !interceptors.contains(metadata2HeaderRestTemplateInterceptor)) {
|
|
|
|
|
interceptors.add(this.metadata2HeaderRestTemplateInterceptor);
|
|
|
|
|
if (null != interceptors && !interceptors
|
|
|
|
|
.contains(encodeTransferMedataRestTemplateInterceptor)) {
|
|
|
|
|
interceptors.add(this.encodeTransferMedataRestTemplateInterceptor);
|
|
|
|
|
restTemplate.setInterceptors(interceptors);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
Haotian Zhang
3 years ago
committed by
GitHub