ft: allow state overrides as method

rahul0tripathi · Jun 18, 2024 · 2d7de23 · 2d7de23
1 parent a59e611
commit 2d7de23
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/services/smelter_rpc.go b/services/smelter_rpc.go
@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 
 	"github.com/ethereum/go-ethereum/common"
+	"github.com/rahul0tripathi/smelter/entity"
 )
 
 type SmelterRpc struct {
@@ -48,3 +49,13 @@ func (s *SmelterRpc) GetState(ctx context.Context) (json.RawMessage, error) {
 
 	return v, nil
 }
+
+func (s *SmelterRpc) SetStateOverrides(ctx context.Context, overrides entity.StateOverrides) error {
+	execCtx, err := s.execStorage.GetOrCreate(ctx)
+	if err != nil {
+		return err
+	}
+
+	execCtx.Overrides = overrides
+	return nil
+}