mirror of
https://github.com/status-im/go-waku.git
synced 2025-01-27 14:05:18 +00:00
chore: remove --store-message-db-vacuum (#883)
This commit is contained in:
parent
712febd32f
commit
ff94b1faf0
@ -330,12 +330,6 @@ var (
|
|||||||
Destination: &options.Store.DatabaseURL,
|
Destination: &options.Store.DatabaseURL,
|
||||||
EnvVars: []string{"WAKUNODE2_STORE_MESSAGE_DB_URL"},
|
EnvVars: []string{"WAKUNODE2_STORE_MESSAGE_DB_URL"},
|
||||||
})
|
})
|
||||||
StoreMessageDBVacuum = altsrc.NewBoolFlag(&cli.BoolFlag{
|
|
||||||
Name: "store-message-db-vacuum",
|
|
||||||
Usage: "Enable database vacuuming at start.",
|
|
||||||
Destination: &options.Store.Vacuum,
|
|
||||||
EnvVars: []string{"WAKUNODE2_STORE_MESSAGE_DB_VACUUM"},
|
|
||||||
})
|
|
||||||
StoreMessageDBMigration = altsrc.NewBoolFlag(&cli.BoolFlag{
|
StoreMessageDBMigration = altsrc.NewBoolFlag(&cli.BoolFlag{
|
||||||
Name: "store-message-db-migration",
|
Name: "store-message-db-migration",
|
||||||
Usage: "Enable database migration at start.",
|
Usage: "Enable database migration at start.",
|
||||||
|
@ -65,7 +65,6 @@ func main() {
|
|||||||
StoreMessageDBURL,
|
StoreMessageDBURL,
|
||||||
StoreMessageRetentionTime,
|
StoreMessageRetentionTime,
|
||||||
StoreMessageRetentionCapacity,
|
StoreMessageRetentionCapacity,
|
||||||
StoreMessageDBVacuum,
|
|
||||||
StoreMessageDBMigration,
|
StoreMessageDBMigration,
|
||||||
FilterFlag,
|
FilterFlag,
|
||||||
FilterNode,
|
FilterNode,
|
||||||
|
@ -110,9 +110,7 @@ func Execute(options NodeOptions) error {
|
|||||||
var db *sql.DB
|
var db *sql.DB
|
||||||
var migrationFn func(*sql.DB) error
|
var migrationFn func(*sql.DB) error
|
||||||
if requiresDB(options) && options.Store.Migration {
|
if requiresDB(options) && options.Store.Migration {
|
||||||
dbSettings := dbutils.DBSettings{
|
dbSettings := dbutils.DBSettings{}
|
||||||
Vacuum: options.Store.Vacuum,
|
|
||||||
}
|
|
||||||
db, migrationFn, err = dbutils.ExtractDBAndMigration(options.Store.DatabaseURL, dbSettings, logger)
|
db, migrationFn, err = dbutils.ExtractDBAndMigration(options.Store.DatabaseURL, dbSettings, logger)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nonRecoverErrorMsg("could not connect to DB: %w", err)
|
return nonRecoverErrorMsg("could not connect to DB: %w", err)
|
||||||
|
@ -81,7 +81,6 @@ type StoreOptions struct {
|
|||||||
RetentionMaxMessages int
|
RetentionMaxMessages int
|
||||||
//ResumeNodes []multiaddr.Multiaddr
|
//ResumeNodes []multiaddr.Multiaddr
|
||||||
Nodes []multiaddr.Multiaddr
|
Nodes []multiaddr.Multiaddr
|
||||||
Vacuum bool
|
|
||||||
Migration bool
|
Migration bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -13,7 +13,7 @@ import (
|
|||||||
|
|
||||||
func MemoryDB(t *testing.T) *persistence.DBStore {
|
func MemoryDB(t *testing.T) *persistence.DBStore {
|
||||||
var db *sql.DB
|
var db *sql.DB
|
||||||
db, err := sqlite.NewDB(":memory:", false, utils.Logger())
|
db, err := sqlite.NewDB(":memory:", utils.Logger())
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
||||||
|
@ -128,7 +128,7 @@ func NewNode(configJSON string) error {
|
|||||||
if *config.EnableStore {
|
if *config.EnableStore {
|
||||||
var db *sql.DB
|
var db *sql.DB
|
||||||
var migrationFn func(*sql.DB) error
|
var migrationFn func(*sql.DB) error
|
||||||
db, migrationFn, err = dbutils.ExtractDBAndMigration(*config.DatabaseURL, dbutils.DBSettings{Vacuum: true}, utils.Logger())
|
db, migrationFn, err = dbutils.ExtractDBAndMigration(*config.DatabaseURL, dbutils.DBSettings{}, utils.Logger())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -1,10 +1,8 @@
|
|||||||
package postgres
|
package postgres
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/golang-migrate/migrate/v4/database"
|
"github.com/golang-migrate/migrate/v4/database"
|
||||||
"github.com/golang-migrate/migrate/v4/database/pgx"
|
"github.com/golang-migrate/migrate/v4/database/pgx"
|
||||||
@ -15,55 +13,13 @@ import (
|
|||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
)
|
)
|
||||||
|
|
||||||
func executeVacuum(db *sql.DB, logger *zap.Logger) error {
|
|
||||||
logger.Info("starting PostgreSQL database vacuuming")
|
|
||||||
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
errCh := make(chan error)
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
defer cancel()
|
|
||||||
_, err := db.Exec("VACUUM FULL")
|
|
||||||
if err != nil {
|
|
||||||
errCh <- err
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
t := time.NewTicker(2 * time.Minute)
|
|
||||||
defer t.Stop()
|
|
||||||
|
|
||||||
loop:
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-ctx.Done():
|
|
||||||
break loop
|
|
||||||
case err := <-errCh:
|
|
||||||
return err
|
|
||||||
case <-t.C:
|
|
||||||
logger.Info("still vacuuming...")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
logger.Info("finished PostgreSQL database vacuuming")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewDB connects to postgres DB in the specified path
|
// NewDB connects to postgres DB in the specified path
|
||||||
func NewDB(dburl string, shouldVacuum bool, logger *zap.Logger) (*sql.DB, error) {
|
func NewDB(dburl string, logger *zap.Logger) (*sql.DB, error) {
|
||||||
db, err := sql.Open("pgx", dburl)
|
db, err := sql.Open("pgx", dburl)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if shouldVacuum {
|
|
||||||
err := executeVacuum(db, logger)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return db, nil
|
return db, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,11 +1,9 @@
|
|||||||
package sqlite
|
package sqlite
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/golang-migrate/migrate/v4/database"
|
"github.com/golang-migrate/migrate/v4/database"
|
||||||
"github.com/golang-migrate/migrate/v4/database/sqlite3"
|
"github.com/golang-migrate/migrate/v4/database/sqlite3"
|
||||||
@ -32,43 +30,8 @@ func addSqliteURLDefaults(dburl string) string {
|
|||||||
return dburl
|
return dburl
|
||||||
}
|
}
|
||||||
|
|
||||||
func executeVacuum(db *sql.DB, logger *zap.Logger) error {
|
|
||||||
logger.Info("starting sqlite database vacuuming")
|
|
||||||
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
errCh := make(chan error)
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
defer cancel()
|
|
||||||
_, err := db.Exec("VACUUM")
|
|
||||||
if err != nil {
|
|
||||||
errCh <- err
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
t := time.NewTicker(2 * time.Minute)
|
|
||||||
defer t.Stop()
|
|
||||||
|
|
||||||
loop:
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-ctx.Done():
|
|
||||||
break loop
|
|
||||||
case err := <-errCh:
|
|
||||||
return err
|
|
||||||
case <-t.C:
|
|
||||||
logger.Info("still vacuuming...")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
logger.Info("finished sqlite database vacuuming")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewDB creates a sqlite3 DB in the specified path
|
// NewDB creates a sqlite3 DB in the specified path
|
||||||
func NewDB(dburl string, shouldVacuum bool, logger *zap.Logger) (*sql.DB, error) {
|
func NewDB(dburl string, logger *zap.Logger) (*sql.DB, error) {
|
||||||
db, err := sql.Open("sqlite3", addSqliteURLDefaults(dburl))
|
db, err := sql.Open("sqlite3", addSqliteURLDefaults(dburl))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
@ -77,13 +40,6 @@ func NewDB(dburl string, shouldVacuum bool, logger *zap.Logger) (*sql.DB, error)
|
|||||||
// Disable concurrent access as not supported by the driver
|
// Disable concurrent access as not supported by the driver
|
||||||
db.SetMaxOpenConns(1)
|
db.SetMaxOpenConns(1)
|
||||||
|
|
||||||
if shouldVacuum {
|
|
||||||
err := executeVacuum(db, logger)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return db, nil
|
return db, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -22,7 +22,7 @@ func validateDBUrl(val string) error {
|
|||||||
|
|
||||||
// DBSettings hold db specific configuration settings required during the db initialization
|
// DBSettings hold db specific configuration settings required during the db initialization
|
||||||
type DBSettings struct {
|
type DBSettings struct {
|
||||||
Vacuum bool
|
// TODO: add any DB specific setting here
|
||||||
}
|
}
|
||||||
|
|
||||||
// ExtractDBAndMigration will return a database connection, and migration function that should be used depending on a database connection string
|
// ExtractDBAndMigration will return a database connection, and migration function that should be used depending on a database connection string
|
||||||
@ -50,10 +50,10 @@ func ExtractDBAndMigration(databaseURL string, dbSettings DBSettings, logger *za
|
|||||||
dbParams := dbURLParts[1]
|
dbParams := dbURLParts[1]
|
||||||
switch dbEngine {
|
switch dbEngine {
|
||||||
case "sqlite3":
|
case "sqlite3":
|
||||||
db, err = sqlite.NewDB(dbParams, dbSettings.Vacuum, logger)
|
db, err = sqlite.NewDB(dbParams, logger)
|
||||||
migrationFn = sqlite.Migrations
|
migrationFn = sqlite.Migrations
|
||||||
case "postgresql":
|
case "postgresql":
|
||||||
db, err = postgres.NewDB(dbURL, dbSettings.Vacuum, logger)
|
db, err = postgres.NewDB(dbURL, logger)
|
||||||
migrationFn = postgres.Migrations
|
migrationFn = postgres.Migrations
|
||||||
default:
|
default:
|
||||||
err = errors.New("unsupported database engine")
|
err = errors.New("unsupported database engine")
|
||||||
|
@ -70,7 +70,7 @@ func TestConnectionStatusChanges(t *testing.T) {
|
|||||||
err = node2.Start(ctx)
|
err = node2.Start(ctx)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
db, err := sqlite.NewDB(":memory:", false, utils.Logger())
|
db, err := sqlite.NewDB(":memory:", utils.Logger())
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
@ -242,7 +242,7 @@ func TestDecoupledStoreFromRelay(t *testing.T) {
|
|||||||
defer subs[0].Unsubscribe()
|
defer subs[0].Unsubscribe()
|
||||||
|
|
||||||
// NODE2: Filter Client/Store
|
// NODE2: Filter Client/Store
|
||||||
db, err := sqlite.NewDB(":memory:", false, utils.Logger())
|
db, err := sqlite.NewDB(":memory:", utils.Logger())
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
@ -13,7 +13,7 @@ import (
|
|||||||
|
|
||||||
func MemoryDB(t *testing.T) *persistence.DBStore {
|
func MemoryDB(t *testing.T) *persistence.DBStore {
|
||||||
var db *sql.DB
|
var db *sql.DB
|
||||||
db, err := sqlite.NewDB(":memory:", false, utils.Logger())
|
db, err := sqlite.NewDB(":memory:", utils.Logger())
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
dbStore, err := persistence.NewDBStore(prometheus.DefaultRegisterer, utils.Logger(), persistence.WithDB(db), persistence.WithMigrations(sqlite.Migrations))
|
||||||
|
@ -45,7 +45,7 @@ func TestRendezvous(t *testing.T) {
|
|||||||
host1, err := tests.MakeHost(ctx, port1, rand.Reader)
|
host1, err := tests.MakeHost(ctx, port1, rand.Reader)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
db, err := sqlite.NewDB(":memory:", false, utils.Logger())
|
db, err := sqlite.NewDB(":memory:", utils.Logger())
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
err = sqlite.Migrations(db)
|
err = sqlite.Migrations(db)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user