diff options
Diffstat (limited to 'rpc/api/net.go')
-rw-r--r-- | rpc/api/net.go | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/rpc/api/net.go b/rpc/api/net.go new file mode 100644 index 000000000..d6888ee4a --- /dev/null +++ b/rpc/api/net.go @@ -0,0 +1,89 @@ +package api + +import ( + "github.com/ethereum/go-ethereum/eth" + "github.com/ethereum/go-ethereum/rpc/codec" + "github.com/ethereum/go-ethereum/rpc/shared" + "github.com/ethereum/go-ethereum/xeth" +) + +const ( + NetApiVersion = "1.0" +) + +var ( + // mapping between methods and handlers + netMapping = map[string]nethandler{ + "net_version": (*netApi).Version, + "net_peerCount": (*netApi).PeerCount, + "net_listening": (*netApi).IsListening, + "net_peers": (*netApi).Peers, + } +) + +// net callback handler +type nethandler func(*netApi, *shared.Request) (interface{}, error) + +// net api provider +type netApi struct { + xeth *xeth.XEth + ethereum *eth.Ethereum + methods map[string]nethandler + codec codec.ApiCoder +} + +// create a new net api instance +func NewNetApi(xeth *xeth.XEth, eth *eth.Ethereum, coder codec.Codec) *netApi { + return &netApi{ + xeth: xeth, + ethereum: eth, + methods: netMapping, + codec: coder.New(nil), + } +} + +// collection with supported methods +func (self *netApi) Methods() []string { + methods := make([]string, len(self.methods)) + i := 0 + for k := range self.methods { + methods[i] = k + i++ + } + return methods +} + +// Execute given request +func (self *netApi) Execute(req *shared.Request) (interface{}, error) { + if callback, ok := self.methods[req.Method]; ok { + return callback(self, req) + } + + return nil, shared.NewNotImplementedError(req.Method) +} + +func (self *netApi) Name() string { + return NetApiName +} + +func (self *netApi) ApiVersion() string { + return NetApiVersion +} + +// Network version +func (self *netApi) Version(req *shared.Request) (interface{}, error) { + return self.xeth.NetworkVersion(), nil +} + +// Number of connected peers +func (self *netApi) PeerCount(req *shared.Request) (interface{}, error) { + return self.xeth.PeerCount(), nil +} + +func (self *netApi) IsListening(req *shared.Request) (interface{}, error) { + return self.xeth.IsListening(), nil +} + +func (self *netApi) Peers(req *shared.Request) (interface{}, error) { + return self.ethereum.PeersInfo(), nil +} |