Action Protos - Action fields for JiT execution (#2076)

Protobuf changes required for a basic JiT execution
in @dataform/core.

Author: ikholopov-omni

protos/BUILD

@@ -13,6 +13,7 @@ proto_library(
         "execution.proto",
         "profiles.proto",
     ],
+    deps = ["@com_google_protobuf//:struct_proto"],
 )
 
 ts_proto_library(

protos/core.proto

@@ -100,13 +100,25 @@ message CompilationError {
   string stack = 3;
 }
 
+// Compilation mode, unspecified is interpreted to AoT.
+enum ActionCompilationMode {
+  ACTION_COMPILATION_MODE_UNSPECIFIED = 0;
+  // Ahead-of-time compilation (regular Dataform compilation).
+  ACTION_COMPILATION_MODE_AOT = 1;
+  // Just-in-time compilation.
+  // Will only populate jit_code fields in compiled graph,
+  // to be re-compiled at execution time.
+  ACTION_COMPILATION_MODE_JIT = 2;
+}
+
 message ActionDescriptor {
   string description = 1;
   // For Operations, 'columns' may be set iff has_output == true.
   // For Assertions, 'columns' will always be empty.
   repeated ColumnDescriptor columns = 2;
   map<string, string> bigquery_labels = 3;
   Metadata metadata = 4;
+  ActionCompilationMode compilation_mode = 5;
 }
 
 message Metadata {
@@ -188,6 +200,8 @@ message Table {
   repeated string incremental_pre_ops = 28;
   repeated string incremental_post_ops = 29;
 
+  string jit_code = 38;
+
   // Warehouse specific features.
   BigQueryOptions bigquery = 22;
 
@@ -207,6 +221,7 @@ message Operation {
   bool disabled = 14;
 
   repeated string queries = 6;
+  string jit_code = 15;
   bool has_output = 8;
   repeated string tags = 9;
 
@@ -264,6 +279,8 @@ message Test {
   string test_query = 2;
   string expected_output_query = 3;
 
+  string jit_code = 10;
+
   // Generated.
   string file_name = 4;
 
@@ -379,6 +396,8 @@ message CompiledGraph {
 
   repeated Target targets = 11;
 
+  google.protobuf.Struct jit_context = 15;
+
   reserved 5, 6;
 }