You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
131 lines
4.0 KiB
131 lines
4.0 KiB
package main
|
|
|
|
import (
|
|
"errors"
|
|
"net"
|
|
"net/http"
|
|
|
|
rpctypes "github.com/tendermint/tendermint/rpc/lib/types"
|
|
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
rpc "github.com/tendermint/tendermint/rpc/lib/server"
|
|
"github.com/tendermint/tendermint/tools/tm-monitor/monitor"
|
|
)
|
|
|
|
func startRPC(listenAddr string, m *monitor.Monitor, logger log.Logger) net.Listener {
|
|
routes := routes(m)
|
|
|
|
mux := http.NewServeMux()
|
|
wm := rpc.NewWebsocketManager(routes, nil)
|
|
mux.HandleFunc("/websocket", wm.WebsocketHandler)
|
|
rpc.RegisterRPCFuncs(mux, routes, cdc, logger)
|
|
config := rpc.DefaultConfig()
|
|
listener, err := rpc.Listen(listenAddr, config)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
go rpc.StartHTTPServer(listener, mux, logger, config)
|
|
return listener
|
|
}
|
|
|
|
func routes(m *monitor.Monitor) map[string]*rpc.RPCFunc {
|
|
return map[string]*rpc.RPCFunc{
|
|
"status": rpc.NewRPCFunc(RPCStatus(m), ""),
|
|
"status/network": rpc.NewRPCFunc(RPCNetworkStatus(m), ""),
|
|
"status/node": rpc.NewRPCFunc(RPCNodeStatus(m), "name"),
|
|
"monitor": rpc.NewRPCFunc(RPCMonitor(m), "endpoint"),
|
|
"unmonitor": rpc.NewRPCFunc(RPCUnmonitor(m), "endpoint"),
|
|
|
|
// "start_meter": rpc.NewRPCFunc(network.StartMeter, "chainID,valID,event"),
|
|
// "stop_meter": rpc.NewRPCFunc(network.StopMeter, "chainID,valID,event"),
|
|
// "meter": rpc.NewRPCFunc(GetMeterResult(network), "chainID,valID,event"),
|
|
}
|
|
}
|
|
|
|
// RPCStatus returns common statistics for the network and statistics per node.
|
|
func RPCStatus(m *monitor.Monitor) func(*rpctypes.Context) (networkAndNodes, error) {
|
|
return func(_ *rpctypes.Context) (networkAndNodes, error) {
|
|
return networkAndNodes{m.Network, m.Nodes}, nil
|
|
}
|
|
}
|
|
|
|
// RPCNetworkStatus returns common statistics for the network.
|
|
func RPCNetworkStatus(m *monitor.Monitor) func(*rpctypes.Context) (*monitor.Network, error) {
|
|
return func(_ *rpctypes.Context) (*monitor.Network, error) {
|
|
return m.Network, nil
|
|
}
|
|
}
|
|
|
|
// RPCNodeStatus returns statistics for the given node.
|
|
func RPCNodeStatus(m *monitor.Monitor) func(*rpctypes.Context, string) (*monitor.Node, error) {
|
|
return func(_ *rpctypes.Context, name string) (*monitor.Node, error) {
|
|
if i, n := m.NodeByName(name); i != -1 {
|
|
return n, nil
|
|
}
|
|
return nil, errors.New("cannot find node with that name")
|
|
}
|
|
}
|
|
|
|
// RPCMonitor allows to dynamically add a endpoint to under the monitor. Safe
|
|
// to call multiple times.
|
|
func RPCMonitor(m *monitor.Monitor) func(*rpctypes.Context, string) (*monitor.Node, error) {
|
|
return func(_ *rpctypes.Context, endpoint string) (*monitor.Node, error) {
|
|
i, n := m.NodeByName(endpoint)
|
|
if i == -1 {
|
|
n = monitor.NewNode(endpoint)
|
|
if err := m.Monitor(n); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return n, nil
|
|
}
|
|
}
|
|
|
|
// RPCUnmonitor removes the given endpoint from under the monitor.
|
|
func RPCUnmonitor(m *monitor.Monitor) func(*rpctypes.Context, string) (bool, error) {
|
|
return func(_ *rpctypes.Context, endpoint string) (bool, error) {
|
|
if i, n := m.NodeByName(endpoint); i != -1 {
|
|
m.Unmonitor(n)
|
|
return true, nil
|
|
}
|
|
return false, errors.New("cannot find node with that name")
|
|
}
|
|
}
|
|
|
|
// func (tn *TendermintNetwork) StartMeter(chainID, valID, eventID string) error {
|
|
// tn.mtx.Lock()
|
|
// defer tn.mtx.Unlock()
|
|
// val, err := tn.getChainVal(chainID, valID)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// return val.EventMeter().Subscribe(eventID, nil)
|
|
// }
|
|
|
|
// func (tn *TendermintNetwork) StopMeter(chainID, valID, eventID string) error {
|
|
// tn.mtx.Lock()
|
|
// defer tn.mtx.Unlock()
|
|
// val, err := tn.getChainVal(chainID, valID)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// return val.EventMeter().Unsubscribe(eventID)
|
|
// }
|
|
|
|
// func (tn *TendermintNetwork) GetMeter(chainID, valID, eventID string) (*eventmeter.EventMetric, error) {
|
|
// tn.mtx.Lock()
|
|
// defer tn.mtx.Unlock()
|
|
// val, err := tn.getChainVal(chainID, valID)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
|
|
// return val.EventMeter().GetMetric(eventID)
|
|
// }
|
|
|
|
//--> types
|
|
|
|
type networkAndNodes struct {
|
|
Network *monitor.Network `json:"network"`
|
|
Nodes []*monitor.Node `json:"nodes"`
|
|
}
|