diff --git a/atsf_server/model/main.go b/atsf_server/model/main.go index 45146a5d..c9d65fe0 100644 --- a/atsf_server/model/main.go +++ b/atsf_server/model/main.go @@ -10,6 +10,16 @@ import ( var DB *gorm.DB +func migrateProxyRouteEnableHTTPSColumn(db *gorm.DB) error { + if !db.Migrator().HasTable(&ProxyRoute{}) { + return nil + } + if db.Migrator().HasColumn(&ProxyRoute{}, "enable_https") || !db.Migrator().HasColumn(&ProxyRoute{}, "enable_http_s") { + return nil + } + return db.Migrator().RenameColumn(&ProxyRoute{}, "enable_http_s", "enable_https") +} + func createRootAccountIfNeed() error { var user User //if user.Status != common.UserStatusEnabled { @@ -52,6 +62,9 @@ func InitDB() (err error) { } if err == nil { DB = db + if err = migrateProxyRouteEnableHTTPSColumn(db); err != nil { + return err + } err := db.AutoMigrate(&File{}) if err != nil { return err diff --git a/atsf_server/model/proxy_route.go b/atsf_server/model/proxy_route.go index 2fab4b6f..d0863fba 100644 --- a/atsf_server/model/proxy_route.go +++ b/atsf_server/model/proxy_route.go @@ -7,7 +7,7 @@ type ProxyRoute struct { Domain string `json:"domain" gorm:"uniqueIndex;size:255;not null"` OriginURL string `json:"origin_url" gorm:"size:2048;not null"` Enabled bool `json:"enabled" gorm:"not null;default:true"` - EnableHTTPS bool `json:"enable_https" gorm:"not null;default:false"` + EnableHTTPS bool `json:"enable_https" gorm:"column:enable_https;not null;default:false"` CertID *uint `json:"cert_id"` RedirectHTTP bool `json:"redirect_http" gorm:"not null;default:false"` CustomHeaders string `json:"custom_headers" gorm:"type:text;not null;default:'[]'"`