Support create table as select with column list (#214)

goccy · Jul 1, 2024 · 9d65c7d · 9d65c7d
1 parent ed459f5
commit 9d65c7d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/internal/analyzer.go b/internal/analyzer.go
@@ -75,6 +75,7 @@ func newAnalyzerOptions() (*zetasql.AnalyzerOptions, error) {
 		zetasql.FeatureV11WithOnSubquery,
 		zetasql.FeatureV13Pivot,
 		zetasql.FeatureV13Unpivot,
+		zetasql.FeatureCreateTableAsSelectColumnList,
 	})
 	langOpt.SetSupportedStatementKinds([]ast.Kind{
 		ast.BeginStmt,

diff --git a/query_test.go b/query_test.go
@@ -5887,6 +5887,17 @@ SELECT c1 * ? * ? FROM t1;
 			args:         []interface{}{int64(1), int64(2), int64(3)},
 			expectedRows: [][]interface{}{{int64(6)}},
 		},
+		{
+			name: "create table as select with column list",
+			query: `
+CREATE TABLE table1 (field_a STRING NOT NULL);
+INSERT INTO table1 (field_a) VALUES ("test");
+CREATE TEMP TABLE table2 (field_x STRING NOT NULL)
+AS (SELECT field_a FROM table1);
+SELECT * FROM table2;
+`,
+			expectedRows: [][]interface{}{{"test"}},
+		},
 	} {
 		test := test
 		t.Run(test.name, func(t *testing.T) {