fix bug

SolaceProducts · Oct 11, 2023 · 48290c2 · 48290c2
1 parent 8699fe0
commit 48290c2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...src/main/java/com/solace/maas/ep/event/management/agent/publisher/HeartbeatGenerator.java b/...src/main/java/com/solace/maas/ep/event/management/agent/publisher/HeartbeatGenerator.java
@@ -29,7 +29,7 @@ public HeartbeatGenerator(SolaceConfiguration solaceConfiguration,
         this.solacePublisher = solacePublisher;
         this.runtimeAgentId = eventPortalProperties.getRuntimeAgentId();
         topic = solaceConfiguration.getTopicPrefix() + "heartbeat/v1";
-        this.runtimeAgentVersion = buildProperties.getVersion().substring(0, buildProperties.getVersion().indexOf('-'));
+        this.runtimeAgentVersion = getFormattedVersion(buildProperties.getVersion());
     }
 
     @Scheduled(fixedRate = 5000)
@@ -38,4 +38,10 @@ public void sendHeartbeat() {
         solacePublisher.publish(message, topic);
     }
 
+    private String getFormattedVersion(String version) {
+        if (version.endsWith("-SNAPSHOT")) {
+            return version.replace("-SNAPSHOT", "");
+        }
+        return version;
+    }
 }