postgresql: Add support for renaming types (#971)

Author: kyleconroy

internal/endtoend/testdata/ddl_alter_type_rename/postgresql/go/db.go

@@ -0,0 +1,11 @@
+CREATE TYPE event AS enum ('START', 'STOP');
+
+ALTER TYPE event RENAME TO "new_event";
+
+CREATE TABLE log_lines (
+  id     BIGSERIAL    PRIMARY KEY,
+  status "new_event"  NOT NULL
+);
+
+-- name: ListAuthors :many
+SELECT * FROM log_lines;

internal/endtoend/testdata/ddl_alter_type_rename/postgresql/go/models.go

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "name": "querytest",
+      "schema": "query.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/ddl_alter_type_rename/postgresql/go/query.sql.go

@@ -562,6 +562,16 @@ func translate(node *nodes.Node) (ast.Node, error) {
 				NewName: makeString(n.Newname),
 			}, nil
 
+		case nodes.ObjectType_OBJECT_TYPE:
+			rel, err := parseRelation(n.Object)
+			if err != nil {
+				return nil, fmt.Errorf("nodes.RenameStmt: TYPE: %w", err)
+			}
+			return &ast.RenameTypeStmt{
+				Type:    rel.TypeName(),
+				NewName: makeString(n.Newname),
+			}, nil
+
 		}
 		return nil, errSkip
 

internal/endtoend/testdata/ddl_alter_type_rename/postgresql/query.sql

@@ -0,0 +1,10 @@
+package ast
+
+type RenameTypeStmt struct {
+	Type    *TypeName
+	NewName *string
+}
+
+func (n *RenameTypeStmt) Pos() int {
+	return 0
+}

internal/endtoend/testdata/ddl_alter_type_rename/postgresql/sqlc.json

@@ -204,6 +204,9 @@ func (a *application) apply(parent ast.Node, name string, iter *iterator, n ast.
 	case *ast.RenameTableStmt:
 		a.apply(n, "Table", nil, n.Table)
 
+	case *ast.RenameTypeStmt:
+		a.apply(n, "Type", nil, n.Type)
+
 	case *ast.Statement:
 		a.apply(n, "Raw", nil, n.Raw)
 

internal/engine/postgresql/parse.go

@@ -128,6 +128,11 @@ func Walk(f Visitor, node ast.Node) {
 			Walk(f, n.Table)
 		}
 
+	case *ast.RenameTypeStmt:
+		if n.Type != nil {
+			Walk(f, n.Type)
+		}
+
 	case *ast.Statement:
 		if n.Raw != nil {
 			Walk(f, n.Raw)

internal/sql/ast/rename_type_stmt.go

@@ -340,6 +340,9 @@ func (c *Catalog) Update(stmt ast.Statement) error {
 	case *ast.RenameTableStmt:
 		err = c.renameTable(n)
 
+	case *ast.RenameTypeStmt:
+		err = c.renameType(n)
+
 	}
 	return err
 }