adding filtration for state contract

rarimo · Mar 28, 2024 · 937bc5e · 937bc5e
1 parent cc43ba6
commit 937bc5e
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 9 deletions.
diff --git a/config-example.yaml b/config-example.yaml
@@ -24,5 +24,4 @@ evm:
 
 relay:
   catchup_disabled: true
-  issuer_id: [ '', '' ]
-  disable_filtration: true
+  source_contract: 0x...
diff --git a/internal/config/relay.go b/internal/config/relay.go
@@ -7,7 +7,8 @@ import (
 )
 
 type RelayConfig struct {
-	CatchupDisabled bool `fig:"catchup_disabled"`
+	CatchupDisabled bool   `fig:"catchup_disabled"`
+	SourceContract  string `fig:"source_contract"`
 }
 
 func (c *config) Relay() RelayConfig {

diff --git a/internal/services/ingester/state_transfer.go b/internal/services/ingester/state_transfer.go
@@ -14,18 +14,20 @@ import (
 )
 
 type stateIngester struct {
-	log        *logan.Entry
-	rarimocore rarimocore.QueryClient
-	storage    *pg.Storage
+	log            *logan.Entry
+	rarimocore     rarimocore.QueryClient
+	storage        *pg.Storage
+	sourceContract string
 }
 
 var _ Processor = &stateIngester{}
 
 func NewStateIngester(cfg config.Config) Processor {
 	return &stateIngester{
-		log:        cfg.Log(),
-		rarimocore: rarimocore.NewQueryClient(cfg.Cosmos()),
-		storage:    pg.New(cfg.DB()),
+		log:            cfg.Log(),
+		rarimocore:     rarimocore.NewQueryClient(cfg.Cosmos()),
+		storage:        pg.New(cfg.DB()),
+		sourceContract: cfg.Relay().SourceContract,
 	}
 }
 
@@ -106,6 +108,11 @@ func (s *stateIngester) trySave(ctx context.Context, operation rarimocore.Operat
 			})
 		}
 
+		if op.Contract != s.sourceContract {
+			s.log.WithField("operation_index", operation.Index).Info("Invalid state contract. Skipping")
+			return nil
+		}
+
 		err = s.storage.StateQ().UpsertCtx(ctx, &data.State{
 			ID:        op.State,
 			Operation: operation.Index,