diff --git a/network/p2p/p2p_genesis.go b/network/p2p/p2p_genesis.go index d2d691d455..32022a66d4 100644 --- a/network/p2p/p2p_genesis.go +++ b/network/p2p/p2p_genesis.go @@ -5,6 +5,7 @@ import ( "github.com/pkg/errors" genesisspectypes "github.com/ssvlabs/ssv-spec-pre-cc/types" + spectypes "github.com/ssvlabs/ssv-spec/types" "github.com/ssvlabs/ssv/logging/fields" "github.com/ssvlabs/ssv/network/commons" p2pprotocol "github.com/ssvlabs/ssv/protocol/v2/p2p" @@ -59,3 +60,8 @@ func (p *GenesisP2p) Broadcast(message *genesisspectypes.SSVMessage) error { } return nil } + +// Subscribe subscribes to validator subnet +func (n *GenesisP2p) Subscribe(pk genesisspectypes.ValidatorPK) error { + return n.Network.Subscribe(spectypes.ValidatorPK(pk[:])) +}