Merge 6bb0cb6 into a58c7d5

partiql · Mar 19, 2024 · f57111f · f57111f
2 parents a58c7d5 + 6bb0cb6
commit f57111f
Show file tree

Hide file tree

Showing 7 changed files with 1,113 additions and 52 deletions.
diff --git a/partiql-cli/src/main/kotlin/org/partiql/cli/Main.kt b/partiql-cli/src/main/kotlin/org/partiql/cli/Main.kt
@@ -15,15 +15,17 @@
 
 package org.partiql.cli
 
-import AstPrinter
 import com.amazon.ion.system.IonSystemBuilder
+import com.amazon.ion.system.IonTextWriterBuilder
 import org.partiql.cli.pico.PartiQLCommand
 import org.partiql.cli.shell.info
 import org.partiql.lang.eval.EvaluationSession
 import org.partiql.parser.PartiQLParser
+import org.partiql.plan.Statement
 import org.partiql.plan.debug.PlanPrinter
 import org.partiql.planner.PartiQLPlanner
 import org.partiql.plugins.local.LocalConnector
+import org.partiql.plugins.local.toIon
 import picocli.CommandLine
 import java.io.PrintStream
 import java.nio.file.Paths
@@ -80,6 +82,16 @@ object Debug {
         out.info("-- Plan ----------")
         PlanPrinter.append(out, result.statement)
 
+        when (val plan = result.statement) {
+            is Statement.Query -> {
+                out.info("-- Schema ----------")
+                val outputSchema = java.lang.StringBuilder()
+                val ionWriter = IonTextWriterBuilder.minimal().withPrettyPrinting().build(outputSchema)
+                plan.root.type.toIon().writeTo(ionWriter)
+                out.info(outputSchema.toString())
+            }
+        }
+
         return "OK"
     }
 }