Disable migrations in API

celenium-io · Nov 4, 2024 · 2742ef0 · 2742ef0
1 parent 6c0b40f
commit 2742ef0
Show file tree

Hide file tree

Showing 10 changed files with 20 additions and 13 deletions.
diff --git a/cmd/api/init.go b/cmd/api/init.go
@@ -261,7 +261,7 @@ func initDispatcher(ctx context.Context, db postgres.Storage) {
 func initDatabase(cfg config.Database, viewsDir string) postgres.Storage {
 	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer cancel()
-	db, err := postgres.Create(ctx, cfg, viewsDir)
+	db, err := postgres.Create(ctx, cfg, viewsDir, false)
 	if err != nil {
 		panic(err)
 	}

diff --git a/cmd/quotes/main.go b/cmd/quotes/main.go
@@ -41,7 +41,7 @@ func main() {
 		return
 	}
 
-	pg, err := postgres.Create(ctx, cfg.Database, cfg.Indexer.ScriptsDir)
+	pg, err := postgres.Create(ctx, cfg.Database, cfg.Indexer.ScriptsDir, false)
 	if err != nil {
 		log.Panic().Err(err).Msg("can't create database connection")
 		return

diff --git a/internal/storage/postgres/block_stats_test.go b/internal/storage/postgres/block_stats_test.go
@@ -47,7 +47,7 @@ func (s *BlockStatsTestSuite) SetupSuite() {
 		Password: s.psqlContainer.Config.Password,
 		Host:     s.psqlContainer.Config.Host,
 		Port:     s.psqlContainer.MappedPort().Int(),
-	}, "../../../database")
+	}, "../../../database", false)
 	s.Require().NoError(err)
 	s.storage = strg
 

diff --git a/internal/storage/postgres/core.go b/internal/storage/postgres/core.go
@@ -57,8 +57,12 @@ type Storage struct {
 }
 
 // Create -
-func Create(ctx context.Context, cfg config.Database, scriptsDir string) (Storage, error) {
-	strg, err := postgres.Create(ctx, cfg, initDatabase)
+func Create(ctx context.Context, cfg config.Database, scriptsDir string, withMigrations bool) (Storage, error) {
+	init := initDatabase
+	if withMigrations {
+		init = initDatabaseWithMigrations
+	}
+	strg, err := postgres.Create(ctx, cfg, init)
 	if err != nil {
 		return Storage{}, err
 	}
@@ -145,10 +149,13 @@ func initDatabase(ctx context.Context, conn *database.Bun) error {
 		return errors.Wrap(err, "create hypertables")
 	}
 
-	if err := createIndices(ctx, conn); err != nil {
+	return createIndices(ctx, conn)
+}
+
+func initDatabaseWithMigrations(ctx context.Context, conn *database.Bun) error {
+	if err := initDatabase(ctx, conn); err != nil {
 		return err
 	}
-
 	return migrateDatabase(ctx, conn)
 }
 

diff --git a/internal/storage/postgres/stats_test.go b/internal/storage/postgres/stats_test.go
@@ -46,7 +46,7 @@ func (s *StatsTestSuite) SetupSuite() {
 		Password: s.psqlContainer.Config.Password,
 		Host:     s.psqlContainer.Config.Host,
 		Port:     s.psqlContainer.MappedPort().Int(),
-	}, "../../../database")
+	}, "../../../database", false)
 	s.Require().NoError(err)
 	s.storage = strg
 

diff --git a/internal/storage/postgres/storage_test.go b/internal/storage/postgres/storage_test.go
@@ -51,7 +51,7 @@ func (s *StorageTestSuite) SetupSuite() {
 		Password: s.psqlContainer.Config.Password,
 		Host:     s.psqlContainer.Config.Host,
 		Port:     s.psqlContainer.MappedPort().Int(),
-	}, "../../../database")
+	}, "../../../database", false)
 	s.Require().NoError(err)
 	s.storage = strg
 

diff --git a/internal/storage/postgres/transaction_test.go b/internal/storage/postgres/transaction_test.go
@@ -52,7 +52,7 @@ func (s *TransactionTestSuite) SetupSuite() {
 		Password: s.psqlContainer.Config.Password,
 		Host:     s.psqlContainer.Config.Host,
 		Port:     s.psqlContainer.MappedPort().Int(),
-	}, "../../../database")
+	}, "../../../database", false)
 	s.Require().NoError(err)
 	s.storage = strg
 }

diff --git a/pkg/indexer/indexer.go b/pkg/indexer/indexer.go
@@ -43,7 +43,7 @@ type Indexer struct {
 }
 
 func New(ctx context.Context, cfg config.Config, stopperModule modules.Module) (Indexer, error) {
-	pg, err := postgres.Create(ctx, cfg.Database, cfg.Indexer.ScriptsDir)
+	pg, err := postgres.Create(ctx, cfg.Database, cfg.Indexer.ScriptsDir, true)
 	if err != nil {
 		return Indexer{}, errors.Wrap(err, "while creating pg context")
 	}

diff --git a/pkg/indexer/rollback/rollback_test.go b/pkg/indexer/rollback/rollback_test.go
@@ -57,7 +57,7 @@ func (s *ModuleTestSuite) SetupSuite() {
 		Password: s.psqlContainer.Config.Password,
 		Host:     s.psqlContainer.Config.Host,
 		Port:     s.psqlContainer.MappedPort().Int(),
-	}, "../../../database")
+	}, "../../../database", false)
 	s.Require().NoError(err)
 	s.storage = st
 }

diff --git a/pkg/indexer/storage/storage_test.go b/pkg/indexer/storage/storage_test.go
@@ -52,7 +52,7 @@ func (s *ModuleTestSuite) SetupSuite() {
 		Password: s.psqlContainer.Config.Password,
 		Host:     s.psqlContainer.Config.Host,
 		Port:     s.psqlContainer.MappedPort().Int(),
-	}, "../../../database")
+	}, "../../../database", false)
 	s.Require().NoError(err)
 	s.storage = strg
 }