Merge branch 'generate_output_mappings' of github.com:weibocom/rill-f…

…low into dag_descriptor_converter
weibocom · Oct 25, 2024 · a816361 · a816361
2 parents db25f4d + b3e8348
commit a816361
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...versal/src/main/java/com/weibo/rill/flow/olympicene/traversal/runners/PassTaskRunner.java b/...versal/src/main/java/com/weibo/rill/flow/olympicene/traversal/runners/PassTaskRunner.java
@@ -20,18 +20,17 @@
 import com.google.common.collect.Sets;
 import com.weibo.rill.flow.interfaces.model.task.TaskInfo;
 import com.weibo.rill.flow.interfaces.model.task.TaskStatus;
+import com.weibo.rill.flow.olympicene.core.model.task.ExecutionResult;
 import com.weibo.rill.flow.olympicene.core.model.task.TaskCategory;
 import com.weibo.rill.flow.olympicene.core.runtime.DAGContextStorage;
 import com.weibo.rill.flow.olympicene.core.runtime.DAGInfoStorage;
 import com.weibo.rill.flow.olympicene.core.runtime.DAGStorageProcedure;
-import com.weibo.rill.flow.olympicene.core.model.task.ExecutionResult;
 import com.weibo.rill.flow.olympicene.core.switcher.SwitcherManager;
 import com.weibo.rill.flow.olympicene.traversal.helper.ContextHelper;
 import com.weibo.rill.flow.olympicene.traversal.mappings.InputOutputMapping;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections.CollectionUtils;
 
-import java.util.HashMap;
 import java.util.Map;
 
 @Slf4j
@@ -60,7 +59,7 @@ protected ExecutionResult doRun(String executionId, TaskInfo taskInfo, Map<Strin
         log.info("pass task begin to run executionId:{}, taskInfoName:{}", executionId, taskInfo.getName());
         if (CollectionUtils.isNotEmpty(taskInfo.getTask().getOutputMappings())) {
             Map<String, Object> context = ContextHelper.getInstance().getContext(dagContextStorage, executionId, taskInfo);
-            outputMappings(context, input, new HashMap<>(), taskInfo.getTask().getOutputMappings());
+            outputMappings(context, input, input, taskInfo.getTask().getOutputMappings());
             saveContext(executionId, context, Sets.newHashSet(taskInfo));
         }