This is an automated email from the ASF dual-hosted git repository.

dockerzhang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/inlong.git


The following commit(s) were added to refs/heads/master by this push:
     new bc63709f92 [INLONG-8739][Manager] Fix The file collection task was 
mistakenly deleted (#8740)
bc63709f92 is described below

commit bc63709f92bbf8acf417be9bd43403667534879d
Author: fuweng11 <76141879+fuwen...@users.noreply.github.com>
AuthorDate: Wed Aug 16 14:12:31 2023 +0800

    [INLONG-8739][Manager] Fix The file collection task was mistakenly deleted 
(#8740)
---
 .../service/heartbeat/HeartbeatServiceImpl.java    |  1 -
 .../service/core/impl/AgentServiceTest.java        | 30 +++++++++++++++++++---
 2 files changed, 26 insertions(+), 5 deletions(-)

diff --git 
a/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/heartbeat/HeartbeatServiceImpl.java
 
b/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/heartbeat/HeartbeatServiceImpl.java
index 5b2cd5e46d..d3f634f769 100644
--- 
a/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/heartbeat/HeartbeatServiceImpl.java
+++ 
b/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/heartbeat/HeartbeatServiceImpl.java
@@ -87,7 +87,6 @@ public class HeartbeatServiceImpl implements HeartbeatService 
{
         ComponentTypeEnum componentType = 
ComponentTypeEnum.forType(request.getComponentType());
         switch (componentType) {
             case Agent:
-                return updateAgentHeartbeatOpt(request);
             case Sort:
             case DataProxy:
             case Cache:
diff --git 
a/inlong-manager/manager-service/src/test/java/org/apache/inlong/manager/service/core/impl/AgentServiceTest.java
 
b/inlong-manager/manager-service/src/test/java/org/apache/inlong/manager/service/core/impl/AgentServiceTest.java
index c8d127c505..0806c603d4 100644
--- 
a/inlong-manager/manager-service/src/test/java/org/apache/inlong/manager/service/core/impl/AgentServiceTest.java
+++ 
b/inlong-manager/manager-service/src/test/java/org/apache/inlong/manager/service/core/impl/AgentServiceTest.java
@@ -236,16 +236,38 @@ class AgentServiceTest extends ServiceBaseTest {
         // unbind group and mismatch
         bindGroup(false, "group1");
         TaskResult t1 = agent.pullTask();
-        Assertions.assertEquals(0, t1.getDataConfigs().size());
+        Assertions.assertEquals(1, t1.getDataConfigs().size());
+        Assertions.assertEquals(1, t1.getDataConfigs().stream()
+                .filter(dataConfig -> Integer.valueOf(dataConfig.getOp()) == 
ManagerOpEnum.FROZEN.getType())
+                .collect(Collectors.toSet())
+                .size());
+        DataConfig d1 = t1.getDataConfigs().get(0);
+        Assertions.assertEquals(sourceId, d1.getTaskId());
 
         // bind group and rematch
         bindGroup(true, "group1");
         TaskResult t2 = agent.pullTask();
-        Assertions.assertEquals(1, t2.getDataConfigs().size());
-        Assertions.assertEquals(1, t2.getDataConfigs().stream()
-                .filter(dataConfig -> Integer.valueOf(dataConfig.getOp()) == 
ManagerOpEnum.ADD.getType())
+        Assertions.assertEquals(0, t2.getDataConfigs().size());
+        Assertions.assertEquals(0, t2.getDataConfigs().stream()
+                .filter(dataConfig -> Integer.valueOf(dataConfig.getOp()) == 
ManagerOpEnum.ACTIVE.getType())
+                .collect(Collectors.toSet())
+                .size());
+
+        // update group to config success
+        final String groupId = sourceService.listSource(groupStream.getLeft(), 
groupStream.getRight()).stream()
+                .filter(source -> source.getTemplateId() != null)
+                .findAny()
+                .get()
+                .getInlongGroupId();
+        groupMapper.updateStatus(groupId, 
GroupStatus.CONFIG_SUCCESSFUL.getCode(), GLOBAL_OPERATOR);
+        TaskResult t3 = agent.pullTask();
+        Assertions.assertEquals(1, t3.getDataConfigs().size());
+        Assertions.assertEquals(1, t3.getDataConfigs().stream()
+                .filter(dataConfig -> Integer.valueOf(dataConfig.getOp()) == 
ManagerOpEnum.ACTIVE.getType())
                 .collect(Collectors.toSet())
                 .size());
+        DataConfig d3 = t3.getDataConfigs().get(0);
+        Assertions.assertEquals(sourceId, d3.getTaskId());
     }
 
     /**

Reply via email to