diff --git a/CHANGELOG.md b/CHANGELOG.md index b014e5a4c..3e57251d0 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -5,3 +5,4 @@ - [feat:support webclient and gateway report call metrics](https://github.com/Tencent/spring-cloud-tencent/pull/942) - [docs:prevent the release of the final version of the sdk.](https://github.com/Tencent/spring-cloud-tencent/pull/945) - [fix: fix nacos CircuitBreaker disable bug.](https://github.com/Tencent/spring-cloud-tencent/pull/947) +- [feature: add config for customized local port.](https://github.com/Tencent/spring-cloud-tencent/pull/956) diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisRegistration.java b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisRegistration.java index 7a5fb4609..1c9d902f9 100644 --- a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisRegistration.java +++ b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisRegistration.java @@ -25,6 +25,7 @@ import java.util.Objects; import com.tencent.cloud.common.metadata.StaticMetadataManager; import com.tencent.cloud.polaris.PolarisDiscoveryProperties; +import com.tencent.cloud.polaris.context.config.PolarisContextProperties; import com.tencent.cloud.polaris.extend.consul.ConsulContextProperties; import com.tencent.cloud.polaris.extend.nacos.NacosContextProperties; import com.tencent.polaris.client.api.SDKContext; @@ -76,6 +77,7 @@ public class PolarisRegistration implements Registration { public PolarisRegistration( PolarisDiscoveryProperties polarisDiscoveryProperties, + @Nullable PolarisContextProperties polarisContextProperties, @Nullable ConsulContextProperties consulContextProperties, SDKContext context, StaticMetadataManager staticMetadataManager, @Nullable NacosContextProperties nacosContextProperties, @@ -89,6 +91,9 @@ public class PolarisRegistration implements Registration { this.servletWebServerApplicationContext = servletWebServerApplicationContext; this.reactiveWebServerApplicationContext = reactiveWebServerApplicationContext; host = polarisContext.getConfig().getGlobal().getAPI().getBindIP(); + if (polarisContextProperties != null) { + port = polarisContextProperties.getLocalPort(); + } } @Override diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java index a421ec307..be5d244f9 100644 --- a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java +++ b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java @@ -20,6 +20,7 @@ package com.tencent.cloud.polaris.registry; import com.tencent.cloud.common.metadata.StaticMetadataManager; import com.tencent.cloud.polaris.PolarisDiscoveryProperties; +import com.tencent.cloud.polaris.context.config.PolarisContextProperties; import com.tencent.cloud.polaris.discovery.PolarisDiscoveryAutoConfiguration; import com.tencent.cloud.polaris.discovery.PolarisDiscoveryHandler; import com.tencent.cloud.polaris.extend.consul.ConsulContextProperties; @@ -65,11 +66,12 @@ public class PolarisServiceRegistryAutoConfiguration { @ConditionalOnBean(AutoServiceRegistrationProperties.class) public PolarisRegistration polarisRegistration( PolarisDiscoveryProperties polarisDiscoveryProperties, + PolarisContextProperties polarisContextProperties, @Autowired(required = false) ConsulContextProperties consulContextProperties, SDKContext context, StaticMetadataManager staticMetadataManager, NacosContextProperties nacosContextProperties, @Autowired(required = false) ServletWebServerApplicationContext servletWebServerApplicationContext, @Autowired(required = false) ReactiveWebServerApplicationContext reactiveWebServerApplicationContext) { - return new PolarisRegistration(polarisDiscoveryProperties, consulContextProperties, context, + return new PolarisRegistration(polarisDiscoveryProperties, polarisContextProperties, consulContextProperties, context, staticMetadataManager, nacosContextProperties, servletWebServerApplicationContext, reactiveWebServerApplicationContext); } diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisRegistrationTest.java b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisRegistrationTest.java index ebcc5ea4b..742ce4052 100644 --- a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisRegistrationTest.java +++ b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisRegistrationTest.java @@ -22,6 +22,7 @@ import java.util.Map; import com.tencent.cloud.common.metadata.StaticMetadataManager; import com.tencent.cloud.polaris.PolarisDiscoveryProperties; +import com.tencent.cloud.polaris.context.config.PolarisContextProperties; import com.tencent.cloud.polaris.extend.consul.ConsulContextProperties; import com.tencent.cloud.polaris.extend.nacos.NacosContextProperties; import com.tencent.polaris.api.config.Configuration; @@ -62,6 +63,10 @@ public class PolarisRegistrationTest { private PolarisRegistration polarisRegistration3; + private PolarisRegistration polarisRegistration4; + + private static int testLocalPort = 10086; + @BeforeEach void setUp() { // mock PolarisDiscoveryProperties @@ -70,23 +75,27 @@ public class PolarisRegistrationTest { doReturn("http").when(polarisDiscoveryProperties).getProtocol(); doReturn(true).when(polarisDiscoveryProperties).isRegisterEnabled(); + // mock PolarisContextProperties + PolarisContextProperties polarisContextProperties = mock(PolarisContextProperties.class); + doReturn(testLocalPort).when(polarisContextProperties).getLocalPort(); + // mock ConsulContextProperties ConsulContextProperties consulContextProperties = mock(ConsulContextProperties.class); doReturn(true).when(consulContextProperties).isEnabled(); doReturn(true).when(consulContextProperties).isRegister(); // mock NacosContextProperties - nacosContextProperties = new NacosContextProperties(); - nacosContextProperties.setEnabled(true); - nacosContextProperties.setRegisterEnabled(true); - nacosContextProperties.setContextPath("/"); - nacosContextProperties.setClusterName("cluster"); - nacosContextProperties.setGroup(""); - nacosContextProperties.setDiscoveryEnabled(true); - nacosContextProperties.setPassword(""); - nacosContextProperties.setUsername(""); - nacosContextProperties.setServerAddr(""); - nacosContextProperties.toString(); + // mock NacosContextProperties + nacosContextProperties = mock(NacosContextProperties.class); + doReturn(true).when(nacosContextProperties).isEnabled(); + doReturn(true).when(nacosContextProperties).isRegisterEnabled(); + doReturn("/").when(nacosContextProperties).getContextPath(); + doReturn("cluster").when(nacosContextProperties).getClusterName(); + doReturn("").when(nacosContextProperties).getGroup(); + doReturn(true).when(nacosContextProperties).isDiscoveryEnabled(); + doReturn("").when(nacosContextProperties).getPassword(); + doReturn("").when(nacosContextProperties).getUsername(); + doReturn("").when(nacosContextProperties).getServerAddr(); // mock SDKContext APIConfig apiConfig = mock(APIConfig.class); @@ -114,15 +123,19 @@ public class PolarisRegistrationTest { ReactiveWebServerApplicationContext reactiveWebServerApplicationContext = mock(ReactiveWebServerApplicationContext.class); doReturn(reactiveWebServer).when(reactiveWebServerApplicationContext).getWebServer(); - polarisRegistration1 = new PolarisRegistration(polarisDiscoveryProperties, consulContextProperties, + polarisRegistration1 = new PolarisRegistration(polarisDiscoveryProperties, null, consulContextProperties, polarisContext, staticMetadataManager, nacosContextProperties, servletWebServerApplicationContext, null); - polarisRegistration2 = new PolarisRegistration(polarisDiscoveryProperties, consulContextProperties, + polarisRegistration2 = new PolarisRegistration(polarisDiscoveryProperties, null, consulContextProperties, polarisContext, staticMetadataManager, nacosContextProperties, null, reactiveWebServerApplicationContext); - polarisRegistration3 = new PolarisRegistration(polarisDiscoveryProperties, consulContextProperties, + polarisRegistration3 = new PolarisRegistration(polarisDiscoveryProperties, null, consulContextProperties, + polarisContext, staticMetadataManager, nacosContextProperties, + null, null); + + polarisRegistration4 = new PolarisRegistration(polarisDiscoveryProperties, polarisContextProperties, consulContextProperties, polarisContext, staticMetadataManager, nacosContextProperties, null, null); } @@ -147,6 +160,7 @@ public class PolarisRegistrationTest { catch (RuntimeException e) { assertThat(e.getMessage()).isEqualTo("Unsupported web type."); } + assertThat(polarisRegistration4.getPort()).isEqualTo(testLocalPort); } @Test diff --git a/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextAutoConfiguration.java b/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextAutoConfiguration.java index d51cf35a4..298d89fa4 100644 --- a/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextAutoConfiguration.java +++ b/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextAutoConfiguration.java @@ -52,7 +52,8 @@ public class PolarisContextAutoConfiguration { @ConditionalOnMissingBean public SDKContext polarisContext(PolarisContextProperties properties, Environment environment, List modifierList) throws PolarisException { return SDKContext.initContextByConfig(properties.configuration(modifierList, - () -> environment.getProperty("spring.cloud.client.ip-address"))); + () -> environment.getProperty("spring.cloud.client.ip-address"), + () -> environment.getProperty("spring.cloud.polaris.local-port", Integer.class, 0))); } @Bean diff --git a/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextProperties.java b/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextProperties.java index 696218a28..f77eaa4f5 100644 --- a/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextProperties.java +++ b/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/config/PolarisContextProperties.java @@ -52,6 +52,11 @@ public class PolarisContextProperties { */ private String localIpAddress; + /** + * current server local port. + */ + private Integer localPort; + /** * If polaris enabled. */ @@ -67,7 +72,7 @@ public class PolarisContextProperties { */ private String service; - public Configuration configuration(List modifierList, Supplier ipAddressSupplier) { + public Configuration configuration(List modifierList, Supplier ipAddressSupplier, Supplier portSupplier) { // 1. Read user-defined polaris.yml configuration ConfigurationImpl configuration = (ConfigurationImpl) ConfigAPIFactory .defaultConfig(ConfigProvider.DEFAULT_CONFIG); @@ -78,6 +83,9 @@ public class PolarisContextProperties { defaultHost = ipAddressSupplier.get(); this.localIpAddress = defaultHost; } + if (this.localPort == null || this.localPort <= 0) { + this.localPort = portSupplier.get(); + } configuration.getGlobal().getAPI().setBindIP(defaultHost); @@ -110,6 +118,14 @@ public class PolarisContextProperties { this.localIpAddress = localIpAddress; } + public Integer getLocalPort() { + return localPort; + } + + public void setLocalPort(Integer localPort) { + this.localPort = localPort; + } + public Boolean getEnabled() { return enabled; }