forked from TrueCloudLab/neoneo-go
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package result
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
type (
|
|
// GetPeers payload for outputting peers in `getpeers` RPC call.
|
|
GetPeers struct {
|
|
Unconnected Peers `json:"unconnected"`
|
|
Connected Peers `json:"connected"`
|
|
Bad Peers `json:"bad"`
|
|
}
|
|
|
|
// Peers represent a slice of peers.
|
|
Peers []Peer
|
|
|
|
// Peer represents the peer.
|
|
Peer struct {
|
|
Address string `json:"address"`
|
|
Port string `json:"port"`
|
|
}
|
|
)
|
|
|
|
// NewGetPeers creates a new GetPeers structure.
|
|
func NewGetPeers() GetPeers {
|
|
return GetPeers{
|
|
Unconnected: []Peer{},
|
|
Connected: []Peer{},
|
|
Bad: []Peer{},
|
|
}
|
|
}
|
|
|
|
// AddUnconnected adds a set of peers to the unconnected peers slice.
|
|
func (g *GetPeers) AddUnconnected(addrs []string) {
|
|
g.Unconnected.addPeers(addrs)
|
|
}
|
|
|
|
// AddConnected adds a set of peers to the connected peers slice.
|
|
func (g *GetPeers) AddConnected(addrs []string) {
|
|
g.Connected.addPeers(addrs)
|
|
}
|
|
|
|
// AddBad adds a set of peers to the bad peers slice.
|
|
func (g *GetPeers) AddBad(addrs []string) {
|
|
g.Bad.addPeers(addrs)
|
|
}
|
|
|
|
// addPeers adds a set of peers to the given peer slice.
|
|
func (p *Peers) addPeers(addrs []string) {
|
|
for i := range addrs {
|
|
addressParts := strings.Split(addrs[i], ":")
|
|
peer := Peer{
|
|
Address: addressParts[0],
|
|
Port: addressParts[1],
|
|
}
|
|
|
|
*p = append(*p, peer)
|
|
}
|
|
}
|