feat: database migrate (#653)

package.json

@@ -1,6 +1,6 @@
 {
   "name": "nebula-graph-studio",
-  "version": "3.7.0",
+  "version": "3.8.0",
   "description": "nebula-graph-studio",
   "private": true,
   "repository": {

server/api/studio/internal/model/db.go

@@ -104,7 +104,13 @@ func InitDB(config *config.Config, db *gorm.DB) {
 	}
 
 	if config.DB.AutoMigrate {
-		err := db.AutoMigrate(
+		migrateTables := []string{"task_infos", "task_effects", "sketches", "schema_snapshots", "favorites", "files", "datasources"}
+		err := MigrateAlterBID(db, migrateTables)
+		if err != nil {
+			zap.L().Fatal(fmt.Sprintf("migrate tables fail: %s", err))
+			panic(err)
+		}
+		err = db.AutoMigrate(
 			&Datasource{},
 			&TaskInfo{},
 			&TaskEffect{},

server/api/studio/internal/model/db_migrate.go

@@ -0,0 +1,30 @@
+package db
+
+import (
+	"gorm.io/gorm"
+)
+
+// Add the `b_id“ field to certain database tables for versions prior to v3.7 (inclusive).
+// We need to perform this operation manually, otherwise the `db.AutoMigrate` method will throw an exception
+func MigrateAlterBID(db *gorm.DB, tables []string) error {
+	for _, table := range tables {
+		isTableExist := db.Migrator().HasTable(table)
+		// if table not exists, skip
+		if !isTableExist {
+			continue
+		}
+		isColumnExist := db.Migrator().HasColumn(table, "b_id")
+		if isColumnExist {
+			continue
+		}
+		err := db.Exec("ALTER TABLE `" + table + "` ADD COLUMN `b_id` CHAR(32) NOT NULL DEFAULT ''").Error
+		if err != nil {
+			return err
+		}
+		err = db.Exec("UPDATE `" + table + "` SET `b_id` = `id`").Error
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}