fix(gms): GMS hang when upgrade image #8270 (#8271)

Co-authored-by: yangjd33 <[email protected]> Co-authored-by: david-leifker <[email protected]>
datahub-project · Jul 29, 2023 · 753b0f1 · 753b0f1
1 parent 55c6592
commit 753b0f1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...factories/src/main/java/com/linkedin/metadata/boot/kafka/DataHubUpgradeKafkaListener.java b/...factories/src/main/java/com/linkedin/metadata/boot/kafka/DataHubUpgradeKafkaListener.java
@@ -108,6 +108,10 @@ public void checkSystemVersion(final ConsumerRecord<String, GenericRecord> consu
   }
 
   public void waitForUpdate() {
+    if (!_configurationProvider.getSystemUpdate().isWaitForSystemUpdate()) {
+      log.warn("Wait for system update is disabled. Proceeding with startup.");
+      IS_UPDATED.getAndSet(true);
+    }
     int maxBackOffs = Integer.parseInt(_configurationProvider.getSystemUpdate().getMaxBackOffs());
     long initialBackOffMs = Long.parseLong(_configurationProvider.getSystemUpdate().getInitialBackOffMs());
     int backOffFactor = Integer.parseInt(_configurationProvider.getSystemUpdate().getBackOffFactor());