Task provenance

docs/config.rst

@@ -254,6 +254,7 @@ Name                Description
 enabled             When ``true`` turns on the generation of the DAG file (default: ``false``).
 file                Graph file name (default: ``dag-<timestamp>.dot``).
 overwrite           When ``true`` overwrites any existing DAG file with the same name.
+type                Can be ``abstract`` to render the abstract DAG or ``concrete`` to render the concrete (task) DAG (default: ``abstract``).
 ================== ================
 
 The above options can be used by prefixing them with the ``dag`` scope or surrounding them by curly

modules/nextflow/src/main/groovy/nextflow/Session.groovy

@@ -40,6 +40,7 @@ import nextflow.conda.CondaConfig
 import nextflow.config.Manifest
 import nextflow.container.ContainerConfig
 import nextflow.dag.DAG
+import nextflow.dag.ConcreteDAG
 import nextflow.exception.AbortOperationException
 import nextflow.exception.AbortSignalException
 import nextflow.exception.IllegalConfigException
@@ -193,6 +194,8 @@ class Session implements ISession {
 
     private DAG dag
 
+    private ConcreteDAG concreteDag
+
     private CacheDB cache
 
     private Barrier processesBarrier = new Barrier()
@@ -345,6 +348,7 @@ class Session implements ISession {
 
         // -- DAG object
         this.dag = new DAG()
+        this.concreteDag = new ConcreteDAG()
 
         // -- init work dir
         this.workDir = ((config.workDir ?: 'work') as Path).complete()
@@ -800,6 +804,8 @@ class Session implements ISession {
 
     DAG getDag() { this.dag }
 
+    ConcreteDAG getConcreteDAG() { this.concreteDag }
+
     ExecutorService getExecService() { execService }
 
     /**
@@ -1008,6 +1014,9 @@ class Session implements ISession {
         final trace = handler.safeTraceRecord()
         cache.putTaskAsync(handler, trace)
 
+        // save the task meta file to the task directory
+        handler.writeMetaFile()
+
         // notify the event to the observers
         for( int i=0; i<observers.size(); i++ ) {
             final observer = observers.get(i)

modules/nextflow/src/main/groovy/nextflow/dag/ConcreteDAG.groovy

@@ -0,0 +1,105 @@
+/*
+ * Copyright 2013-2023, Seqera Labs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package nextflow.dag
+
+import java.nio.file.Path
+import java.util.regex.Pattern
+
+import groovy.transform.MapConstructor
+import groovy.transform.ToString
+import groovy.util.logging.Slf4j
+import nextflow.processor.TaskRun
+import nextflow.script.params.FileOutParam
+/**
+ * Model the conrete (task) graph of a pipeline execution.
+ *
+ * @author Ben Sherman <bentshermann@gmail.com>
+ */
+@Slf4j
+class ConcreteDAG {
+
+    Map<String,Task> nodes = new HashMap<>(100)
+
+    /**
+     * Add a task to the graph
+     *
+     * @param task
+     */
+    synchronized void addTask( TaskRun task ) {
+        final hash = task.hash.toString()
+        final label = "[${hash.substring(0,2)}/${hash.substring(2,8)}] ${task.name}"
+        final inputs = task.getInputFilesMap()
+            .collect { name, path ->
+                new Input(name: name, path: path, predecessor: getPredecessorHash(path))
+            }
+
+        nodes[hash] = new Task(
+            index: nodes.size(),
+            label: label,
+            inputs: inputs
+        )
+    }
+
+    static public String getPredecessorHash(Path path) {
+        final pattern = Pattern.compile('.*/([0-9a-f]{2}/[0-9a-f]{30})')
+        final matcher = pattern.matcher(path.toString())
+
+        matcher.find() ? matcher.group(1).replace('/', '') : null
+    }
+
+    /**
+     * Add a task's outputs to the graph
+     *
+     * @param task
+     */
+    synchronized void addTaskOutputs( TaskRun task ) {
+        final hash = task.hash.toString()
+        final outputs = task.getOutputsByType(FileOutParam)
+            .values()
+            .flatten()
+            .collect { path ->
+                new Output(name: path.name, path: path)
+            }
+
+        nodes[hash].outputs = outputs
+    }
+
+    @MapConstructor
+    @ToString(includeNames = true, includes = 'label', includePackage=false)
+    static protected class Task {
+        int index
+        String label
+        List<Input> inputs
+        List<Output> outputs
+
+        String getSlug() { "t${index}" }
+    }
+
+    @MapConstructor
+    static protected class Input {
+        String name
+        Path path
+        String predecessor
+    }
+
+    @MapConstructor
+    static protected class Output {
+        String name
+        Path path
+    }
+
+}

modules/nextflow/src/main/groovy/nextflow/dag/CytoscapeHtmlRenderer.groovy

@@ -29,7 +29,7 @@ import java.nio.file.Path
 class CytoscapeHtmlRenderer implements DagRenderer {
 
     @Override
-    void renderDocument(DAG dag, Path file) {
+    void renderAbstractGraph(DAG dag, Path file) {
         String tmplPage = readTemplate()
         String network = CytoscapeJsRenderer.renderNetwork(dag)
         file.text = tmplPage.replaceAll(~/\/\* REPLACE_WITH_NETWORK_DATA \*\//, network)

modules/nextflow/src/main/groovy/nextflow/dag/CytoscapeJsRenderer.groovy

@@ -29,7 +29,7 @@ import java.nio.file.Path
 class CytoscapeJsRenderer implements DagRenderer {
 
     @Override
-    void renderDocument(DAG dag, Path file) {
+    void renderAbstractGraph(DAG dag, Path file) {
         file.text = renderNetwork(dag)
     }
 

modules/nextflow/src/main/groovy/nextflow/dag/DAG.groovy

@@ -41,7 +41,7 @@ import nextflow.script.params.OutputsList
 import nextflow.script.params.TupleInParam
 import nextflow.script.params.TupleOutParam
 /**
- * Model a direct acyclic graph of the pipeline execution.
+ * Model the abstract graph of a pipeline execution.
  *
  * @author Paolo Di Tommaso <paolo.ditommaso@gmail.com>
  */

modules/nextflow/src/main/groovy/nextflow/dag/DagRenderer.groovy

@@ -24,10 +24,19 @@ import java.nio.file.Path
  * @author Paolo Di Tommaso <paolo.ditommaso@gmail.com>
  * @author Mike Smoot <mes@aescon.com>
  */
-interface DagRenderer {
+trait DagRenderer {
 
     /**
-     * Render the dag to the specified file.
+     * Render an abstract (process/operator) DAG.
      */
-    void renderDocument(DAG dag, Path file);
+    void renderAbstractGraph(DAG dag, Path file) {
+        throw new UnsupportedOperationException("Abstract graph rendering is not supported for this file format")
+    }
+
+    /**
+     * Render a concrete (task) DAG.
+     */
+    void renderConcreteGraph(ConcreteDAG dag, Path file) {
+        throw new UnsupportedOperationException("Concrete graph rendering is not supported for this file format")
+    }
 }

modules/nextflow/src/main/groovy/nextflow/dag/DotRenderer.groovy

@@ -46,7 +46,7 @@ class DotRenderer implements DagRenderer {
     static String normalise(String str) { str.replaceAll(/[^0-9_A-Za-z]/,'') }
 
     @Override
-    void renderDocument(DAG dag, Path file) {
+    void renderAbstractGraph(DAG dag, Path file) {
         file.text = renderNetwork(dag)
     }
 

modules/nextflow/src/main/groovy/nextflow/dag/GexfRenderer.groovy

@@ -41,7 +41,7 @@ class GexfRenderer implements DagRenderer {
     }
 
     @Override
-    void renderDocument(DAG dag, Path file) {
+    void renderAbstractGraph(DAG dag, Path file) {
         final Charset charset = Charset.defaultCharset()
         Writer bw = Files.newBufferedWriter(file, charset) 
         final XMLOutputFactory xof = XMLOutputFactory.newFactory()

modules/nextflow/src/main/groovy/nextflow/dag/GraphVizRenderer.groovy

@@ -42,7 +42,7 @@ class GraphvizRenderer implements DagRenderer {
      * See http://www.graphviz.org for more info.
      */
     @Override
-    void renderDocument(DAG dag, Path target) {
+    void renderAbstractGraph(DAG dag, Path target) {
         def result = Files.createTempFile('nxf-',".$format")
         def temp = Files.createTempFile('nxf-','.dot')
         // save the DAG as `dot` to a temp file

modules/nextflow/src/main/groovy/nextflow/dag/MermaidRenderer.groovy

@@ -27,11 +27,7 @@ import java.nio.file.Path
 class MermaidRenderer implements DagRenderer {
 
     @Override
-    void renderDocument(DAG dag, Path file) {
-        file.text = renderNetwork(dag)
-    }
-
-    String renderNetwork(DAG dag) {
+    void renderAbstractGraph(DAG dag, Path file) {
         def lines = []
         lines << "flowchart TD"
 
@@ -45,7 +41,7 @@ class MermaidRenderer implements DagRenderer {
 
         lines << ""
 
-        return lines.join('\n')
+        file.text = lines.join('\n')
     }
 
     private String renderVertex(DAG.Vertex vertex) {
@@ -76,4 +72,49 @@ class MermaidRenderer implements DagRenderer {
 
         return "${edge.from.name} -->${label} ${edge.to.name}"
     }
+
+    @Override
+    void renderConcreteGraph(ConcreteDAG graph, Path file) {
+        def renderedOutputs = [] as Set<Path>
+        def numInputs = 0
+        def numOutputs = 0
+
+        def lines = []
+        lines << "flowchart TD"
+
+        // render tasks and task inputs
+        graph.nodes.values().each { task ->
+            // render task node
+            lines << "    ${task.getSlug()}[\"${task.label}\"]"
+
+            task.inputs.each { input ->
+                // render task input from predecessor
+                if( input.predecessor != null ) {
+                    final pred = graph.nodes[input.predecessor]
+                    lines << "    ${pred.getSlug()} -->|${input.name}| ${task.getSlug()}"
+                    renderedOutputs << input.path
+                }
+
+                // render task input from source node
+                else {
+                    numInputs += 1
+                    lines << "    i${numInputs}(( )) -->|${input.name}| ${task.getSlug()}"
+                }
+            }
+        }
+
+        // render task outputs with sink nodes
+        graph.nodes.values().each { task ->
+            task.outputs.each { output ->
+                if( output.path !in renderedOutputs ) {
+                    numOutputs += 1
+                    lines << "    ${task.getSlug()} -->|${output.name}| o${numOutputs}(( ))"
+                }
+            }
+        }
+
+        lines << ""
+
+        file.text = lines.join('\n')
+    }
 }

modules/nextflow/src/main/groovy/nextflow/dag/NodeMarker.groovy

@@ -41,7 +41,7 @@ class NodeMarker {
     static private Session getSession() { Global.session as Session }
 
     /**
-     *  Creates a new vertex in the DAG representing a computing `process`
+     * Creates a vertex in the abstract DAG representing a computing `process`
      *
      * @param label The label associated to the process
      * @param inputs The list of inputs entering in the process
@@ -53,7 +53,7 @@ class NodeMarker {
     }
 
     /**
-     * Creates a new DAG vertex representing a dataflow operator
+     * Creates a vertex in the abstract DAG representing a dataflow operator
      *
      * @param label The operator label
      * @param inputs The operator input(s). It can be either a single channel or a list of channels.
@@ -67,7 +67,7 @@ class NodeMarker {
     }
 
     /**
-     * Creates a vertex in the DAG representing a dataflow channel source.
+     * Creates a vertex in the abstract DAG representing a dataflow channel source.
      *
      * @param label The node description
      * @param source Either a dataflow channel or a list of channel.

modules/nextflow/src/main/groovy/nextflow/processor/TaskHandler.groovy

@@ -21,7 +21,11 @@ import static nextflow.processor.TaskStatus.*
 
 import java.nio.file.NoSuchFileException
 
+import groovy.json.JsonBuilder
 import groovy.util.logging.Slf4j
+import nextflow.dag.ConcreteDAG
+import nextflow.extension.FilesEx
+import nextflow.script.params.FileOutParam
 import nextflow.trace.TraceRecord
 /**
  * Actions to handle the underlying job running the user task.
@@ -214,6 +218,31 @@ abstract class TaskHandler {
         return record
     }
 
+    void writeMetaFile() {
+        final record = [
+            hash: task.hash.toString(),
+            inputs: task.getInputFilesMap().collect { name, path ->
+                [
+                    name: name,
+                    path: path.toString(),
+                    predecessor: ConcreteDAG.getPredecessorHash(path)
+                ]
+            },
+            outputs: task.getOutputsByType(FileOutParam).values().flatten().collect { path ->
+                final file = path.toFile()
+
+                [
+                    name: path.name,
+                    path: path.toString(),
+                    size: file.size(),
+                    checksum: file.isFile() ? FilesEx.getChecksum(path) : null
+                ]
+            }
+        ]
+
+        task.workDir.resolve(TaskRun.CMD_META).text = new JsonBuilder(record).toString()
+    }
+
     /**
      * Determine if a process can be forked i.e. can launch
      * a parallel task execution. This is only enforced when

modules/nextflow/src/main/groovy/nextflow/processor/TaskProcessor.groovy

@@ -524,7 +524,7 @@ class TaskProcessor {
         def invoke = new InvokeTaskAdapter(this, opInputs.size())
         session.allOperators << (operator = new DataflowOperator(group, params, invoke))
 
-        // notify the creation of a new vertex the execution DAG
+        // notify the creation of a new process in the abstract DAG
         NodeMarker.addProcessNode(this, config.getInputs(), config.getOutputs())
 
         // fix issue #41

modules/nextflow/src/main/groovy/nextflow/processor/TaskRun.groovy

@@ -543,6 +543,7 @@ class TaskRun implements Cloneable {
     static final public String CMD_RUN = '.command.run'
     static final public String CMD_TRACE = '.command.trace'
     static final public String CMD_ENV = '.command.env'
+    static final public String CMD_META = '.command.meta.json'
 
 
     String toString( ) {