Add support for DDL

Author: kyleconroy

parser.go

@@ -51,6 +51,40 @@ func parse(s *postgres.Schema, tree pg.ParsetreeList) {
 			continue
 		}
 		switch n := raw.Stmt.(type) {
+		case nodes.AlterTableStmt:
+			idx := -1
+			for i, table := range s.Tables {
+				if table.Name == *n.Relation.Relname {
+					idx = i
+				}
+			}
+			if idx < 0 {
+				panic("could not find table " + *n.Relation.Relname)
+			}
+
+			for _, cmd := range n.Cmds.Items {
+				switch cmd := cmd.(type) {
+				case nodes.AlterTableCmd:
+					switch cmd.Subtype {
+					case nodes.AT_AddColumn:
+						switch n := cmd.Def.(type) {
+						case nodes.ColumnDef:
+							s.Tables[idx].Columns = append(s.Tables[idx].Columns, postgres.Column{
+								Name:    *n.Colname,
+								Type:    join(n.TypeName.Names, "."),
+								GoName:  structName(*n.Colname),
+								NotNull: isNotNull(n),
+							})
+						}
+					case nodes.AT_DropColumn:
+						for i, c := range s.Tables[idx].Columns {
+							if c.Name == *cmd.Name {
+								s.Tables[idx].Columns = append(s.Tables[idx].Columns[:i], s.Tables[idx].Columns[i+1:]...)
+							}
+						}
+					}
+				}
+			}
 		case nodes.CreateStmt:
 			table := postgres.Table{
 				Name:   *n.Relation.Relname,
@@ -69,6 +103,21 @@ func parse(s *postgres.Schema, tree pg.ParsetreeList) {
 				}
 			}
 			s.Tables = append(s.Tables, table)
+		case nodes.RenameStmt:
+			switch n.RenameType {
+			case nodes.OBJECT_TABLE:
+				idx := -1
+				for i, table := range s.Tables {
+					if table.Name == *n.Relation.Relname {
+						idx = i
+					}
+				}
+				if idx < 0 {
+					panic("could not find table " + *n.Relation.Relname)
+				}
+				s.Tables[idx].Name = *n.Newname
+				s.Tables[idx].GoName = structName(*n.Newname)
+			}
 		default:
 			// spew.Dump(n)
 		}

testdata/ondeck/db.go

@@ -13,12 +13,12 @@ type City struct {
 
 type Venue struct {
 	ID              int
-	CreatedAt       time.Time
 	Slug            string
 	Name            string
 	City            sql.NullString
 	SpotifyPlaylist string
 	SongkickID      sql.NullString
+	CreatedAt       time.Time
 }
 
 type dbtx interface {
@@ -192,7 +192,7 @@ func (q *Queries) GetCity(ctx context.Context, slug string) (City, error) {
 }
 
 const getVenue = `-- name: GetVenue :one
-SELECT id, created_at, slug, name, city, spotify_playlist, songkick_id
+SELECT id, slug, name, city, spotify_playlist, songkick_id, created_at
 FROM venue
 WHERE slug = $1 AND city = $2
 `
@@ -208,7 +208,7 @@ func (q *Queries) GetVenue(ctx context.Context, slug string, city string) (Venue
 		row = q.db.QueryRowContext(ctx, getVenue, slug, city)
 	}
 	var i Venue
-	err := row.Scan(&i.ID, &i.CreatedAt, &i.Slug, &i.Name, &i.City, &i.SpotifyPlaylist, &i.SongkickID)
+	err := row.Scan(&i.ID, &i.Slug, &i.Name, &i.City, &i.SpotifyPlaylist, &i.SongkickID, &i.CreatedAt)
 	return i, err
 }
 
@@ -251,7 +251,7 @@ func (q *Queries) ListCities(ctx context.Context) ([]City, error) {
 }
 
 const listVenues = `-- name: ListVenues :many
-SELECT id, created_at, slug, name, city, spotify_playlist, songkick_id
+SELECT id, slug, name, city, spotify_playlist, songkick_id, created_at
 FROM venue
 WHERE city = $1
 ORDER BY name
@@ -275,7 +275,7 @@ func (q *Queries) ListVenues(ctx context.Context, city string) ([]Venue, error)
 	items := []Venue{}
 	for rows.Next() {
 		var i Venue
-		if err := rows.Scan(&i.ID, &i.CreatedAt, &i.Slug, &i.Name, &i.City, &i.SpotifyPlaylist, &i.SongkickID); err != nil {
+		if err := rows.Scan(&i.ID, &i.Slug, &i.Name, &i.City, &i.SpotifyPlaylist, &i.SongkickID, &i.CreatedAt); err != nil {
 			return nil, err
 		}
 		items = append(items, i)

testdata/ondeck/schema/0002_venue.sql

@@ -1,6 +1,6 @@
-CREATE TABLE venue (
+CREATE TABLE venues (
     id               SERIAL primary key,
-    created_at       TIMESTAMP NOT NULL DEFAULT NOW(),
+    dropped          text,
     slug             text not null,
     name             text not null,
     city             text references city(slug),

testdata/ondeck/schema/0003_add_column.sql

@@ -0,0 +1,3 @@
+ALTER TABLE venues RENAME TO venue;
+ALTER TABLE venue ADD COLUMN created_at TIMESTAMP NOT NULL DEFAULT NOW();
+ALTER TABLE venue DROP COLUMN dropped;