mirror of
https://github.com/lightninglabs/lightning-terminal.git
synced 2026-08-13 12:33:36 +02:00
85 lines
2.6 KiB
Go
85 lines
2.6 KiB
Go
//go:build dev
|
|
|
|
package migsets
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
|
|
"github.com/golang-migrate/migrate/v4"
|
|
"github.com/golang-migrate/migrate/v4/database/pgx/v5"
|
|
"github.com/lightninglabs/lightning-terminal/db"
|
|
"github.com/lightningnetwork/lnd/clock"
|
|
"github.com/lightningnetwork/lnd/lnrpc"
|
|
"github.com/lightningnetwork/lnd/sqldb/v2"
|
|
)
|
|
|
|
// MakeMigrationSets creates the migration sets for the dev environments.
|
|
func MakeMigrationSets(ctx context.Context,
|
|
basicClient lnrpc.LightningClient, macPath, litDir, network string,
|
|
clock clock.Clock) []sqldb.MigrationSet {
|
|
|
|
accountsDir := filepath.Dir(macPath)
|
|
networkDir := filepath.Join(litDir, network)
|
|
|
|
// Create the prod migration set.
|
|
migSet := sqldb.MigrationSet{
|
|
TrackingTableName: pgx.DefaultMigrationsTable,
|
|
SQLFileDirectory: "sqlc/migrations",
|
|
SQLFiles: db.SqlSchemas,
|
|
|
|
// LatestMigrationVersion is the latest migration version of the
|
|
// database. This is used to implement downgrade protection for
|
|
// the daemon.
|
|
//
|
|
// NOTE: This MUST be updated when a new migration is added.
|
|
LatestMigrationVersion: db.LatestMigrationVersion,
|
|
Descriptors: db.MakeMigrationDescriptors(),
|
|
|
|
MakeProgrammaticMigrations: func(baseDB *sqldb.BaseDB) (
|
|
map[uint]migrate.ProgrammaticMigrEntry, error) {
|
|
|
|
// Any programmatic migrations added to this map will be
|
|
// executed when the migration number for the uint key
|
|
// is applied. If no entry exists for a given uint, then
|
|
// no programmatic migration will be executed for that
|
|
// migration number.
|
|
res := make(map[uint]migrate.ProgrammaticMigrEntry)
|
|
|
|
res[db.KVDBtoSQLMigVersion] = Mig6ProgrammaticMigration(
|
|
ctx, basicClient, baseDB, accountsDir,
|
|
networkDir, clock, db.KVDBtoSQLMigVersion,
|
|
)
|
|
|
|
return res, nil
|
|
},
|
|
}
|
|
|
|
// If there are no dev migrations in the sqlc/migrations_dev folder, we
|
|
// can return early.
|
|
if !db.HasDevMigrations() {
|
|
return []sqldb.MigrationSet{migSet}
|
|
}
|
|
|
|
// Create the dev migration set.
|
|
migSetDev := sqldb.MigrationSet{
|
|
TrackingTableName: pgx.DefaultMigrationsTable + "_dev",
|
|
SQLFileDirectory: "sqlc/migrations_dev",
|
|
SQLFiles: db.SqlSchemas,
|
|
|
|
// LatestMigrationVersion is the latest migration version of the
|
|
// dev migrations database. This is used to implement downgrade
|
|
// protection for the daemon.
|
|
//
|
|
// NOTE: This MUST be updated when a new dev migration is added.
|
|
LatestMigrationVersion: db.LatestDevMigrationVersion,
|
|
|
|
MakeProgrammaticMigrations: func(_ *sqldb.BaseDB) (
|
|
map[uint]migrate.ProgrammaticMigrEntry, error) {
|
|
|
|
return make(map[uint]migrate.ProgrammaticMigrEntry), nil
|
|
},
|
|
}
|
|
|
|
return []sqldb.MigrationSet{migSet, migSetDev}
|
|
}
|