From 11a10ed6c6823c22af7e9ee9570ca4a3f43d9d5c Mon Sep 17 00:00:00 2001
From: seanyu <seansyyu@tencent.com>
Date: Fri, 21 Apr 2023 20:17:15 +0800
Subject: [PATCH] add InstanceId

---
 .../cloud/polaris/PolarisDiscoveryProperties.java  | 14 ++++++++++++++
 .../polaris/registry/PolarisServiceRegistry.java   |  1 +
 2 files changed, 15 insertions(+)

diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/PolarisDiscoveryProperties.java b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/PolarisDiscoveryProperties.java
index ea480d406..3f83bfd44 100644
--- a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/PolarisDiscoveryProperties.java
+++ b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/PolarisDiscoveryProperties.java
@@ -45,6 +45,11 @@ public class PolarisDiscoveryProperties {
 	@Value("${spring.cloud.polaris.discovery.service:${spring.cloud.polaris.service:${spring.application.name:}}}")
 	private String service;
 
+	/**
+	 * Service instance id.
+	 */
+	private String instanceId;
+
 	/**
 	 * The polaris authentication token.
 	 */
@@ -96,6 +101,14 @@ public class PolarisDiscoveryProperties {
 	 */
 	private Long serviceListRefreshInterval = 60000L;
 
+	public String getInstanceId() {
+		return instanceId;
+	}
+
+	public void setInstanceId(String instanceId) {
+		this.instanceId = instanceId;
+	}
+
 	public String getNamespace() {
 		return namespace;
 	}
@@ -204,6 +217,7 @@ public class PolarisDiscoveryProperties {
 		return "PolarisDiscoveryProperties{" +
 				"namespace='" + namespace + '\'' +
 				", service='" + service + '\'' +
+				", instanceId='" + instanceId + '\'' +
 				", token='" + token + '\'' +
 				", weight=" + weight +
 				", version='" + version + '\'' +
diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java
index 8a4177026..63e340386 100644
--- a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java
+++ b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java
@@ -114,6 +114,7 @@ public class PolarisServiceRegistry implements ServiceRegistry<PolarisRegistrati
 		instanceRegisterRequest.setMetadata(registration.getMetadata());
 		instanceRegisterRequest.setProtocol(polarisDiscoveryProperties.getProtocol());
 		instanceRegisterRequest.setVersion(polarisDiscoveryProperties.getVersion());
+		instanceRegisterRequest.setInstanceId(polarisDiscoveryProperties.getInstanceId());
 		try {
 			ProviderAPI providerClient = polarisDiscoveryHandler.getProviderAPI();
 			InstanceRegisterResponse instanceRegisterResponse;