add subscriber

ssvlabs · Jul 17, 2024 · 264a3c3 · 264a3c3
1 parent ad6844d
commit 264a3c3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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[:]))
+}