shedfreewu
2 years ago
parent
3f5eda54ff
commit
d721d27441
@ -0,0 +1,74 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<parent>
|
||||
<artifactId>spring-cloud-tencent-plugin-starters</artifactId>
|
||||
<groupId>com.tencent.cloud</groupId>
|
||||
<version>${revision}</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
|
||||
<artifactId>spring-cloud-tencent-lossless-plugin</artifactId>
|
||||
<name>Spring Cloud Tencent Lossless Plugin</name>
|
||||
|
||||
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>com.tencent.cloud</groupId>
|
||||
<artifactId>spring-cloud-tencent-polaris-context</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.tencent.cloud</groupId>
|
||||
<artifactId>spring-cloud-tencent-commons</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.tencent.cloud</groupId>
|
||||
<artifactId>spring-cloud-starter-tencent-polaris-discovery</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.tencent.polaris</groupId>
|
||||
<artifactId>lossless-register</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.tencent.polaris</groupId>
|
||||
<artifactId>lossless-deregister</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-aop</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-test</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-web</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.tencent.polaris</groupId>
|
||||
<artifactId>polaris-test-mock-discovery</artifactId>
|
||||
<scope>test</scope>
|
||||
<exclusions>
|
||||
<exclusion>
|
||||
<groupId>junit</groupId>
|
||||
<artifactId>junit</artifactId>
|
||||
</exclusion>
|
||||
</exclusions>
|
||||
</dependency>
|
||||
|
||||
</dependencies>
|
||||
|
||||
</project>
|
||||
@ -0,0 +1,95 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless;
|
||||
|
||||
import com.tencent.cloud.common.util.ReflectionUtils;
|
||||
import com.tencent.cloud.plugin.lossless.config.LosslessProperties;
|
||||
import com.tencent.cloud.polaris.context.PolarisSDKContextManager;
|
||||
import com.tencent.polaris.api.pojo.BaseInstance;
|
||||
import com.tencent.polaris.api.pojo.DefaultBaseInstance;
|
||||
|
||||
import org.springframework.beans.BeansException;
|
||||
import org.springframework.beans.factory.config.BeanPostProcessor;
|
||||
import org.springframework.cloud.client.serviceregistry.AbstractAutoServiceRegistration;
|
||||
import org.springframework.cloud.client.serviceregistry.Registration;
|
||||
import org.springframework.cloud.client.serviceregistry.ServiceRegistry;
|
||||
|
||||
/**
|
||||
* Wrap Spring Bean and proxy for serviceRegistry.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
public class LosslessBeanPostProcessor implements BeanPostProcessor {
|
||||
|
||||
private PolarisSDKContextManager polarisSDKContextManager;
|
||||
|
||||
private LosslessProperties losslessProperties;
|
||||
|
||||
private Registration registration;
|
||||
|
||||
private Integer port;
|
||||
|
||||
public LosslessBeanPostProcessor(PolarisSDKContextManager polarisSDKContextManager,
|
||||
LosslessProperties losslessProperties, Registration registration, Integer port) {
|
||||
this.polarisSDKContextManager = polarisSDKContextManager;
|
||||
this.losslessProperties = losslessProperties;
|
||||
this.registration = registration;
|
||||
this.port = port;
|
||||
}
|
||||
|
||||
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
|
||||
if (bean instanceof AbstractAutoServiceRegistration) {
|
||||
wrap(bean, polarisSDKContextManager, losslessProperties, registration, port);
|
||||
}
|
||||
return bean;
|
||||
}
|
||||
|
||||
public static void wrap(Object bean, PolarisSDKContextManager polarisSDKContextManager,
|
||||
LosslessProperties losslessProperties, Registration registration, Integer port) {
|
||||
LosslessProxyServiceRegistry proxyServiceRegistry;
|
||||
String clsName = bean.getClass().getCanonicalName();
|
||||
if (clsName.contains("org.springframework.cloud.client.serviceregistry.AbstractAutoServiceRegistration")) {
|
||||
ServiceRegistry<Registration> registry =
|
||||
(ServiceRegistry<Registration>) ReflectionUtils.
|
||||
getObjectByFieldName(bean, "serviceRegistry");
|
||||
proxyServiceRegistry = new LosslessProxyServiceRegistry(registry, polarisSDKContextManager, registration);
|
||||
ReflectionUtils.setValueByFieldName(bean, "serviceRegistry", proxyServiceRegistry);
|
||||
}
|
||||
else {
|
||||
ServiceRegistry<Registration> registry =
|
||||
(ServiceRegistry<Registration>) ReflectionUtils.
|
||||
getSuperObjectByFieldName(bean, "serviceRegistry");
|
||||
proxyServiceRegistry = new LosslessProxyServiceRegistry(registry, polarisSDKContextManager, registration);
|
||||
ReflectionUtils.setSuperValueByFieldName(bean, "serviceRegistry", proxyServiceRegistry);
|
||||
}
|
||||
SpringCloudLosslessActionProvider losslessActionProvider =
|
||||
new SpringCloudLosslessActionProvider(proxyServiceRegistry, losslessProperties);
|
||||
polarisSDKContextManager.getLosslessAPI().setLosslessActionProvider(
|
||||
getBaseInstance(registration, port), losslessActionProvider);
|
||||
}
|
||||
|
||||
public static BaseInstance getBaseInstance(Registration registration, Integer port) {
|
||||
// registration 通用，不设置 ns
|
||||
DefaultBaseInstance baseInstance = new DefaultBaseInstance();
|
||||
baseInstance.setService(registration.getServiceId());
|
||||
// 由于 PolarisRegistration 的 port 在 web 启动后才能生成，需从外部传入
|
||||
baseInstance.setPort(port);
|
||||
baseInstance.setHost(registration.getHost());
|
||||
return baseInstance;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,93 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless;
|
||||
|
||||
import java.util.concurrent.atomic.AtomicBoolean;
|
||||
|
||||
import com.tencent.cloud.polaris.context.PolarisSDKContextManager;
|
||||
|
||||
import org.springframework.cloud.client.serviceregistry.Registration;
|
||||
import org.springframework.cloud.client.serviceregistry.ServiceRegistry;
|
||||
|
||||
/**
|
||||
* Lossless proxy for {@link ServiceRegistry}.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
public class LosslessProxyServiceRegistry implements ServiceRegistry<Registration> {
|
||||
|
||||
private final ServiceRegistry<Registration> target;
|
||||
|
||||
private Registration registration;
|
||||
|
||||
private PolarisSDKContextManager polarisSDKContextManager;
|
||||
|
||||
private final AtomicBoolean doneDeregister = new AtomicBoolean(false);
|
||||
|
||||
public LosslessProxyServiceRegistry(ServiceRegistry<Registration> target,
|
||||
PolarisSDKContextManager polarisSDKContextManager, Registration registration) {
|
||||
this.target = target;
|
||||
this.polarisSDKContextManager = polarisSDKContextManager;
|
||||
this.registration = registration;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void register(Registration registration) {
|
||||
this.registration = registration;
|
||||
// web started, get port from registration
|
||||
polarisSDKContextManager.getLosslessAPI().losslessRegister(
|
||||
LosslessBeanPostProcessor.getBaseInstance(registration, registration.getPort()));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deregister(Registration registration) {
|
||||
if (doneDeregister.compareAndSet(false, true)) {
|
||||
target.deregister(registration);
|
||||
}
|
||||
}
|
||||
|
||||
public void deregister() {
|
||||
// 需要兼容其他 discovery, spring cloud deregister 统一幂等处理
|
||||
if (registration != null && doneDeregister.compareAndSet(false, true)) {
|
||||
target.deregister(registration);
|
||||
}
|
||||
}
|
||||
|
||||
public ServiceRegistry<Registration> getTarget() {
|
||||
return target;
|
||||
}
|
||||
|
||||
public Registration getRegistration() {
|
||||
return registration;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() {
|
||||
target.close();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setStatus(Registration registration, String status) {
|
||||
target.setStatus(registration, status);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> T getStatus(Registration registration) {
|
||||
return target.getStatus(registration);
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,89 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
import com.tencent.cloud.plugin.lossless.config.LosslessProperties;
|
||||
import com.tencent.cloud.polaris.util.OkHttpUtil;
|
||||
import com.tencent.polaris.api.plugin.lossless.InstanceProperties;
|
||||
import com.tencent.polaris.api.plugin.lossless.LosslessActionProvider;
|
||||
import com.tencent.polaris.api.utils.StringUtils;
|
||||
|
||||
import org.springframework.cloud.client.serviceregistry.Registration;
|
||||
import org.springframework.http.HttpHeaders;
|
||||
|
||||
/**
|
||||
* LosslessActionProvider for Spring Cloud.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
public class SpringCloudLosslessActionProvider implements LosslessActionProvider {
|
||||
|
||||
private LosslessProxyServiceRegistry losslessProxyServiceRegistry;
|
||||
|
||||
private LosslessProperties losslessProperties;
|
||||
|
||||
private Consumer<Registration> registrationConsumer;
|
||||
|
||||
private Registration registration;
|
||||
|
||||
public SpringCloudLosslessActionProvider(
|
||||
LosslessProxyServiceRegistry losslessProxyServiceRegistry,
|
||||
LosslessProperties losslessProperties) {
|
||||
this.losslessProxyServiceRegistry = losslessProxyServiceRegistry;
|
||||
this.losslessProperties = losslessProperties;
|
||||
this.registrationConsumer = losslessProxyServiceRegistry.getTarget()::register;
|
||||
this.registration = losslessProxyServiceRegistry.getRegistration();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "spring-cloud";
|
||||
}
|
||||
|
||||
@Override
|
||||
public void doRegister(InstanceProperties instanceProperties) {
|
||||
registrationConsumer.accept(registration);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void doDeregister() {
|
||||
losslessProxyServiceRegistry.deregister();
|
||||
}
|
||||
|
||||
/**
|
||||
* Check whether health check is enable.
|
||||
* @return true: register after passing doHealthCheck, false: register after delayRegisterInterval.
|
||||
*/
|
||||
@Override
|
||||
public boolean isEnableHealthCheck() {
|
||||
return StringUtils.isNotBlank(losslessProperties.getHealthCheckPath());
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean doHealthCheck() {
|
||||
Map<String, String> headers = new HashMap<>(1);
|
||||
headers.put(HttpHeaders.USER_AGENT, "polaris");
|
||||
|
||||
return OkHttpUtil.checkUrl("localhost", registration.getPort(),
|
||||
losslessProperties.getHealthCheckPath(), headers);
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless.config;
|
||||
|
||||
import com.tencent.cloud.plugin.lossless.LosslessBeanPostProcessor;
|
||||
import com.tencent.cloud.polaris.context.PolarisSDKContextManager;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
||||
import org.springframework.cloud.client.serviceregistry.Registration;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.context.annotation.Import;
|
||||
|
||||
/**
|
||||
* Autoconfiguration of lossless.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
@Import(LosslessPropertiesAutoConfiguration.class)
|
||||
public class LosslessAutoConfiguration {
|
||||
|
||||
@Value("${server.port:8080}")
|
||||
private Integer port;
|
||||
|
||||
@Bean
|
||||
@ConditionalOnMissingBean
|
||||
public LosslessBeanPostProcessor losslessBeanPostProcessor(Registration registration,
|
||||
PolarisSDKContextManager polarisSDKContextManager,
|
||||
LosslessProperties losslessProperties) {
|
||||
return new LosslessBeanPostProcessor(polarisSDKContextManager, losslessProperties,
|
||||
registration, port);
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,59 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless.config;
|
||||
|
||||
import java.util.Objects;
|
||||
|
||||
import com.tencent.cloud.common.constant.OrderConstant.Modifier;
|
||||
import com.tencent.cloud.polaris.context.PolarisConfigModifier;
|
||||
import com.tencent.polaris.factory.config.ConfigurationImpl;
|
||||
import com.tencent.polaris.factory.config.provider.LosslessConfigImpl;
|
||||
|
||||
/**
|
||||
* Config modifier for lossless.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
public class LosslessConfigModifier implements PolarisConfigModifier {
|
||||
|
||||
private final LosslessProperties losslessProperties;
|
||||
|
||||
public LosslessConfigModifier(LosslessProperties losslessProperties) {
|
||||
this.losslessProperties = losslessProperties;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void modify(ConfigurationImpl configuration) {
|
||||
if (losslessProperties.isEnabled()) {
|
||||
LosslessConfigImpl losslessConfig = (LosslessConfigImpl) configuration.getProvider().getLossless();
|
||||
losslessConfig.setEnable(true);
|
||||
losslessConfig.setPort(losslessProperties.getPort());
|
||||
if (Objects.nonNull(losslessProperties.getDelayRegisterInterval())) {
|
||||
losslessConfig.setDelayRegisterInterval(losslessProperties.getDelayRegisterInterval());
|
||||
}
|
||||
if (Objects.nonNull(losslessProperties.getHealthCheckInterval())) {
|
||||
losslessConfig.setHealthCheckInterval(losslessProperties.getHealthCheckInterval());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public int getOrder() {
|
||||
return Modifier.LOSSLESS_ORDER;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,75 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless.config;
|
||||
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
|
||||
@ConfigurationProperties("spring.cloud.polaris.lossless")
|
||||
public class LosslessProperties {
|
||||
|
||||
private boolean enabled = true;
|
||||
|
||||
private int port = 28080;
|
||||
|
||||
private String healthCheckPath;
|
||||
|
||||
private Long delayRegisterInterval;
|
||||
|
||||
private Long healthCheckInterval;
|
||||
|
||||
public boolean isEnabled() {
|
||||
return enabled;
|
||||
}
|
||||
|
||||
public void setEnabled(boolean enabled) {
|
||||
this.enabled = enabled;
|
||||
}
|
||||
|
||||
public int getPort() {
|
||||
return port;
|
||||
}
|
||||
|
||||
public void setPort(int port) {
|
||||
this.port = port;
|
||||
}
|
||||
|
||||
public String getHealthCheckPath() {
|
||||
return healthCheckPath;
|
||||
}
|
||||
|
||||
public void setHealthCheckPath(String healthCheckPath) {
|
||||
this.healthCheckPath = healthCheckPath;
|
||||
}
|
||||
|
||||
public Long getDelayRegisterInterval() {
|
||||
return delayRegisterInterval;
|
||||
}
|
||||
|
||||
public void setDelayRegisterInterval(Long delayRegisterInterval) {
|
||||
this.delayRegisterInterval = delayRegisterInterval;
|
||||
}
|
||||
|
||||
public Long getHealthCheckInterval() {
|
||||
return healthCheckInterval;
|
||||
}
|
||||
|
||||
public void setHealthCheckInterval(Long healthCheckInterval) {
|
||||
this.healthCheckInterval = healthCheckInterval;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless.config;
|
||||
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
||||
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
/**
|
||||
* Autoconfiguration of lossless properties.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
@EnableConfigurationProperties(LosslessProperties.class)
|
||||
public class LosslessPropertiesAutoConfiguration {
|
||||
|
||||
@Bean
|
||||
@ConditionalOnMissingBean
|
||||
public LosslessConfigModifier losslessConfigModifier(LosslessProperties losslessProperties) {
|
||||
return new LosslessConfigModifier(losslessProperties);
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,33 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless.config;
|
||||
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.context.annotation.Import;
|
||||
|
||||
|
||||
/**
|
||||
* BootstrapConfiguration of lossless properties.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
@Import(LosslessPropertiesAutoConfiguration.class)
|
||||
public class LosslessPropertiesBootstrapConfiguration {
|
||||
|
||||
}
|
||||
@ -0,0 +1,10 @@
|
||||
{
|
||||
"properties": [
|
||||
{
|
||||
"name": "spring.cloud.polaris.lossless.enabled",
|
||||
"type": "java.lang.Boolean",
|
||||
"defaultValue": true,
|
||||
"description": "the switch for lossless plugin."
|
||||
}
|
||||
]
|
||||
}
|
||||
@ -0,0 +1,4 @@
|
||||
org.springframework.cloud.bootstrap.BootstrapConfiguration=\
|
||||
com.tencent.cloud.plugin.lossless.config.LosslessPropertiesBootstrapConfiguration
|
||||
org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
|
||||
com.tencent.cloud.plugin.lossless.config.LosslessAutoConfiguration
|
||||
@ -0,0 +1,89 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless;
|
||||
|
||||
import com.tencent.cloud.plugin.lossless.config.LosslessConfigModifier;
|
||||
import com.tencent.cloud.polaris.context.PolarisSDKContextManager;
|
||||
import com.tencent.polaris.api.config.provider.LosslessConfig;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
|
||||
/**
|
||||
* Test for {@link LosslessConfigModifier}.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
public class LosslessConfigModifierTest {
|
||||
|
||||
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
||||
.withConfiguration(AutoConfigurations.of(TestApplication.class))
|
||||
.withPropertyValues("spring.cloud.polaris.enabled=true")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.enabled=true")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.port=20000")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.healthCheckPath=/xxx")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.delayRegisterInterval=10")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.healthCheckInterval=5")
|
||||
.withPropertyValues("spring.application.name=test")
|
||||
.withPropertyValues("spring.cloud.gateway.enabled=false");
|
||||
private final ApplicationContextRunner disabledContextRunner = new ApplicationContextRunner()
|
||||
.withConfiguration(AutoConfigurations.of(TestApplication.class))
|
||||
.withPropertyValues("spring.cloud.polaris.enabled=true")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.enabled=false")
|
||||
.withPropertyValues("spring.application.name=test")
|
||||
.withPropertyValues("spring.cloud.gateway.enabled=false");
|
||||
|
||||
@BeforeEach
|
||||
void setUp() {
|
||||
PolarisSDKContextManager.innerDestroy();
|
||||
}
|
||||
|
||||
@Test
|
||||
void testModify() {
|
||||
contextRunner.run(context -> {
|
||||
PolarisSDKContextManager polarisSDKContextManager = context.getBean(PolarisSDKContextManager.class);
|
||||
LosslessConfig losslessConfig = polarisSDKContextManager.getSDKContext().
|
||||
getConfig().getProvider().getLossless();
|
||||
assertThat(losslessConfig.getHost()).isEqualTo("0.0.0.0");
|
||||
assertThat(losslessConfig.getPort()).isEqualTo(20000);
|
||||
assertThat(losslessConfig.getDelayRegisterInterval()).isEqualTo(10);
|
||||
assertThat(losslessConfig.getHealthCheckInterval()).isEqualTo(5);
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
void testDisabled() {
|
||||
disabledContextRunner.run(context -> {
|
||||
PolarisSDKContextManager polarisSDKContextManager = context.getBean(PolarisSDKContextManager.class);
|
||||
LosslessConfig losslessConfig = polarisSDKContextManager.getSDKContext().
|
||||
getConfig().getProvider().getLossless();
|
||||
assertThat(losslessConfig.isEnable()).isFalse();
|
||||
});
|
||||
}
|
||||
|
||||
@SpringBootApplication
|
||||
protected static class TestApplication {
|
||||
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless;
|
||||
|
||||
import com.tencent.cloud.plugin.lossless.config.LosslessProperties;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
|
||||
/**
|
||||
* Test for {@link LosslessProperties}.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
public class LosslessPropertiesTest {
|
||||
|
||||
@Test
|
||||
void testGetAndSet() {
|
||||
LosslessProperties polarisStatProperties = new LosslessProperties();
|
||||
|
||||
// healthCheckPath
|
||||
polarisStatProperties.setHealthCheckPath("/xxx");
|
||||
assertThat(polarisStatProperties.getHealthCheckPath()).isEqualTo("/xxx");
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,190 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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.tencent.cloud.plugin.lossless;
|
||||
|
||||
import java.util.Collections;
|
||||
|
||||
import com.tencent.cloud.plugin.lossless.config.LosslessAutoConfiguration;
|
||||
import com.tencent.cloud.plugin.lossless.config.LosslessPropertiesBootstrapConfiguration;
|
||||
import com.tencent.cloud.polaris.context.PolarisSDKContextManager;
|
||||
import com.tencent.cloud.polaris.context.config.PolarisContextAutoConfiguration;
|
||||
import com.tencent.cloud.polaris.discovery.PolarisDiscoveryAutoConfiguration;
|
||||
import com.tencent.cloud.polaris.discovery.PolarisDiscoveryClientConfiguration;
|
||||
import com.tencent.cloud.polaris.registry.PolarisRegistration;
|
||||
import com.tencent.cloud.polaris.registry.PolarisServiceRegistry;
|
||||
import com.tencent.cloud.polaris.util.OkHttpUtil;
|
||||
import com.tencent.polaris.api.pojo.ServiceKey;
|
||||
import com.tencent.polaris.test.mock.discovery.NamingServer;
|
||||
import org.junit.jupiter.api.AfterAll;
|
||||
import org.junit.jupiter.api.BeforeAll;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
|
||||
import org.springframework.boot.test.context.runner.WebApplicationContextRunner;
|
||||
import org.springframework.cloud.client.serviceregistry.AbstractAutoServiceRegistration;
|
||||
import org.springframework.cloud.client.serviceregistry.AutoServiceRegistrationUtils;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.assertThatCode;
|
||||
|
||||
/**
|
||||
* Test for {@link LosslessProxyServiceRegistry}.
|
||||
*
|
||||
* @author Shedfree Wu
|
||||
*/
|
||||
public class LosslessServiceRegistryTest {
|
||||
|
||||
private static String NAMESPACE_TEST = "Test";
|
||||
|
||||
private static String SERVICE_PROVIDER = "java_provider_test";
|
||||
|
||||
private static String HOST = "127.0.0.1";
|
||||
|
||||
private static int APPLICATION_PORT = 19091;
|
||||
|
||||
private static int LOSSLESS_PORT_1 = 28081;
|
||||
|
||||
private static NamingServer namingServer;
|
||||
|
||||
private final WebApplicationContextRunner contextRunner = new WebApplicationContextRunner()
|
||||
.withConfiguration(AutoConfigurations.of(
|
||||
LosslessAutoConfiguration.class,
|
||||
LosslessPropertiesBootstrapConfiguration.class,
|
||||
PolarisContextAutoConfiguration.class,
|
||||
PolarisPropertiesConfiguration.class,
|
||||
PolarisDiscoveryClientConfiguration.class,
|
||||
PolarisDiscoveryAutoConfiguration.class))
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.delayRegisterInterval=5000")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.healthCheckPath=")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.port=" + LOSSLESS_PORT_1)
|
||||
.withPropertyValues("spring.application.name=" + SERVICE_PROVIDER)
|
||||
.withPropertyValues("server.port=" + APPLICATION_PORT)
|
||||
.withPropertyValues("spring.cloud.polaris.localIpAddress=" + HOST)
|
||||
.withPropertyValues("spring.cloud.polaris.localPort=" + APPLICATION_PORT)
|
||||
.withPropertyValues("spring.cloud.polaris.address=grpc://127.0.0.1:10081")
|
||||
.withPropertyValues("spring.cloud.polaris.discovery.namespace=" + NAMESPACE_TEST)
|
||||
.withPropertyValues("spring.cloud.polaris.discovery.token=xxxxxx");
|
||||
|
||||
private final WebApplicationContextRunner contextRunner2 = new WebApplicationContextRunner()
|
||||
.withConfiguration(AutoConfigurations.of(
|
||||
LosslessAutoConfiguration.class,
|
||||
LosslessPropertiesBootstrapConfiguration.class,
|
||||
PolarisContextAutoConfiguration.class,
|
||||
PolarisPropertiesConfiguration.class,
|
||||
PolarisDiscoveryClientConfiguration.class,
|
||||
PolarisDiscoveryAutoConfiguration.class))
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.healthCheckInterval=1000")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.healthCheckPath=/test")
|
||||
.withPropertyValues("spring.cloud.polaris.lossless.port=28082")
|
||||
.withPropertyValues("spring.application.name=" + SERVICE_PROVIDER)
|
||||
.withPropertyValues("server.port=" + APPLICATION_PORT)
|
||||
.withPropertyValues("spring.cloud.polaris.localIpAddress=" + HOST)
|
||||
.withPropertyValues("spring.cloud.polaris.localPort=" + APPLICATION_PORT)
|
||||
.withPropertyValues("spring.cloud.polaris.address=grpc://127.0.0.1:10081")
|
||||
.withPropertyValues("spring.cloud.polaris.discovery.namespace=" + NAMESPACE_TEST)
|
||||
.withPropertyValues("spring.cloud.polaris.discovery.token=xxxxxx");
|
||||
|
||||
@BeforeAll
|
||||
static void beforeAll() throws Exception {
|
||||
namingServer = NamingServer.startNamingServer(10081);
|
||||
|
||||
// add service
|
||||
namingServer.getNamingService().addService(new ServiceKey(NAMESPACE_TEST, SERVICE_PROVIDER));
|
||||
}
|
||||
|
||||
@AfterAll
|
||||
static void afterAll() {
|
||||
if (null != namingServer) {
|
||||
namingServer.terminate();
|
||||
}
|
||||
}
|
||||
|
||||
@BeforeEach
|
||||
void setUp() {
|
||||
PolarisSDKContextManager.innerDestroy();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRegister() {
|
||||
this.contextRunner.run(context -> {
|
||||
|
||||
AbstractAutoServiceRegistration autoServiceRegistration = context.getBean(AbstractAutoServiceRegistration.class);
|
||||
|
||||
assertThatCode(() -> {
|
||||
AutoServiceRegistrationUtils.register(autoServiceRegistration);
|
||||
}).doesNotThrowAnyException();
|
||||
Thread.sleep(1000);
|
||||
// before register online status is false
|
||||
assertThatCode(() -> {
|
||||
assertThat(OkHttpUtil.checkUrl(HOST, LOSSLESS_PORT_1, "/online", Collections.EMPTY_MAP)).isFalse();
|
||||
}).doesNotThrowAnyException();
|
||||
// delay register after 5s
|
||||
Thread.sleep(5000);
|
||||
PolarisServiceRegistry registry = context.getBean(PolarisServiceRegistry.class);
|
||||
PolarisRegistration registration = context.getBean(PolarisRegistration.class);
|
||||
|
||||
assertThatCode(() -> {
|
||||
assertThat(registry.getStatus(registration)).isEqualTo("DOWN");
|
||||
}).doesNotThrowAnyException();
|
||||
|
||||
assertThatCode(() -> {
|
||||
assertThat(OkHttpUtil.checkUrl(HOST, LOSSLESS_PORT_1, "/online", Collections.EMPTY_MAP)).isTrue();
|
||||
}).doesNotThrowAnyException();
|
||||
|
||||
assertThatCode(() -> {
|
||||
assertThat(OkHttpUtil.checkUrl(HOST, LOSSLESS_PORT_1, "/offline", Collections.EMPTY_MAP)).isTrue();
|
||||
}).doesNotThrowAnyException();
|
||||
|
||||
assertThatCode(() -> {
|
||||
AutoServiceRegistrationUtils.deRegister(autoServiceRegistration);
|
||||
}).doesNotThrowAnyException();
|
||||
|
||||
assertThatCode(() -> {
|
||||
assertThat(registry.getStatus(registration)).isEqualTo("DOWN");
|
||||
}).doesNotThrowAnyException();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRegister2() {
|
||||
this.contextRunner2.run(context -> {
|
||||
|
||||
AbstractAutoServiceRegistration autoServiceRegistration = context.getBean(AbstractAutoServiceRegistration.class);
|
||||
|
||||
assertThatCode(() -> {
|
||||
AutoServiceRegistrationUtils.register(autoServiceRegistration);
|
||||
}).doesNotThrowAnyException();
|
||||
|
||||
Thread.sleep(2000);
|
||||
|
||||
assertThatCode(() -> {
|
||||
AutoServiceRegistrationUtils.deRegister(autoServiceRegistration);
|
||||
}).doesNotThrowAnyException();
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
@Configuration
|
||||
@EnableAutoConfiguration
|
||||
static class PolarisPropertiesConfiguration {
|
||||
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,29 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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 org.springframework.cloud.client.serviceregistry;
|
||||
|
||||
public class AutoServiceRegistrationUtils {
|
||||
|
||||
public static void register(AbstractAutoServiceRegistration autoServiceRegistration) {
|
||||
autoServiceRegistration.register();
|
||||
}
|
||||
|
||||
public static void deRegister(AbstractAutoServiceRegistration autoServiceRegistration) {
|
||||
autoServiceRegistration.deregister();
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue