Add query to show output

Signed-off-by: Chen Dai <daichen@amazon.com>
opensearch-project · Oct 19, 2023 · 4a7a36e · 4a7a36e
1 parent ef8031a
commit 4a7a36e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/...c/main/scala/org/opensearch/flint/spark/sql/mv/FlintSparkMaterializedViewAstBuilder.scala b/...c/main/scala/org/opensearch/flint/spark/sql/mv/FlintSparkMaterializedViewAstBuilder.scala
@@ -53,15 +53,16 @@ trait FlintSparkMaterializedViewAstBuilder extends FlintSparkSqlExtensionsVisito
   override def visitShowMaterializedViewStatement(
       ctx: ShowMaterializedViewStatementContext): AnyRef = {
     val outputSchema = Seq(
-      AttributeReference("materialized_view_name", StringType, nullable = false)())
+      AttributeReference("materialized_view_name", StringType, nullable = false)(),
+      AttributeReference("query", StringType, nullable = false)())
 
     FlintSparkSqlCommand(outputSchema) { flint =>
       val catalogDbName = ctx.catalogDb.getText
       val indexNamePattern = FlintSparkIndex.flintIndexNamePrefix(catalogDbName) + "*"
       flint
         .describeIndexes(indexNamePattern)
         .collect { case mv: FlintSparkMaterializedView =>
-          Row(mv.mvName)
+          Row(mv.mvName, mv.query)
         }
     }
   }

diff --git a/...test/src/test/scala/org/opensearch/flint/spark/FlintSparkMaterializedViewSqlITSuite.scala b/...test/src/test/scala/org/opensearch/flint/spark/FlintSparkMaterializedViewSqlITSuite.scala
@@ -134,13 +134,15 @@ class FlintSparkMaterializedViewSqlITSuite extends FlintSparkSuite {
     flint.materializedView().name("spark_catalog.default.mv1").query(testQuery).create()
     checkAnswer(
       sql(s"SHOW MATERIALIZED VIEW IN spark_catalog"),
-      Seq(Row("spark_catalog.default.mv1")))
+      Seq(Row("spark_catalog.default.mv1", testQuery)))
 
     // Show in catalog.database
     flint.materializedView().name("spark_catalog.default.mv2").query(testQuery).create()
     checkAnswer(
       sql(s"SHOW MATERIALIZED VIEW IN spark_catalog.default"),
-      Seq(Row("spark_catalog.default.mv1"), Row("spark_catalog.default.mv2")))
+      Seq(
+        Row("spark_catalog.default.mv1", testQuery),
+        Row("spark_catalog.default.mv2", testQuery)))
 
     checkAnswer(sql(s"SHOW MATERIALIZED VIEW IN spark_catalog.other"), Seq.empty)
   }