cerc-io/plugeth
18
0
Fork 1
Code Issues 4 Pull Requests Packages Projects Releases Wiki Activity

Get peers returns now both in and outbound peers

Browse Source
This commit is contained in:
obscuren 2014-01-31 11:57:56 +01:00
parent dfa38b3f91
commit da66eddfcc
2 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ethereum.go
View File

@ -144,6 +144,18 @@ func (s *Ethereum) InboundPeers() []*Peer {
return inboundPeers[:length] return inboundPeers[:length]
} }
func (s *Ethereum) InOutPeers() []*Peer {
// Create a new peer slice with at least the length of the total peers
inboundPeers := make([]*Peer, s.peers.Len())
length := 0
eachPeer(s.peers, func(p *Peer, e *list.Element) {
inboundPeers[length] = p
length++
})
return inboundPeers[:length]
}
func (s *Ethereum) Broadcast(msgType ethwire.MsgType, data interface{}) { func (s *Ethereum) Broadcast(msgType ethwire.MsgType, data interface{}) {
msg := ethwire.NewMessage(msgType, data) msg := ethwire.NewMessage(msgType, data)
eachPeer(s.peers, func(p *Peer, e *list.Element) { eachPeer(s.peers, func(p *Peer, e *list.Element) {
@ -151,6 +163,10 @@ func (s *Ethereum) Broadcast(msgType ethwire.MsgType, data interface{}) {
}) })
} }
func (s *Ethereum) Peers() *list.List {
return s.peers
}
func (s *Ethereum) ReapDeadPeers() { func (s *Ethereum) ReapDeadPeers() {
for { for {
eachPeer(s.peers, func(p *Peer, e *list.Element) { eachPeer(s.peers, func(p *Peer, e *list.Element) {

4
peer.go
View File

@ -335,9 +335,9 @@ func (p *Peer) pushHandshake() error {
// Pushes the list of outbound peers to the client when requested // Pushes the list of outbound peers to the client when requested
func (p *Peer) pushPeers() { func (p *Peer) pushPeers() {
outPeers := make([]interface{}, len(p.ethereum.OutboundPeers())) outPeers := make([]interface{}, len(p.ethereum.InOutPeers()))
// Serialise each peer // Serialise each peer
for i, peer := range p.ethereum.OutboundPeers() { for i, peer := range p.ethereum.InOutPeers() {
outPeers[i] = peer.RlpData() outPeers[i] = peer.RlpData()
} }