mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
ryan
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package migrate
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type nodeV16 struct {
|
|
NodeType string `gorm:"column:node_type;not null;default:'edge_node'"`
|
|
RelayBindPort int `gorm:"column:relay_bind_port"`
|
|
}
|
|
|
|
type tunnelV16 struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
}
|
|
|
|
type proxyRouteV16 struct {
|
|
UpstreamType string `gorm:"column:upstream_type;not null;default:'direct'"`
|
|
TunnelID *uint `gorm:"column:tunnel_id"`
|
|
}
|
|
|
|
func init() {
|
|
Register(V16())
|
|
}
|
|
|
|
func V16() Migration {
|
|
return Migration{
|
|
FromVersion: 15,
|
|
ToVersion: 16,
|
|
Migrate: migrateV16,
|
|
Validate: validateV16,
|
|
}
|
|
}
|
|
|
|
func (nodeV16) TableName() string {
|
|
return "nodes"
|
|
}
|
|
|
|
func (tunnelV16) TableName() string {
|
|
return "tunnels"
|
|
}
|
|
|
|
func (proxyRouteV16) TableName() string {
|
|
return "proxy_routes"
|
|
}
|
|
|
|
func migrateV16(ctx Context, db *gorm.DB, backend string) error {
|
|
if err := ctx.ApplyCurrentSchema(db, backend); err != nil {
|
|
return err
|
|
}
|
|
if err := db.AutoMigrate(&tunnelV16{}); err != nil {
|
|
return fmt.Errorf("auto migrate tunnelV16: %w", err)
|
|
}
|
|
if err := db.Exec("UPDATE nodes SET node_type = 'edge_node' WHERE node_type = '' OR node_type IS NULL").Error; err != nil {
|
|
return fmt.Errorf("backfill nodes.node_type: %w", err)
|
|
}
|
|
if err := db.Exec("UPDATE proxy_routes SET upstream_type = 'direct' WHERE upstream_type = '' OR upstream_type IS NULL").Error; err != nil {
|
|
return fmt.Errorf("backfill proxy_routes.upstream_type: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func validateV16(ctx Context, db *gorm.DB, backend string) error {
|
|
if err := ctx.ValidateDatabaseSchemaVersion(db, backend, 15); err != nil {
|
|
return err
|
|
}
|
|
if !db.Migrator().HasColumn(&proxyRouteV16{}, "tunnel_node_id") {
|
|
if db == nil || !db.Migrator().HasTable(&tunnelV16{}) {
|
|
return fmt.Errorf("table tunnels is missing")
|
|
}
|
|
if !db.Migrator().HasColumn(&proxyRouteV16{}, "tunnel_id") {
|
|
return fmt.Errorf("column proxy_routes.tunnel_id is missing")
|
|
}
|
|
}
|
|
if !db.Migrator().HasColumn(&nodeV16{}, "node_type") {
|
|
return fmt.Errorf("column nodes.node_type is missing")
|
|
}
|
|
if !db.Migrator().HasColumn(&proxyRouteV16{}, "upstream_type") {
|
|
return fmt.Errorf("column proxy_routes.upstream_type is missing")
|
|
}
|
|
if !db.Migrator().HasColumn(&nodeV16{}, "relay_bind_port") {
|
|
return fmt.Errorf("column nodes.relay_bind_port is missing")
|
|
}
|
|
return nil
|
|
}
|