alihan/fn-serverless
1
0
Fork 0
mirror of https://github.com/fnproject/fn.git synced 2022-10-28 21:29:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix SQL UP migration (#895)

Browse Source
This commit is contained in:
Denis Makogon
2018-03-27 21:28:03 +03:00
committed by Reed Allman
parent 4537a5eb91
commit fe0763825b
1 changed files with 84 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
api/datastore/sql/migrations/10_add_app_id.go
View File

@@ -10,6 +10,56 @@ import (
"github.com/jmoiron/sqlx" "github.com/jmoiron/sqlx"
) )
var sqlStatements = [...]string{`CREATE TABLE IF NOT EXISTS routes (
app_id varchar(256) NOT NULL,
path varchar(256) NOT NULL,
image varchar(256) NOT NULL,
format varchar(16) NOT NULL,
memory int NOT NULL,
cpus int,
timeout int NOT NULL,
idle_timeout int NOT NULL,
type varchar(16) NOT NULL,
headers text NOT NULL,
config text NOT NULL,
annotations text NOT NULL,
created_at text,
updated_at varchar(256),
PRIMARY KEY (app_id, path)
);`,
`CREATE TABLE IF NOT EXISTS calls (
created_at varchar(256) NOT NULL,
started_at varchar(256) NOT NULL,
completed_at varchar(256) NOT NULL,
status varchar(256) NOT NULL,
id varchar(256) NOT NULL,
app_id varchar(256) NOT NULL,
path varchar(256) NOT NULL,
stats text,
error text,
PRIMARY KEY (id)
);`,
`CREATE TABLE IF NOT EXISTS logs (
id varchar(256) NOT NULL PRIMARY KEY,
app_id varchar(256) NOT NULL,
log text NOT NULL
);`,
}
var tables = [...]map[string]string{
{"table": "routes", "statement": sqlStatements[0],
"columns": "app_id,path,image,format,memory,cpus,timeout," +
"idle_timeout,type,headers,config,annotations,created_at,updated_at"},
{"table": "calls", "statement": sqlStatements[1],
"columns": "created_at,started_at,completed_at,status,id,app_id,path,stats,error"},
{"table": "logs", "statement": sqlStatements[2],
"columns": "id,app_id,log"},
}
func up10(ctx context.Context, tx *sqlx.Tx) error { func up10(ctx context.Context, tx *sqlx.Tx) error {
addAppIDStatements := []string{ addAppIDStatements := []string{
"ALTER TABLE apps ADD id VARCHAR(256);", "ALTER TABLE apps ADD id VARCHAR(256);",
@@ -69,6 +119,7 @@ func up10(ctx context.Context, tx *sqlx.Tx) error {
} }
} }
if tx.DriverName() != "sqlite3" {
dropAppNameStatements := []string{ dropAppNameStatements := []string{
"ALTER TABLE routes DROP COLUMN app_name;", "ALTER TABLE routes DROP COLUMN app_name;",
"ALTER TABLE calls DROP COLUMN app_name;", "ALTER TABLE calls DROP COLUMN app_name;",
@@ -80,6 +131,30 @@ func up10(ctx context.Context, tx *sqlx.Tx) error {
return err return err
} }
} }
} else {
for _, t := range tables {
statement := t["statement"]
tableName := t["table"]
columns := t["columns"]
_, err = tx.ExecContext(ctx, tx.Rebind(fmt.Sprintf("ALTER TABLE %s RENAME TO old_%s;", tableName, tableName)))
if err != nil {
return err
}
_, err = tx.ExecContext(ctx, tx.Rebind(statement))
if err != nil {
return err
}
_, err = tx.ExecContext(ctx, tx.Rebind(fmt.Sprintf("INSERT INTO %s (%s) SELECT %s FROM old_%s;", tableName, columns, columns, tableName)))
if err != nil {
return err
}
_, err = tx.ExecContext(ctx, tx.Rebind(fmt.Sprintf("DROP TABLE old_%s;", tableName)))
if err != nil {
return err
}
}
}
return nil return nil
} }