feature:add User-Agent:polaris for healthyCheck api. (#873)

Co-authored-by: lingxiao.wu <51630311+lingxiao-wu@users.noreply.github.com>

CHANGELOG.md

@@ -9,3 +9,4 @@
 - [feat:enable stat reporting as default.](https://github.com/Tencent/spring-cloud-tencent/pull/864)
 - [refactor:update to junit 5.](https://github.com/Tencent/spring-cloud-tencent/pull/867)
 - [docs:support auto snapshot release in GitHub Action.](https://github.com/Tencent/spring-cloud-tencent/pull/870)
+- [feature:add User-Agent:polaris for healthyCheck api.](https://github.com/Tencent/spring-cloud-tencent/pull/873)

spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java

@@ -18,6 +18,8 @@
 
 package com.tencent.cloud.polaris.registry;
 
+import java.util.HashMap;
+import java.util.Map;
 import java.util.Objects;
 import java.util.concurrent.Executors;
 import java.util.concurrent.ScheduledExecutorService;
@@ -45,6 +47,7 @@ import org.slf4j.LoggerFactory;
 
 import org.springframework.beans.BeanUtils;
 import org.springframework.cloud.client.serviceregistry.ServiceRegistry;
+import org.springframework.http.HttpHeaders;
 
 import static java.util.concurrent.TimeUnit.SECONDS;
 import static org.springframework.util.ReflectionUtils.rethrowRuntimeException;
@@ -224,7 +227,9 @@ public class PolarisServiceRegistry implements ServiceRegistry<PolarisRegistrati
 				String healthCheckUrl = String.format("http://%s:%s%s", heartbeatRequest.getHost(),
 						heartbeatRequest.getPort(), healthCheckEndpoint);
 
-				if (!OkHttpUtil.get(healthCheckUrl, null)) {
+				Map<String, String> headers = new HashMap<>(1);
+				headers.put(HttpHeaders.USER_AGENT, "polaris");
+				if (!OkHttpUtil.get(healthCheckUrl, headers)) {
 					LOGGER.error("backend service health check failed. health check endpoint = {}",
 							healthCheckEndpoint);
 					return;

spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/util/OkHttpUtil.java

@@ -27,6 +27,7 @@ import java.util.concurrent.TimeUnit;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 
 /**
@@ -60,6 +61,9 @@ public final class OkHttpUtil {
 			conn.setRequestMethod("GET");
 			conn.setConnectTimeout((int) TimeUnit.SECONDS.toMillis(2));
 			conn.setReadTimeout((int) TimeUnit.SECONDS.toMillis(2));
+			if (!CollectionUtils.isEmpty(headers)) {
+				headers.forEach(conn::setRequestProperty);
+			}
 			BufferedReader reader = new BufferedReader(new InputStreamReader(conn.getInputStream()));
 			StringBuffer buffer = new StringBuffer();
 			String str;

spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryTest.java

@@ -61,6 +61,19 @@ public class PolarisServiceRegistryTest {
 			.withPropertyValues("spring.cloud.polaris.discovery.namespace=" + NAMESPACE_TEST)
 			.withPropertyValues("spring.cloud.polaris.discovery.token=xxxxxx");
 
+	private final WebApplicationContextRunner contextRunner2 = new WebApplicationContextRunner()
+			.withConfiguration(AutoConfigurations.of(
+					PolarisContextAutoConfiguration.class,
+					PolarisPropertiesConfiguration.class,
+					PolarisDiscoveryClientConfiguration.class,
+					PolarisDiscoveryAutoConfiguration.class))
+			.withPropertyValues("spring.application.name=" + SERVICE_PROVIDER)
+			.withPropertyValues("server.port=" + 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")
+			.withPropertyValues("spring.cloud.polaris.discovery.health-check-url=/test");
+
 	@BeforeAll
 	static void beforeAll() throws Exception {
 		namingServer = NamingServer.startNamingServer(10081);
@@ -112,6 +125,23 @@ public class PolarisServiceRegistryTest {
 		});
 	}
 
+	@Test
+	void testHeartbeat() {
+		this.contextRunner2.run(context -> {
+			PolarisServiceRegistry registry = context.getBean(PolarisServiceRegistry.class);
+			PolarisRegistration registration = Mockito.mock(PolarisRegistration.class);
+			when(registration.getHost()).thenReturn("127.0.0.1");
+			when(registration.getPort()).thenReturn(8080);
+			when(registration.getServiceId()).thenReturn(SERVICE_PROVIDER);
+
+			assertThatCode(() -> registry.register(registration)).doesNotThrowAnyException();
+
+			Thread.sleep(6000);
+
+			assertThatCode(() -> registry.deregister(registration)).doesNotThrowAnyException();
+		});
+	}
+
 	@Configuration
 	@EnableAutoConfiguration
 	static class PolarisPropertiesConfiguration {

spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/util/OkHttpUtilTest.java

@@ -26,6 +26,7 @@ import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.web.server.LocalServerPort;
 import org.springframework.test.context.junit.jupiter.SpringExtension;
 import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestHeader;
 import org.springframework.web.bind.annotation.RestController;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -53,8 +54,8 @@ public class OkHttpUtilTest {
 	@RestController
 	static class TestApplication {
 		@GetMapping("/test")
-		public String test() {
-			return "test";
+		public String test(@RequestHeader("key") String key) {
+			return key;
 		}
 	}
 }