From b8354124bebcd3988afd2f41c320f834a69b949e Mon Sep 17 00:00:00 2001 From: obscuren Date: Fri, 26 Sep 2014 13:45:26 +0200 Subject: [PATCH] Added protocol caps accessors --- ethchain/state_manager.go | 1 + ethpipe/js_types.go | 10 +++++++++- peer.go | 4 ++++ 3 files changed, 14 insertions(+), 1 deletion(-) diff --git a/ethchain/state_manager.go b/ethchain/state_manager.go index 888c68bb9..8088f6735 100644 --- a/ethchain/state_manager.go +++ b/ethchain/state_manager.go @@ -32,6 +32,7 @@ type Peer interface { Version() string PingTime() string Connected() *int32 + Caps() *ethutil.Value } type EthManager interface { diff --git a/ethpipe/js_types.go b/ethpipe/js_types.go index 7266a5be4..cf5686a4b 100644 --- a/ethpipe/js_types.go +++ b/ethpipe/js_types.go @@ -1,6 +1,7 @@ package ethpipe import ( + "fmt" "strconv" "strings" @@ -151,6 +152,7 @@ type JSPeer struct { Version string `json:"version"` LastResponse string `json:"lastResponse"` Latency string `json:"latency"` + Caps string `json:"caps"` } func NewJSPeer(peer ethchain.Peer) *JSPeer { @@ -164,7 +166,13 @@ func NewJSPeer(peer ethchain.Peer) *JSPeer { } ipAddress := strings.Join(ip, ".") - return &JSPeer{ref: &peer, Inbound: peer.Inbound(), LastSend: peer.LastSend().Unix(), LastPong: peer.LastPong(), Version: peer.Version(), Ip: ipAddress, Port: int(peer.Port()), Latency: peer.PingTime()} + var caps []string + capsIt := peer.Caps().NewIterator() + for capsIt.Next() { + caps = append(caps, capsIt.Value().Str()) + } + + return &JSPeer{ref: &peer, Inbound: peer.Inbound(), LastSend: peer.LastSend().Unix(), LastPong: peer.LastPong(), Version: peer.Version(), Ip: ipAddress, Port: int(peer.Port()), Latency: peer.PingTime(), Caps: fmt.Sprintf("%v", caps)} } type JSReceipt struct { diff --git a/peer.go b/peer.go index 6f1ad91e3..f1fff59af 100644 --- a/peer.go +++ b/peer.go @@ -813,6 +813,10 @@ func (self *Peer) IsCap(cap string) bool { return false } +func (self *Peer) Caps() *ethutil.Value { + return self.protocolCaps +} + func (p *Peer) String() string { var strBoundType string if p.inbound {