From 8a3d575c4cafd61d3a8dc8572cb5f5f4933a5e36 Mon Sep 17 00:00:00 2001
From: wenweihuang <hww_justin@163.com>
Date: Fri, 27 Oct 2023 18:35:56 +0800
Subject: [PATCH] [INLONG-9136][Agent] Add instance manager

---
 .../agent/core/instance/InstanceManager.java  | 34 ++++++++++++-------
 1 file changed, 21 insertions(+), 13 deletions(-)

diff --git a/inlong-agent/agent-core/src/main/java/org/apache/inlong/agent/core/instance/InstanceManager.java b/inlong-agent/agent-core/src/main/java/org/apache/inlong/agent/core/instance/InstanceManager.java
index a5e0a57349c..a6c8381de36 100644
--- a/inlong-agent/agent-core/src/main/java/org/apache/inlong/agent/core/instance/InstanceManager.java
+++ b/inlong-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());
             }
         });
     }