53 lines
1,020 B
Go
53 lines
1,020 B
Go
package migrate
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/go-pg/migrations"
|
|
)
|
|
|
|
const profileTable = `
|
|
CREATE TABLE profiles (
|
|
id serial NOT NULL,
|
|
created_at timestamp with time zone NOT NULL DEFAULT current_timestamp,
|
|
updated_at timestamp with time zone NOT NULL DEFAULT current_timestamp,
|
|
account_id int NOT NULL REFERENCES accounts(id),
|
|
theme text NOT NULL DEFAULT 'default',
|
|
PRIMARY KEY (id)
|
|
)`
|
|
|
|
const bootstrapAccountProfiles = `
|
|
INSERT INTO profiles(account_id) VALUES(1);
|
|
INSERT INTO profiles(account_id) VALUES(2);
|
|
`
|
|
|
|
func init() {
|
|
up := []string{
|
|
profileTable,
|
|
bootstrapAccountProfiles,
|
|
}
|
|
|
|
down := []string{
|
|
`DROP TABLE profiles`,
|
|
}
|
|
|
|
migrations.Register(func(db migrations.DB) error {
|
|
fmt.Println("create profile table")
|
|
for _, q := range up {
|
|
_, err := db.Exec(q)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}, func(db migrations.DB) error {
|
|
fmt.Println("drop profile table")
|
|
for _, q := range down {
|
|
_, err := db.Exec(q)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|