Temporal persistence layer integration

api/src/main/scala/ai/chronon/api/planner/DependencyResolver.scala

@@ -68,8 +68,9 @@ object DependencyResolver {
     result
   }
 
-  def computeInputRange(queryRange: PartitionRange, tableDep: TableDependency)(implicit
-      partitionSpec: PartitionSpec): Option[PartitionRange] = {
+  def computeInputRange(queryRange: PartitionRange, tableDep: TableDependency): Option[PartitionRange] = {
+
+    implicit val partitionSpec: PartitionSpec = queryRange.partitionSpec
 
     require(queryRange != null, "Query range cannot be null")
     require(queryRange.start != null, "Query range start cannot be null")

api/thrift/orchestration.thrift

@@ -221,6 +221,14 @@ union NodeUnion {
     // TODO: add other types of nodes
 }
 
+enum NodeRunStatus {
+    UNKNOWN = 0,
+    WAITING = 1,
+    RUNNING = 2,
+    SUCCEEDED = 3,
+    FAILED = 4
+}
+
 // ====================== End of Modular Join Spark Job Args ===================
 
 // ====================== Orchestration Service API Types ======================
@@ -250,15 +258,6 @@ struct UploadResponse {
 
 // ====================== End of Orchestration Service API Types ======================
 
-/**
-* Below are dummy thrift objects for execution layer skeleton code using temporal
-* TODO: Need to update these to fill in all the above relevant fields
-**/
-struct DummyNode {
-    1: optional string name
-    2: optional list<DummyNode> dependencies
-}
-
 /**
 * -- Phase 0 plan -- (same as chronon oss)
 * StagingQuery::query - [deps.table] >> query

orchestration/BUILD.bazel

@@ -18,6 +18,7 @@ scala_library(
         maven_artifact("io.temporal:temporal-sdk"),
         maven_artifact("io.temporal:temporal-serviceclient"),
         maven_artifact("com.fasterxml.jackson.core:jackson-databind"),
+        maven_artifact_with_suffix("com.fasterxml.jackson.module:jackson-module-scala"),
         maven_artifact("com.google.protobuf:protobuf-java"),
         maven_artifact("com.google.code.findbugs:jsr305"),
         maven_artifact("io.grpc:grpc-api"),
@@ -50,6 +51,7 @@ test_deps = _VERTX_DEPS + _SCALA_TEST_DEPS + [
     maven_artifact("io.temporal:temporal-serviceclient"),
     maven_artifact("com.fasterxml.jackson.core:jackson-core"),
     maven_artifact("com.fasterxml.jackson.core:jackson-databind"),
+    maven_artifact_with_suffix("com.fasterxml.jackson.module:jackson-module-scala"),
     maven_artifact("io.grpc:grpc-api"),
     maven_artifact("io.grpc:grpc-core"),
     maven_artifact("io.grpc:grpc-stub"),
@@ -84,7 +86,7 @@ scala_test_suite(
         ["src/test/**/*.scala"],
         # Excluding integration tests
         exclude = [
-            "src/test/**/NodeExecutionWorkflowIntegrationSpec.scala",
+            "src/test/**/NodeWorkflowIntegrationSpec.scala",
             "src/test/**/GcpPubSubIntegrationSpec.scala",
         ],
     ),
@@ -96,7 +98,7 @@ scala_test_suite(
     name = "integration_tests",
     srcs = glob(
         [
-            "src/test/**/NodeExecutionWorkflowIntegrationSpec.scala",
+            "src/test/**/NodeWorkflowIntegrationSpec.scala",
             "src/test/**/GcpPubSubIntegrationSpec.scala",
         ],
     ),