[INLONG-9136][Agent] Add instance manager

justinwwhuang · Oct 27, 2023 · 8a3d575 · 8a3d575
1 parent 442751a
commit 8a3d575
Showing 1 changed file with 21 additions and 13 deletions.
diff --git a/...agent/agent-core/src/main/java/org/apache/inlong/agent/core/instance/InstanceManager.java b/...agent/agent-core/src/main/java/org/apache/inlong/agent/core/instance/InstanceManager.java
@@ -132,21 +132,29 @@ private void traverseDbTasksToMemory() {
         instanceDb.getInstances(taskId).forEach((profileFromDb) -> {
             InstanceStateEnum dbState = profileFromDb.getState();
             Instance task = instanceMap.get(profileFromDb.getInstanceId());
-            if (dbState == InstanceStateEnum.DEFAULT) {
-                if (task == null) {
-                    LOGGER.info("traverseDbTasksToMemory add instance to mem taskId {} instanceId {}",
-                            profileFromDb.getTaskId(), profileFromDb.getInstanceId());
-                    addToMemory(profileFromDb);
+            switch (dbState) {
+                case DEFAULT: {
+                    if (task == null) {
+                        LOGGER.info("traverseDbTasksToMemory add instance to mem taskId {} instanceId {}",
+                                profileFromDb.getTaskId(), profileFromDb.getInstanceId());
+                        addToMemory(profileFromDb);
+                    }
+                    break;
                 }
-            } else if (dbState == InstanceStateEnum.FINISHED || dbState == InstanceStateEnum.DELETE) {
-                if (task != null) {
-                    LOGGER.info("traverseDbTasksToMemory delete instance from mem taskId {} instanceId {}",
-                            profileFromDb.getTaskId(), profileFromDb.getInstanceId());
-                    deleteFromMemory(profileFromDb.getInstanceId());
+                case FINISHED:
+                    DELETE: {
+                        if (task != null) {
+                            LOGGER.info("traverseDbTasksToMemory delete instance from mem taskId {} instanceId {}",
+                                    profileFromDb.getTaskId(), profileFromDb.getInstanceId());
+                            deleteFromMemory(profileFromDb.getInstanceId());
+                        }
+                        break;
+                    }
+                default: {
+                    LOGGER.error("instance invalid state {} taskId {} instanceId {}", dbState,
+                            profileFromDb.getTaskId(),
+                            profileFromDb.getInstanceId());
                 }
-            } else {
-                LOGGER.error("instance invalid state {} taskId {} instanceId {}", dbState, profileFromDb.getTaskId(),
-                        profileFromDb.getInstanceId());
             }
         });
     }