Merge pull request #7 from tendermint/feature/extract-lib

[tm-monitor] extract lib to monitor/ dir

tm-monitor/Makefile

@@ -3,6 +3,7 @@ VERSION := $(shell perl -ne '/^var version.*"([^"]+)".*$$/ && print "v$$1\n"' ma
 GOTOOLS = \
 					github.com/Masterminds/glide \
 					github.com/mitchellh/gox
+PACKAGES=$(shell go list ./... | grep -v '/vendor/')
 
 tools:
 	go get -v $(GOTOOLS)
@@ -17,7 +18,7 @@ install:
 	go install -ldflags "-X main.version=${VERSION}"
 
 test:
-	go test
+	@go test $(PACKAGES)
 
 build-all: tools
 	gox -verbose \
@@ -41,7 +42,6 @@ build-docker:
 	docker build -t "tendermint/monitor" .
 
 clean:
-	rm -f ./tm-monitor.log
 	rm -f ./tm-monitor
 	rm -rf ./dist
 

tm-monitor/glide.lock

@@ -1,5 +1,5 @@
-hash: 3315dcf12e2554e2927f2a0907f2547cd86d5c8926461d055662a7bee88caa4a
-updated: 2017-03-07T08:38:45.613512657Z
+hash: d21d1f12681cd4ab5b7f0efd7bf00c1d5f7021b1ae6e8700c11bca6822337079
+updated: 2017-03-16T10:01:58.079646405Z
 imports:
 - name: github.com/btcsuite/btcd
   version: 583684b21bfbde9b5fc4403916fd7c807feb0289
@@ -7,6 +7,12 @@ imports:
   - btcec
 - name: github.com/BurntSushi/toml
   version: 99064174e013895bbd9b025c31100bd1d9b590ca
+- name: github.com/go-kit/kit
+  version: b6f30a2e0632f5722fb26d8765d726335b79d3e6
+  subpackages:
+  - log
+- name: github.com/go-logfmt/logfmt
+  version: 390ab7935ee28ec6b286364bba9b4dd6410cb3d5
 - name: github.com/go-stack/stack
   version: 100eb0c0a9c5b306ca2fb4f165df21d80ada4b82
 - name: github.com/golang/protobuf
@@ -19,10 +25,14 @@ imports:
   version: 3ab3a8b8831546bd18fd182c20687ca853b2bb13
 - name: github.com/jmhodges/levigo
   version: c42d9e0ca023e2198120196f842701bb4c55d7b9
+- name: github.com/kr/logfmt
+  version: b84e30acd515aadc4b783ad4ff83aff3299bdfe0
 - name: github.com/mattn/go-colorable
   version: d898aa9fb31c91f35dd28ca75db377eff023c076
 - name: github.com/mattn/go-isatty
   version: dda3de49cbfcec471bd7a70e6cc01fcc3ff90109
+- name: github.com/pkg/errors
+  version: bfd5150e4e41705ded2129ec33379de1cb90b513
 - name: github.com/rcrowley/go-metrics
   version: 1f30fe9094a513ce4c700b9a54458bbb0c96996c
 - name: github.com/stretchr/testify

tm-monitor/glide.yaml

@@ -16,3 +16,8 @@ import:
   subpackages:
   - client
 - package: github.com/tendermint/log15
+- package: github.com/go-kit/kit
+  subpackages:
+  - log
+  - term
+- package: github.com/pkg/errors

tm-monitor/main.go

@@ -6,22 +6,21 @@ import (
 	"os"
 	"strings"
 
+	"github.com/go-kit/kit/log"
+	"github.com/go-kit/kit/log/term"
 	cmn "github.com/tendermint/go-common"
-	logger "github.com/tendermint/go-logger"
-	log15 "github.com/tendermint/log15"
-	em "github.com/tendermint/tools/tm-monitor/eventmeter"
+	monitor "github.com/tendermint/tools/tm-monitor/monitor"
 )
 
 var version = "0.3.0.pre"
 
-var log = logger.New()
+var logger = log.NewNopLogger()
 
 func main() {
 	var listenAddr string
-	var verbose, noton bool
+	var noton bool
 
 	flag.StringVar(&listenAddr, "listen-addr", "tcp://0.0.0.0:46670", "HTTP and Websocket server listen address")
-	flag.BoolVar(&verbose, "v", false, "verbose logging")
 	flag.BoolVar(&noton, "no-ton", false, "Do not show ton (table of nodes)")
 
 	flag.Usage = func() {
@@ -29,7 +28,7 @@ func main() {
 applications, collecting and providing various statistics to the user.
 
 Usage:
-	tm-monitor [-v] [-no-ton] [-listen-addr="tcp://0.0.0.0:46670"] [endpoints]
+	tm-monitor [-no-ton] [-listen-addr="tcp://0.0.0.0:46670"] [endpoints]
 
 Examples:
 	# monitor single instance
@@ -48,17 +47,28 @@ Examples:
 		os.Exit(1)
 	}
 
+	if noton {
+		// Color errors red
+		colorFn := func(keyvals ...interface{}) term.FgBgColor {
+			for i := 1; i < len(keyvals); i += 2 {
+				if _, ok := keyvals[i].(error); ok {
+					return term.FgBgColor{Fg: term.White, Bg: term.Red}
+				}
+			}
+			return term.FgBgColor{}
+		}
+
+		logger = term.NewLogger(os.Stdout, log.NewLogfmtLogger, colorFn)
+	}
+
 	m := startMonitor(flag.Arg(0))
 
 	startRPC(listenAddr, m)
 
 	var ton *Ton
 	if !noton {
-		logToFile("tm-monitor.log", verbose)
 		ton = NewTon(m)
 		ton.Start()
-	} else {
-		logToStdout(verbose)
 	}
 
 	cmn.TrapSignal(func() {
@@ -69,50 +79,21 @@ Examples:
 	})
 }
 
-func startMonitor(endpoints string) *Monitor {
-	m := NewMonitor()
+func startMonitor(endpoints string) *monitor.Monitor {
+	m := monitor.NewMonitor()
+	m.SetLogger(log.With(logger, "component", "monitor"))
 
 	for _, e := range strings.Split(endpoints, ",") {
-		if err := m.Monitor(NewNode(e)); err != nil {
-			log.Crit(err.Error())
-			os.Exit(1)
+		n := monitor.NewNode(e)
+		n.SetLogger(log.With(logger, "node", e))
+		if err := m.Monitor(n); err != nil {
+			panic(err)
 		}
 	}
 
 	if err := m.Start(); err != nil {
-		log.Crit(err.Error())
-		os.Exit(1)
+		panic(err)
 	}
 
 	return m
 }
-
-func logToStdout(verbose bool) {
-	if verbose {
-		log.SetHandler(logger.LvlFilterHandler(
-			logger.LvlDebug,
-			logger.BypassHandler(),
-		))
-	} else {
-		log.SetHandler(logger.LvlFilterHandler(
-			logger.LvlInfo,
-			logger.BypassHandler(),
-		))
-	}
-	em.Log = log
-}
-
-func logToFile(filename string, verbose bool) {
-	if verbose {
-		log.SetHandler(logger.LvlFilterHandler(
-			logger.LvlDebug,
-			log15.Must.FileHandler(filename, log15.LogfmtFormat()),
-		))
-	} else {
-		log.SetHandler(logger.LvlFilterHandler(
-			logger.LvlInfo,
-			log15.Must.FileHandler(filename, log15.LogfmtFormat()),
-		))
-	}
-	em.Log = log
-}

tm-monitor/rpc.go

@@ -5,9 +5,10 @@ import (
 	"net/http"
 
 	rpc "github.com/tendermint/go-rpc/server"
+	monitor "github.com/tendermint/tools/tm-monitor/monitor"
 )
 
-func startRPC(listenAddr string, m *Monitor) {
+func startRPC(listenAddr string, m *monitor.Monitor) {
 	routes := routes(m)
 
 	// serve http and ws
@@ -20,7 +21,7 @@ func startRPC(listenAddr string, m *Monitor) {
 	}
 }
 
-func routes(m *Monitor) map[string]*rpc.RPCFunc {
+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), ""),
@@ -35,9 +36,9 @@ func routes(m *Monitor) map[string]*rpc.RPCFunc {
 }
 
 // RPCStatus returns common statistics for the network and statistics per node.
-func RPCStatus(m *Monitor) interface{} {
+func RPCStatus(m *monitor.Monitor) interface{} {
 	return func() (networkAndNodes, error) {
-		values := make([]*Node, len(m.Nodes))
+		values := make([]*monitor.Node, len(m.Nodes))
 		i := 0
 		for _, v := range m.Nodes {
 			values[i] = v
@@ -49,15 +50,15 @@ func RPCStatus(m *Monitor) interface{} {
 }
 
 // RPCNetworkStatus returns common statistics for the network.
-func RPCNetworkStatus(m *Monitor) interface{} {
-	return func() (*Network, error) {
+func RPCNetworkStatus(m *monitor.Monitor) interface{} {
+	return func() (*monitor.Network, error) {
 		return m.Network, nil
 	}
 }
 
 // RPCNodeStatus returns statistics for the given node.
-func RPCNodeStatus(m *Monitor) interface{} {
-	return func(name string) (*Node, error) {
+func RPCNodeStatus(m *monitor.Monitor) interface{} {
+	return func(name string) (*monitor.Node, error) {
 		if n, ok := m.Nodes[name]; ok {
 			return n, nil
 		}
@@ -66,9 +67,9 @@ func RPCNodeStatus(m *Monitor) interface{} {
 }
 
 // RPCMonitor allows to dynamically add a endpoint to under the monitor.
-func RPCMonitor(m *Monitor) interface{} {
-	return func(endpoint string) (*Node, error) {
-		n := NewNode(endpoint)
+func RPCMonitor(m *monitor.Monitor) interface{} {
+	return func(endpoint string) (*monitor.Node, error) {
+		n := monitor.NewNode(endpoint)
 		if err := m.Monitor(n); err != nil {
 			return nil, err
 		}
@@ -77,7 +78,7 @@ func RPCMonitor(m *Monitor) interface{} {
 }
 
 // RPCUnmonitor removes the given endpoint from under the monitor.
-func RPCUnmonitor(m *Monitor) interface{} {
+func RPCUnmonitor(m *monitor.Monitor) interface{} {
 	return func(endpoint string) (bool, error) {
 		if n, ok := m.Nodes[endpoint]; ok {
 			m.Unmonitor(n)
@@ -121,6 +122,6 @@ func RPCUnmonitor(m *Monitor) interface{} {
 //--> types
 
 type networkAndNodes struct {
-	Network *Network `json:"network"`
-	Nodes   []*Node  `json:"nodes"`
+	Network *monitor.Network `json:"network"`
+	Nodes   []*monitor.Node  `json:"nodes"`
 }

tm-monitor/ton.go

@@ -6,6 +6,8 @@ import (
 	"os"
 	"text/tabwriter"
 	"time"
+
+	monitor "github.com/tendermint/tools/tm-monitor/monitor"
 )
 
 const (
@@ -24,14 +26,14 @@ const (
 // Ton was inspired by [Linux top
 // program](https://en.wikipedia.org/wiki/Top_(software)) as the name suggests.
 type Ton struct {
-	monitor *Monitor
+	monitor *monitor.Monitor
 
 	RefreshRate time.Duration
 	Output      io.Writer
 	quit        chan struct{}
 }
 
-func NewTon(m *Monitor) *Ton {
+func NewTon(m *monitor.Monitor) *Ton {
 	return &Ton{
 		RefreshRate: defaultRefreshRate,
 		Output:      os.Stdout,