[优化] Phase2

openflare_server/model/migrations.go

@@ -74,6 +74,10 @@ func (databaseSchemaMigrationContext) ValidateDatabaseSchemaVersion(db *gorm.DB,
 		return validateDatabaseSchemaV13(db, backend)
 	case 14:
 		return validateDatabaseSchemaV14(db, backend)
+	case 15:
+		return validateDatabaseSchemaV15(db, backend)
+	case 16:
+		return validateDatabaseSchemaV16(db, backend)
 	default:
 		return fmt.Errorf("database schema validation for v%d is not defined", version)
 	}
@@ -1140,6 +1144,32 @@ func validateDatabaseSchemaV14(db *gorm.DB, backend string) error {
 	return nil
 }
 
+func validateDatabaseSchemaV15(db *gorm.DB, backend string) error {
+	if err := validateDatabaseSchemaV14(db, backend); err != nil {
+		return err
+	}
+	if !db.Migrator().HasColumn(&Node{}, "ip_manual_override") {
+		return fmt.Errorf("column nodes.ip_manual_override is missing")
+	}
+	return nil
+}
+
+func validateDatabaseSchemaV16(db *gorm.DB, backend string) error {
+	if err := validateDatabaseSchemaV15(db, backend); err != nil {
+		return err
+	}
+	if !db.Migrator().HasTable(&Tunnel{}) {
+		return fmt.Errorf("table tunnels is missing")
+	}
+	if !db.Migrator().HasColumn(&Node{}, "node_type") {
+		return fmt.Errorf("column nodes.node_type is missing")
+	}
+	if !db.Migrator().HasColumn(&ProxyRoute{}, "upstream_type") {
+		return fmt.Errorf("column proxy_routes.upstream_type is missing")
+	}
+	return nil
+}
+
 func databaseSchemaMigrations() []databaseSchemaMigration {
 	ctx := databaseSchemaMigrationContext{}
 	migrations := []databaseSchemaMigration{}