Create contract indexes

baking-bad · Oct 8, 2023 · 9a0e26f · 9a0e26f
1 parent ca8023f
commit 9a0e26f
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 9 deletions.
diff --git a/cmd/indexer/indexer/indices.go b/cmd/indexer/indexer/indices.go
@@ -44,15 +44,6 @@ func (bi *BlockchainIndexer) createIndices(ctx context.Context) error {
 	if err := bi.Storage.CreateIndex(ctx, "contracts_level_idx", "level", contractModel); err != nil {
 		return err
 	}
-	if err := bi.Storage.CreateIndex(ctx, "contracts_alpha_id_idx", "alpha_id", contractModel); err != nil {
-		return err
-	}
-	if err := bi.Storage.CreateIndex(ctx, "contracts_babylon_id_idx", "babylon_id", contractModel); err != nil {
-		return err
-	}
-	if err := bi.Storage.CreateIndex(ctx, "contracts_jakarta_id_idx", "jakarta_id", contractModel); err != nil {
-		return err
-	}
 
 	// Global constants
 	globalConstant := (*contract.GlobalConstant)(nil)

diff --git a/internal/postgres/core/index.go b/internal/postgres/core/index.go
@@ -85,6 +85,30 @@ func createBaseIndices(ctx context.Context, db bun.IDB) error {
 			Exec(ctx); err != nil {
 			return err
 		}
+		if _, err := db.NewCreateIndex().
+			Model((*contract.Contract)(nil)).
+			IfNotExists().
+			Index("contracts_alpha_id_idx").
+			Column("alpha_id").
+			Exec(ctx); err != nil {
+			return err
+		}
+		if _, err := db.NewCreateIndex().
+			Model((*contract.Contract)(nil)).
+			IfNotExists().
+			Index("contracts_babylon_id_idx").
+			Column("babylon_id").
+			Exec(ctx); err != nil {
+			return err
+		}
+		if _, err := db.NewCreateIndex().
+			Model((*contract.Contract)(nil)).
+			IfNotExists().
+			Index("contracts_jakarta_id_idx").
+			Column("jakarta_id").
+			Exec(ctx); err != nil {
+			return err
+		}
 
 		// Scripts
 		if _, err := db.NewCreateIndex().