Bugfix: the listener will re-register when device in E2E agent discon…

…nected (#334)
microsoft · Mar 7, 2023 · d85a1da · d85a1da
1 parent 4f3940a
commit d85a1da
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/agent/src/main/java/com/microsoft/hydralab/agent/service/AgentWebSocketClientService.java b/agent/src/main/java/com/microsoft/hydralab/agent/service/AgentWebSocketClientService.java
@@ -15,10 +15,10 @@
 import com.microsoft.hydralab.common.entity.common.TestRun;
 import com.microsoft.hydralab.common.entity.common.TestTask;
 import com.microsoft.hydralab.common.entity.common.TestTaskSpec;
+import com.microsoft.hydralab.common.file.StorageServiceClientProxy;
 import com.microsoft.hydralab.common.monitor.MetricPushGateway;
 import com.microsoft.hydralab.common.util.Const;
 import com.microsoft.hydralab.common.util.GlobalConstant;
-import com.microsoft.hydralab.common.file.StorageServiceClientProxy;
 import com.microsoft.hydralab.common.util.ThreadUtils;
 import io.micrometer.core.instrument.MeterRegistry;
 import io.micrometer.core.instrument.Tags;
@@ -72,6 +72,7 @@ public class AgentWebSocketClientService implements TestTaskRunCallback {
     private boolean isPrometheusEnabled;
     @Autowired(required = false)
     private MetricPushGateway pushGateway;
+    boolean isAgentInit = false;
 
     public void onMessage(Message message) {
         log.info("onMessage Receive bytes message {}", message);
@@ -91,8 +92,11 @@ public void onMessage(Message message) {
                  * [agentUser.setTeamName -> meterRegistry.config().commonTags -> deviceManager.init
                  *  -> (deviceControlService.provideDeviceList + deviceStatbilityMonitor.addDeviceMetricRegistration)].
                  */
-                registerAgentMetrics();
-                deviceControlService.deviceManagerInit();
+                if (!isAgentInit) {
+                    registerAgentMetrics();
+                    deviceControlService.deviceManagerInit();
+                    isAgentInit = true;
+                }
                 deviceControlService.provideDeviceList(agentUser.getBatteryStrategy());
                 return;
             case Const.Path.HEARTBEAT: