mb.pie.runtime.exec.TaskExecutor.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pie.runtime Show documentation
PIE runtime
The newest version!
package mb.pie.runtime.exec

import mb.pie.api.*
import mb.pie.api.exec.Cancelled
import mb.pie.api.exec.ExecReason
import mb.pie.api.fs.stamp.FileSystemStamper
import mb.pie.api.stamp.OutputStamper

class TaskExecutor(
  private val taskDefs: TaskDefs,
  private val resourceSystems: ResourceSystems,
  private val visited: MutableMap>,
  private val store: Store,
  private val share: Share,
  private val defaultOutputStamper: OutputStamper,
  private val defaultRequireFileSystemStamper: FileSystemStamper,
  private val defaultProvideFileSystemStamper: FileSystemStamper,
  private val layer: Layer,
  private val logger: Logger,
  private val executorLogger: ExecutorLogger,
  private val postExecFunc: ((TaskKey, TaskData<*, *>) -> Unit)?
) {
  fun  exec(key: TaskKey, task: Task, reason: ExecReason, requireTask: RequireTask, cancel: Cancelled): TaskData {
    cancel.throwIfCancelled()
    executorLogger.executeStart(key, task, reason)
    // OPTO: Inline share functions. Requires statically knowledge of the specific Share type to use.
    val data = share.share(key, { execInternal(key, task, requireTask, cancel) }, { visited[key] })
    executorLogger.executeEnd(key, task, reason, data)
    return data.cast()
  }

  private fun  execInternal(key: TaskKey, task: Task, requireTask: RequireTask, cancel: Cancelled): TaskData {
    cancel.throwIfCancelled()
    // Execute
    val context = ExecContextImpl(requireTask, cancel, taskDefs, resourceSystems, store, defaultOutputStamper, defaultRequireFileSystemStamper, defaultProvideFileSystemStamper, logger)
    val output = task.exec(context)
    Stats.addExecution()
    val (taskRequires, resourceRequires, resourceProvides) = context.deps()
    val data = TaskData(task.input, output, taskRequires, resourceRequires, resourceProvides)
    // Validate well-formedness of the dependency graph, before writing.
    store.readTxn().use {
      layer.validatePreWrite(key, data, it)
    }
    // Call post-execution function.
    postExecFunc?.invoke(key, data)
    // Write output and dependencies to the store.
    store.writeTxn().use {
      it.setData(key, data)
    }
    // Validate well-formedness of the dependency graph, after writing.
    store.readTxn().use {
      layer.validatePostWrite(key, data, it)
    }
    // Mark as visited.
    visited[key] = data
    return data
  }
}