fix tool

streamingfast · Aug 8, 2024 · 9f0a708 · 9f0a708
1 parent 066c955
commit 9f0a708
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/injective/cmd/fireinjective/tools_fix_unknown_type_blocks.go b/injective/cmd/fireinjective/tools_fix_unknown_type_blocks.go
@@ -15,9 +15,9 @@ import (
 	"github.com/streamingfast/dstore"
 	firecore "github.com/streamingfast/firehose-core"
 	pbcosmos "github.com/streamingfast/firehose-cosmos/cosmos/pb/sf/cosmos/type/v2"
+	"github.com/streamingfast/firehose-cosmos/injective"
 	"github.com/streamingfast/logging"
 	"go.uber.org/zap"
-	proto "google.golang.org/protobuf/proto"
 	"google.golang.org/protobuf/types/known/anypb"
 )
 
@@ -105,7 +105,7 @@ func fixUnknownTypeBlocksRunE(zlog *zap.Logger, tracer logging.Tracer) firecore.
 				blockCount++
 
 				injectiveBlock := &pbcosmos.Block{}
-				err = proto.Unmarshal(currentBlock.Payload.Value, injectiveBlock)
+				err = injective.UnmarshallerDiscardUnknown.Unmarshal(currentBlock.Payload.Value, injectiveBlock)
 				if err != nil {
 					return fmt.Errorf("unmarshaling block: %w", err)
 				}

diff --git a/injective/conversion.go b/injective/conversion.go
@@ -14,7 +14,7 @@ import (
 	"google.golang.org/protobuf/proto"
 )
 
-var unmarshallerDiscardUnknown = &proto.UnmarshalOptions{
+var UnmarshallerDiscardUnknown = &proto.UnmarshalOptions{
 	DiscardUnknown: true,
 }
 
@@ -54,7 +54,7 @@ func protoFlip(origin cosmoProto.Message, target proto.Message) error {
 		return fmt.Errorf("mashalling origin object %T: %w", data, err)
 	}
 
-	err = unmarshallerDiscardUnknown.Unmarshal(data, target)
+	err = UnmarshallerDiscardUnknown.Unmarshal(data, target)
 	if err != nil {
 		if e, ok := origin.(*abci.ResponseDeliverTx); ok {
 			fmt.Println("event data:", hex.EncodeToString(data))