Merge pull request #348 from wilhelmguo/feature/add_migration_files

Update db migration workflow

Author: wilhelmguo

src/backend/database/migrations/20190312_154209_v1_6.go

@@ -0,0 +1,81 @@
+package main
+
+import (
+	"encoding/json"
+
+	"github.com/astaxie/beego/migration"
+	"github.com/astaxie/beego/orm"
+
+	"github.com/Qihoo360/wayne/src/backend/models"
+	"github.com/Qihoo360/wayne/src/backend/util/logs"
+)
+
+const (
+	UpSQL   = "ALTER TABLE `namespace` ADD COLUMN `kube_namespace` VARCHAR(128) NOT NULL DEFAULT '' AFTER `deleted`, ADD INDEX `namespace_kube_namespace` (`kube_namespace` ASC);"
+	DownSql = "ALTER TABLE `wayne_dev`.`namespace` DROP COLUMN `kube_namespace`, DROP INDEX `namespace_kube_namespace` ;"
+)
+
+type NamespaceMetaData struct {
+	// kubernetes' namespace
+	Namespace string `json:"namespace,omitempty"`
+}
+
+// DO NOT MODIFY
+type V16_20190312_154209 struct {
+	migration.Migration
+}
+
+// DO NOT MODIFY
+func init() {
+	m := &V16_20190312_154209{}
+	m.Created = "20190312_154209"
+
+	migration.Register("V16_20190312_154209", m)
+}
+
+// Run the migrations
+func (m *V16_20190312_154209) Up() {
+	// For migration namespace metadata namespace to kubeNamespace
+	o := orm.NewOrm()
+	_, err := o.Raw(UpSQL).Exec()
+	if err != nil {
+		logs.Error("Exec UpSQL error.", UpSQL, err)
+		return
+	}
+
+	namespaces := []*models.Namespace{}
+	_, err = o.QueryTable(new(models.Namespace)).
+		All(&namespaces)
+	if err != nil {
+		logs.Error("Get all namespaces error.", err)
+		return
+	}
+	for _, namespace := range namespaces {
+		if namespace.MetaData != "" {
+			metadata := &NamespaceMetaData{}
+			err := json.Unmarshal([]byte(namespace.MetaData), metadata)
+			if err != nil {
+				logs.Error("json.Unmarshal namespaces error.", namespace, err)
+				continue
+			}
+			if metadata.Namespace != "" {
+				namespace.KubeNamespace = metadata.Namespace
+				_, err := o.Update(namespace)
+				if err != nil {
+					logs.Error("update namespaces error.", namespace, err)
+					continue
+				}
+			}
+		}
+	}
+
+	// use m.SQL("CREATE TABLE ...") to make schema update
+
+}
+
+// Reverse the migrations
+func (m *V16_20190312_154209) Down() {
+	// use m.SQL("DROP TABLE ...") to reverse schema update
+	m.SQL(DownSql)
+
+}