mirror of https://github.com/lightninglabs/loop
loopd: Run migration if boltdb exists
parent
69b4df0d9b
commit
a2abcd07d8
@ -0,0 +1,92 @@
|
||||
package loopd
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/lightninglabs/lndclient"
|
||||
"github.com/lightninglabs/loop/loopdb"
|
||||
"github.com/lightningnetwork/lnd/lnrpc"
|
||||
)
|
||||
|
||||
// migrateBoltdb migrates the boltdb to sqlite.
|
||||
func migrateBoltdb(ctx context.Context, cfg *Config) error {
|
||||
// First get the chain params.
|
||||
chainParams, err := lndclient.Network(cfg.Network).ChainParams()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// First open the bolt db.
|
||||
boltdb, err := loopdb.NewBoltSwapStore(cfg.DataDir, chainParams)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer boltdb.Close()
|
||||
|
||||
var db loopdb.SwapStore
|
||||
switch cfg.DatabaseBackend {
|
||||
case DatabaseBackendSqlite:
|
||||
log.Infof("Opening sqlite3 database at: %v",
|
||||
cfg.Sqlite.DatabaseFileName)
|
||||
db, err = loopdb.NewSqliteStore(
|
||||
cfg.Sqlite, chainParams,
|
||||
)
|
||||
|
||||
case DatabaseBackendPostgres:
|
||||
log.Infof("Opening postgres database at: %v",
|
||||
cfg.Postgres.DSN(true))
|
||||
db, err = loopdb.NewPostgresStore(
|
||||
cfg.Postgres, chainParams,
|
||||
)
|
||||
|
||||
default:
|
||||
return fmt.Errorf("unknown database backend: %s",
|
||||
cfg.DatabaseBackend)
|
||||
}
|
||||
if err != nil {
|
||||
return fmt.Errorf("unable to open database: %v", err)
|
||||
}
|
||||
|
||||
defer db.Close()
|
||||
|
||||
// Create a new migrator manager.
|
||||
migrator := loopdb.NewMigratorManager(boltdb, db)
|
||||
|
||||
// Run the migration.
|
||||
err = migrator.RunMigrations(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// If the migration was successfull we'll rename the bolt db to
|
||||
// loop.db.bk.
|
||||
err = os.Rename(
|
||||
filepath.Join(cfg.DataDir, "loop.db"),
|
||||
filepath.Join(cfg.DataDir, "loop.db.bk"),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// needSqlMigration checks if the boltdb exists at it's default location
|
||||
// and returns true if it does.
|
||||
func needSqlMigration(cfg *Config) bool {
|
||||
// First check if the data directory exists.
|
||||
if !lnrpc.FileExists(cfg.DataDir) {
|
||||
return false
|
||||
}
|
||||
|
||||
// Now we'll check if the bolt db exists.
|
||||
if !lnrpc.FileExists(filepath.Join(cfg.DataDir, "loop.db")) {
|
||||
return false
|
||||
}
|
||||
|
||||
// If both the folder and the bolt db exist, we'll return true.
|
||||
return true
|
||||
}
|
Loading…
Reference in New Issue