75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
|
package rpc
|
||
|
|
|
||
|
|
// router implements logic for routing
|
||
|
|
// JSON-RPC requests either to Upstream or
|
||
|
|
// Local node.
|
||
|
|
type router struct {
|
||
|
|
methods map[string]bool
|
||
|
|
upstreamEnabled bool
|
||
|
|
}
|
||
|
|
|
||
|
|
// newRouter inits new router.
|
||
|
|
func newRouter(upstreamEnabled bool) *router {
|
||
|
|
r := &router{
|
||
|
|
methods: make(map[string]bool),
|
||
|
|
upstreamEnabled: upstreamEnabled,
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, m := range localMethods {
|
||
|
|
r.methods[m] = true
|
||
|
|
}
|
||
|
|
|
||
|
|
return r
|
||
|
|
}
|
||
|
|
|
||
|
|
// routeLocally returns true if given method should be routed to
|
||
|
|
// the local node
|
||
|
|
func (r *router) routeLocally(method string) bool {
|
||
|
|
// if upstream is disabled, always route to
|
||
|
|
// the local node
|
||
|
|
if !r.upstreamEnabled {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
// else check route using the methods list
|
||
|
|
return r.methods[method]
|
||
|
|
}
|
||
|
|
|
||
|
|
// localMethods contains methods that should be routed to
|
||
|
|
// the local node; the rest is considered to be routed to
|
||
|
|
// the upstream node.
|
||
|
|
// TODO: in the future, default option may be "local node",
|
||
|
|
// so it'd be convinient to keep track of "remoteMethods" here.
|
||
|
|
var localMethods = [...]string{
|
||
|
|
//Whisper commands
|
||
|
|
"shh_post",
|
||
|
|
"shh_version",
|
||
|
|
"shh_newIdentity",
|
||
|
|
"shh_hasIdentity",
|
||
|
|
"shh_newGroup",
|
||
|
|
"shh_addToGroup",
|
||
|
|
"shh_newFilter",
|
||
|
|
"shh_uninstallFilter",
|
||
|
|
"shh_getFilterChanges",
|
||
|
|
"shh_getMessages",
|
||
|
|
|
||
|
|
// DB commands
|
||
|
|
"db_putString",
|
||
|
|
"db_getString",
|
||
|
|
"db_putHex",
|
||
|
|
"db_getHex",
|
||
|
|
|
||
|
|
// Other commands
|
||
|
|
"net_version",
|
||
|
|
"net_peerCount",
|
||
|
|
"net_listening",
|
||
|
|
|
||
|
|
// blockchain commands
|
||
|
|
"eth_sign",
|
||
|
|
"eth_accounts",
|
||
|
|
"eth_getCompilers",
|
||
|
|
"eth_compileLLL",
|
||
|
|
"eth_compileSolidity",
|
||
|
|
"eth_compileSerpent",
|
||
|
|
}
|