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.
60 lines
1.7 KiB
60 lines
1.7 KiB
package mock
|
|
|
|
import (
|
|
"log"
|
|
"reflect"
|
|
|
|
gokitlog "github.com/go-kit/kit/log"
|
|
ctypes "github.com/tendermint/tendermint/rpc/core/types"
|
|
em "github.com/tendermint/tools/tm-monitor/eventmeter"
|
|
)
|
|
|
|
type EventMeter struct {
|
|
latencyCallback em.LatencyCallbackFunc
|
|
disconnectCallback em.DisconnectCallbackFunc
|
|
eventCallback em.EventCallbackFunc
|
|
}
|
|
|
|
func (e *EventMeter) Start() error { return nil }
|
|
func (e *EventMeter) Stop() {}
|
|
func (e *EventMeter) SetLogger(l gokitlog.Logger) {}
|
|
func (e *EventMeter) RegisterLatencyCallback(cb em.LatencyCallbackFunc) { e.latencyCallback = cb }
|
|
func (e *EventMeter) RegisterDisconnectCallback(cb em.DisconnectCallbackFunc) {
|
|
e.disconnectCallback = cb
|
|
}
|
|
func (e *EventMeter) Subscribe(eventID string, cb em.EventCallbackFunc) error {
|
|
e.eventCallback = cb
|
|
return nil
|
|
}
|
|
func (e *EventMeter) Unsubscribe(eventID string) error {
|
|
e.eventCallback = nil
|
|
return nil
|
|
}
|
|
|
|
func (e *EventMeter) Call(callback string, args ...interface{}) {
|
|
switch callback {
|
|
case "latencyCallback":
|
|
e.latencyCallback(args[0].(float64))
|
|
case "disconnectCallback":
|
|
e.disconnectCallback()
|
|
case "eventCallback":
|
|
e.eventCallback(args[0].(*em.EventMetric), args[1])
|
|
}
|
|
}
|
|
|
|
type RpcClient struct {
|
|
Stubs map[string]ctypes.TMResult
|
|
}
|
|
|
|
func (c *RpcClient) Call(method string, params map[string]interface{}, result interface{}) (interface{}, error) {
|
|
s, ok := c.Stubs[method]
|
|
if !ok {
|
|
log.Fatalf("Call to %s, but no stub is defined for it", method)
|
|
}
|
|
|
|
rv, rt := reflect.ValueOf(result), reflect.TypeOf(result)
|
|
rv, rt = rv.Elem(), rt.Elem()
|
|
rv.Set(reflect.ValueOf(s))
|
|
|
|
return s, nil
|
|
}
|