Minor fixes

NethermindEth · Sep 30, 2024 · 220d93f · 220d93f
1 parent 3f2fd50
commit 220d93f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git a/p2p/modes.go b/p2p/modes.go
@@ -69,14 +69,6 @@ func (s SyncMode) MarshalText() ([]byte, error) {
 	}
 }
 
-func (mode *SyncMode) UnmarshalText(text []byte) error {
-	switch string(text) {
-	case "full":
-		*mode = FullSync
-	case "snap":
-		*mode = SnapSync
-	default:
-		return fmt.Errorf(`unknown sync mode %q, want "full" or "snap"`, text)
-	}
-	return nil
+func (s *SyncMode) UnmarshalText(text []byte) error {
+	return s.Set(string(text))
 }
diff --git a/p2p/snap_syncer.go b/p2p/snap_syncer.go
@@ -9,15 +9,14 @@ import (
 	"sync/atomic"
 	"time"
 
-	"github.com/NethermindEth/juno/p2p/starknet"
-
 	"github.com/NethermindEth/juno/adapters/core2p2p"
 	"github.com/NethermindEth/juno/adapters/p2p2core"
 	"github.com/NethermindEth/juno/blockchain"
 	"github.com/NethermindEth/juno/core"
 	"github.com/NethermindEth/juno/core/crypto"
 	"github.com/NethermindEth/juno/core/felt"
 	"github.com/NethermindEth/juno/core/trie"
+	"github.com/NethermindEth/juno/p2p/starknet"
 	"github.com/NethermindEth/juno/p2p/starknet/spec"
 	"github.com/NethermindEth/juno/service"
 	"github.com/NethermindEth/juno/starknetdata"