Build and print the new IR program for each query

The new IR building and printing is tested in src/test/java/io/trino/sql/dialect/trino
However, those unit tests have limited serialization capacity.
Full serialization requires injected dependencies.

With this change, each query run on the server will be rewritten
into a new IR program and printed on the console. It will use
dependencies for full serialization.

Note that the queries are captured in SqlQueryExecution after they are
planned and optimized, and right before they are fragmented. This is
the moment where CTE reuse will kick in.

Author: kasiafi

core/trino-main/src/main/java/io/trino/execution/SqlQueryExecution.java

@@ -61,6 +61,8 @@
 import io.trino.sql.analyzer.Analysis;
 import io.trino.sql.analyzer.Analyzer;
 import io.trino.sql.analyzer.AnalyzerFactory;
+import io.trino.sql.dialect.trino.ProgramBuilder;
+import io.trino.sql.newir.FormatOptions;
 import io.trino.sql.planner.AdaptivePlanner;
 import io.trino.sql.planner.InputExtractor;
 import io.trino.sql.planner.LogicalPlanner;
@@ -148,6 +150,7 @@ public class SqlQueryExecution
     private final EventDrivenTaskSourceFactory eventDrivenTaskSourceFactory;
     private final TaskDescriptorStorage taskDescriptorStorage;
     private final PlanOptimizersStatsCollector planOptimizersStatsCollector;
+    private final FormatOptions formatOptions;
 
     private SqlQueryExecution(
             PreparedQuery preparedQuery,
@@ -185,7 +188,8 @@ private SqlQueryExecution(
             SqlTaskManager coordinatorTaskManager,
             ExchangeManagerRegistry exchangeManagerRegistry,
             EventDrivenTaskSourceFactory eventDrivenTaskSourceFactory,
-            TaskDescriptorStorage taskDescriptorStorage)
+            TaskDescriptorStorage taskDescriptorStorage,
+            FormatOptions formatOptions)
     {
         try (SetThreadName _ = new SetThreadName("Query-" + stateMachine.getQueryId())) {
             this.slug = requireNonNull(slug, "slug is null");
@@ -240,6 +244,7 @@ private SqlQueryExecution(
             this.eventDrivenTaskSourceFactory = requireNonNull(eventDrivenTaskSourceFactory, "taskSourceFactory is null");
             this.taskDescriptorStorage = requireNonNull(taskDescriptorStorage, "taskDescriptorStorage is null");
             this.planOptimizersStatsCollector = requireNonNull(planOptimizersStatsCollector, "planOptimizersStatsCollector is null");
+            this.formatOptions = requireNonNull(formatOptions, "formatOptions is null");
         }
     }
 
@@ -503,6 +508,13 @@ private PlanRoot doPlanQuery(CachingTableStatsProvider tableStatsProvider)
         Plan plan = logicalPlanner.plan(analysis);
         queryPlan.set(plan);
 
+        try {
+            System.out.println(ProgramBuilder.buildProgram(plan.getRoot()).print(1, formatOptions));
+        }
+        catch (UnsupportedOperationException e) {
+            System.out.println("query not supported in the new IR: " + e.getMessage());
+        }
+
         // fragment the plan
         SubPlan fragmentedPlan;
         try (var _ = scopedSpan(tracer, "fragment-plan")) {
@@ -809,6 +821,7 @@ public static class SqlQueryExecutionFactory
         private final ExchangeManagerRegistry exchangeManagerRegistry;
         private final EventDrivenTaskSourceFactory eventDrivenTaskSourceFactory;
         private final TaskDescriptorStorage taskDescriptorStorage;
+        private final FormatOptions formatOptions;
 
         @Inject
         SqlQueryExecutionFactory(
@@ -841,7 +854,8 @@ public static class SqlQueryExecutionFactory
                 SqlTaskManager coordinatorTaskManager,
                 ExchangeManagerRegistry exchangeManagerRegistry,
                 EventDrivenTaskSourceFactory eventDrivenTaskSourceFactory,
-                TaskDescriptorStorage taskDescriptorStorage)
+                TaskDescriptorStorage taskDescriptorStorage,
+                FormatOptions formatOptions)
         {
             this.tracer = requireNonNull(tracer, "tracer is null");
             this.schedulerStats = requireNonNull(schedulerStats, "schedulerStats is null");
@@ -875,6 +889,7 @@ public static class SqlQueryExecutionFactory
             this.exchangeManagerRegistry = requireNonNull(exchangeManagerRegistry, "exchangeManagerRegistry is null");
             this.eventDrivenTaskSourceFactory = requireNonNull(eventDrivenTaskSourceFactory, "eventDrivenTaskSourceFactory is null");
             this.taskDescriptorStorage = requireNonNull(taskDescriptorStorage, "taskDescriptorStorage is null");
+            this.formatOptions = requireNonNull(formatOptions, "formatOptions is null");
         }
 
         @Override
@@ -925,7 +940,8 @@ public QueryExecution createQueryExecution(
                     coordinatorTaskManager,
                     exchangeManagerRegistry,
                     eventDrivenTaskSourceFactory,
-                    taskDescriptorStorage);
+                    taskDescriptorStorage,
+                    formatOptions);
         }
     }
 }