[INLONG-9953][Manager] Fix the problem of stop stream source failed (#…

…9954)
apache · Apr 10, 2024 · c13e355 · c13e355
1 parent 50539a5
commit c13e355
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...r-service/src/main/java/org/apache/inlong/manager/service/core/impl/AgentServiceImpl.java b/...r-service/src/main/java/org/apache/inlong/manager/service/core/impl/AgentServiceImpl.java
@@ -21,6 +21,7 @@
 import org.apache.inlong.common.constant.MQType;
 import org.apache.inlong.common.db.CommandEntity;
 import org.apache.inlong.common.enums.PullJobTypeEnum;
+import org.apache.inlong.common.enums.TaskStateEnum;
 import org.apache.inlong.common.enums.TaskTypeEnum;
 import org.apache.inlong.common.pojo.agent.CmdConfig;
 import org.apache.inlong.common.pojo.agent.DataConfig;
@@ -672,7 +673,9 @@ private DataConfig getDataConfig(StreamSourceEntity entity, int op) {
         String extParams = sourceOperator.getExtParams(entity);
         if (groupEntity != null && streamEntity != null) {
             dataConfig.setState(
-                    SourceStatus.NORMAL_STATUS_SET.contains(SourceStatus.forCode(entity.getStatus())) ? 1 : 0);
+                    SourceStatus.NORMAL_STATUS_SET.contains(SourceStatus.forCode(entity.getStatus()))
+                            ? TaskStateEnum.RUNNING.getType()
+                            : TaskStateEnum.FROZEN.getType());
             dataConfig.setSyncSend(streamEntity.getSyncSend());
             if (SourceType.FILE.equalsIgnoreCase(entity.getSourceType())
                     && StringUtils.isNotBlank(streamEntity.getDataSeparator())) {