Use lock for instance status update.

Netflix · Dec 23, 2024 · f7d808d · f7d808d
1 parent f4c8ed1
commit f7d808d
Showing 1 changed file with 21 additions and 8 deletions.
diff --git a/eureka-client/src/main/java/com/netflix/discovery/DiscoveryClient.java b/eureka-client/src/main/java/com/netflix/discovery/DiscoveryClient.java
@@ -106,6 +106,7 @@
  *
  * @author Karthik Ranganathan, Greg Kim
  * @author Spencer Gibb
+ * @author Olga Maciaszek-Sharma
  *
  */
 @Singleton
@@ -1381,15 +1382,27 @@ void refreshInstanceInfo() {
         applicationInfoManager.refreshLeaseInfoIfRequired();
 
         InstanceStatus status;
-        try {
-            status = getHealthCheckHandler().getStatus(instanceInfo.getStatus());
-        } catch (Exception e) {
-            logger.warn("Exception from healthcheckHandler.getStatus, setting status to DOWN", e);
-            status = InstanceStatus.DOWN;
-        }
+        ReentrantLock instanceStatusUpdateLock = new ReentrantLock();
+        if (instanceStatusUpdateLock.tryLock()) {
+            try {
+                try {
+                    status = getHealthCheckHandler().getStatus(instanceInfo.getStatus());
+                }
+                catch (Exception e) {
+                    logger.warn("Exception from healthcheckHandler.getStatus, setting status to DOWN", e);
+                    status = InstanceStatus.DOWN;
+                }
 
-        if (null != status) {
-            applicationInfoManager.setInstanceStatus(status);
+                if (null != status) {
+                    applicationInfoManager.setInstanceStatus(status);
+                }
+            }
+            finally {
+                instanceStatusUpdateLock.unlock();
+            }
+        }
+        else {
+            logger.warn("Cannot acquire update lock, aborting instance status refresh");
         }
     }