Process parameters for CREATE SCHEMA

trinodb · Mar 17, 2022 · ee0b666 · ee0b666
1 parent ac6e3f3
commit ee0b666
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/core/trino-parser/src/main/java/io/trino/sql/tree/DefaultTraversalVisitor.java b/core/trino-parser/src/main/java/io/trino/sql/tree/DefaultTraversalVisitor.java
@@ -758,6 +758,16 @@ protected Void visitAddColumn(AddColumn node, C context)
         return null;
     }
 
+    @Override
+    protected Void visitCreateSchema(CreateSchema node, C context)
+    {
+        for (Property property : node.getProperties()) {
+            process(property, context);
+        }
+
+        return null;
+    }
+
     @Override
     protected Void visitCreateTable(CreateTable node, C context)
     {

diff --git a/testing/trino-tests/src/test/java/io/trino/tests/TestMockConnector.java b/testing/trino-tests/src/test/java/io/trino/tests/TestMockConnector.java
@@ -16,6 +16,7 @@
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.ImmutableSet;
+import io.trino.Session;
 import io.trino.connector.MockConnectorFactory;
 import io.trino.connector.MockConnectorPlugin;
 import io.trino.connector.MockConnectorTableHandle;
@@ -210,6 +211,16 @@ public void testSchemaProperties()
                 .hasMessage("Catalog 'mock' schema property 'unknown_property' does not exist");
     }
 
+    @Test
+    public void testExecuteWithSchemaProperties()
+    {
+        String query = "CREATE SCHEMA mock.test_schema WITH (boolean_schema_property = ?)";
+        Session session = Session.builder(getSession())
+                .addPreparedStatement("my_query", query)
+                .build();
+        computeActual(session, "EXECUTE my_query USING true");
+    }
+
     @Test
     public void testTableProperties()
     {