From fe0763825bfd4b6cb26f0173dfddfacb7d2e6fb7 Mon Sep 17 00:00:00 2001
From: Denis Makogon <lildee1991@gmail.com>
Date: Tue, 27 Mar 2018 21:28:03 +0300
Subject: [PATCH] Fix SQL UP migration (#895)

---
 api/datastore/sql/migrations/10_add_app_id.go | 93 +++++++++++++++++--
 1 file changed, 84 insertions(+), 9 deletions(-)

diff --git a/api/datastore/sql/migrations/10_add_app_id.go b/api/datastore/sql/migrations/10_add_app_id.go
index fe421390d..facc23830 100644
--- a/api/datastore/sql/migrations/10_add_app_id.go
+++ b/api/datastore/sql/migrations/10_add_app_id.go
@@ -10,6 +10,56 @@ import (
 	"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 {
 	addAppIDStatements := []string{
 		"ALTER TABLE apps ADD id VARCHAR(256);",
@@ -69,15 +119,40 @@ func up10(ctx context.Context, tx *sqlx.Tx) error {
 		}
 	}
 
-	dropAppNameStatements := []string{
-		"ALTER TABLE routes DROP COLUMN app_name;",
-		"ALTER TABLE calls DROP COLUMN app_name;",
-		"ALTER TABLE logs DROP COLUMN app_name;",
-	}
-	for _, statement := range dropAppNameStatements {
-		_, err := tx.ExecContext(ctx, statement)
-		if err != nil {
-			return err
+	if tx.DriverName() != "sqlite3" {
+		dropAppNameStatements := []string{
+			"ALTER TABLE routes DROP COLUMN app_name;",
+			"ALTER TABLE calls DROP COLUMN app_name;",
+			"ALTER TABLE logs DROP COLUMN app_name;",
+		}
+		for _, statement := range dropAppNameStatements {
+			_, err := tx.ExecContext(ctx, statement)
+			if err != nil {
+				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
+			}
 		}
 	}