Skip to content

Commit e7939f4

Browse files
Elvis339maru-ava
authored andcommitted
refactor: export PeerSample (#3745)
1 parent 049efef commit e7939f4

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

network/p2p/network.go

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ import (
2424
var (
2525
_ validators.Connector = (*Network)(nil)
2626
_ common.AppHandler = (*Network)(nil)
27-
_ NodeSampler = (*peerSampler)(nil)
27+
_ NodeSampler = (*PeerSampler)(nil)
2828

2929
opLabel = "op"
3030
handlerLabel = "handlerID"
@@ -144,8 +144,8 @@ func (n *Network) NewClient(handlerID uint64, options ...ClientOption) *Client {
144144
sender: n.sender,
145145
router: n.router,
146146
options: &clientOptions{
147-
nodeSampler: &peerSampler{
148-
peers: n.Peers,
147+
nodeSampler: &PeerSampler{
148+
Peers: n.Peers,
149149
},
150150
},
151151
}
@@ -197,12 +197,13 @@ func (p *Peers) Sample(limit int) []ids.NodeID {
197197
return p.set.Sample(limit)
198198
}
199199

200-
type peerSampler struct {
201-
peers *Peers
200+
// PeerSampler implements NodeSampler
201+
type PeerSampler struct {
202+
Peers *Peers
202203
}
203204

204-
func (p peerSampler) Sample(_ context.Context, limit int) []ids.NodeID {
205-
return p.peers.Sample(limit)
205+
func (p PeerSampler) Sample(_ context.Context, limit int) []ids.NodeID {
206+
return p.Peers.Sample(limit)
206207
}
207208

208209
func ProtocolPrefix(handlerID uint64) []byte {

0 commit comments

Comments
 (0)