Support PostgreSQL CREATE TABLE LIKE (sqlc-dev#2924)

* Support PostgreSQL CREATE TABLE LIKE

* test: Add test case for CREATE TABLE LIKE

---------

Co-authored-by: Kyle Conroy <[email protected]>

Author: xeraph-dev

internal/endtoend/testdata/ddl_create_table_like/issue.md

@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/481

internal/endtoend/testdata/ddl_create_table_like/postgresql/pgx/go/db.go

@@ -0,0 +1,2 @@
+-- name: AllRanked :many
+SELECT * FROM changes_ranked;

internal/endtoend/testdata/ddl_create_table_like/postgresql/pgx/go/models.go

@@ -0,0 +1,9 @@
+CREATE TABLE changes (
+    ranked INT NOT NULL
+);
+
+CREATE TABLE changes_ranked (
+    LIKE changes INCLUDING ALL,
+    rank_by_effect_size INT NOT NULL,
+    rank_by_abs_percent_change INT NOT NULL
+);

internal/endtoend/testdata/ddl_create_table_like/postgresql/pgx/go/query.sql.go

@@ -0,0 +1,10 @@
+version: "2"
+sql:
+  - engine: "postgresql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"
+        sql_package: "pgx/v5"

internal/endtoend/testdata/ddl_create_table_like/postgresql/pgx/query.sql

@@ -289,10 +289,6 @@ func (c *Catalog) createTable(stmt *ast.CreateTableStmt) error {
 		}
 	}
 
-	if stmt.ReferTable != nil && len(stmt.Cols) != 0 {
-		return errors.New("create table node cannot have both a ReferTable and Cols")
-	}
-
 	if stmt.ReferTable != nil {
 		_, original, err := c.getTable(stmt.ReferTable)
 		if err != nil {
@@ -302,23 +298,23 @@ func (c *Catalog) createTable(stmt *ast.CreateTableStmt) error {
 			newCol := *col // make a copy, so changes to the ReferTable don't propagate
 			tbl.Columns = append(tbl.Columns, &newCol)
 		}
-	} else {
-		for _, col := range stmt.Cols {
-			if notNull, ok := seen[col.Colname]; ok {
-				seen[col.Colname] = notNull || col.IsNotNull
-				if a, ok := coltype[col.Colname]; ok {
-					if !sameType(&a, col.TypeName) {
-						return fmt.Errorf("column %q has a type conflict", col.Colname)
-					}
+	}
+
+	for _, col := range stmt.Cols {
+		if notNull, ok := seen[col.Colname]; ok {
+			seen[col.Colname] = notNull || col.IsNotNull
+			if a, ok := coltype[col.Colname]; ok {
+				if !sameType(&a, col.TypeName) {
+					return fmt.Errorf("column %q has a type conflict", col.Colname)
 				}
-				continue
 			}
-			tc, err := c.defineColumn(stmt.Name, col)
-			if err != nil {
-				return err
-			}
-			tbl.Columns = append(tbl.Columns, tc)
+			continue
+		}
+		tc, err := c.defineColumn(stmt.Name, col)
+		if err != nil {
+			return err
 		}
+		tbl.Columns = append(tbl.Columns, tc)
 	}
 
 	// If one of the merged columns was not null, mark the column as not null