mirror of
https://github.com/getAlby/hub.git
synced 2026-08-13 12:33:39 +02:00
ffee8cbcbe
Allows users running Alby Hub on postgres (e.g. Alby Cloud) to create a migration file from Settings -> Migrate Alby Hub. The contents of the postgres database are copied into a temporary local sqlite database which is included in the migration file, so it can be imported into a fresh sqlite-based hub. - extract the db_migrate CLI copy logic into a shared db.MigrateDB - also copy the swaps and forwards tables (previously silently dropped) - only require VSS in the source when migrating to postgres - show a hint on the migrate page when running on postgres - show database storage type and VSS status on the about page - don't log an error when removing non-existent db files before restore Closes #2500 Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
82 lines
2 KiB
Go
82 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/getAlby/hub/db"
|
|
"github.com/getAlby/hub/logger"
|
|
)
|
|
|
|
func main() {
|
|
var fromDSN, toDSN string
|
|
|
|
logger.Init(strconv.Itoa(int(logrus.DebugLevel)))
|
|
|
|
flag.StringVar(&fromDSN, "from", "", "source DSN")
|
|
flag.StringVar(&toDSN, "to", "", "destination DSN")
|
|
|
|
flag.Parse()
|
|
|
|
if fromDSN == "" || toDSN == "" {
|
|
flag.Usage()
|
|
logger.Logger.Error("missing DSN")
|
|
os.Exit(1)
|
|
}
|
|
|
|
stopDB := func(d *gorm.DB) {
|
|
if err := db.Stop(d); err != nil {
|
|
logger.Logger.WithError(err).Error("failed to close database")
|
|
}
|
|
}
|
|
|
|
logger.Logger.Info("opening source DB...")
|
|
fromDB, err := db.NewDB(fromDSN, false)
|
|
if err != nil {
|
|
logger.Logger.WithError(err).Error("failed to open source database")
|
|
os.Exit(1)
|
|
}
|
|
defer stopDB(fromDB)
|
|
|
|
logger.Logger.Info("opening destination DB...")
|
|
toDB, err := db.NewDB(toDSN, false)
|
|
if err != nil {
|
|
logger.Logger.WithError(err).Error("failed to open destination database")
|
|
os.Exit(1)
|
|
}
|
|
defer stopDB(toDB)
|
|
|
|
// When migrating to Postgres (e.g. a cloud deployment) the node data must
|
|
// be stored in VSS, since only the database is migrated by this tool.
|
|
if toDB.Dialector.Name() == "postgres" {
|
|
var vssConfig db.UserConfig
|
|
result := fromDB.Where("key = ?", "LdkVssEnabled").First(&vssConfig)
|
|
if result.Error != nil {
|
|
if result.Error == gorm.ErrRecordNotFound {
|
|
logger.Logger.Error("LdkVssEnabled config not found in source DB. Migration will not proceed.")
|
|
} else {
|
|
logger.Logger.WithError(result.Error).Error("failed to query LdkVssEnabled config from source DB")
|
|
}
|
|
os.Exit(1)
|
|
}
|
|
|
|
if vssConfig.Value != "true" {
|
|
logger.Logger.Error("VSS is not enabled in the source DB (LdkVssEnabled is not 'true'). Migration will not proceed.")
|
|
os.Exit(1)
|
|
}
|
|
logger.Logger.Info("LdkVssEnabled check passed.")
|
|
}
|
|
|
|
logger.Logger.Info("migrating...")
|
|
err = db.MigrateDB(fromDB, toDB)
|
|
if err != nil {
|
|
logger.Logger.WithError(err).Error("failed to migrate database")
|
|
os.Exit(1)
|
|
}
|
|
|
|
logger.Logger.Info("migration complete")
|
|
}
|