// Program estream is a manual testing tool for polling the event stream
|
|
// of a running Tendermint consensus node.
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"path/filepath"
|
|
|
|
"github.com/tendermint/tendermint/rpc/client/eventstream"
|
|
rpcclient "github.com/tendermint/tendermint/rpc/client/http"
|
|
"github.com/tendermint/tendermint/rpc/coretypes"
|
|
)
|
|
|
|
var (
|
|
query = flag.String("query", "", "Filter query")
|
|
batchSize = flag.Int("batch", 0, "Batch size")
|
|
resumeFrom = flag.String("resume", "", "Resume cursor")
|
|
numItems = flag.Int("count", 0, "Number of items to read (0 to stream)")
|
|
waitTime = flag.Duration("poll", 0, "Long poll interval")
|
|
rpcAddr = flag.String("addr", "http://localhost:26657", "RPC service address")
|
|
)
|
|
|
|
func init() {
|
|
flag.Usage = func() {
|
|
fmt.Fprintf(os.Stderr, `Usage: %[1]s [options]
|
|
|
|
Connect to the Tendermint node whose RPC service is at -addr, and poll for events
|
|
matching the specified -query. If no query is given, all events are fetched.
|
|
The resulting event data are written to stdout as JSON.
|
|
|
|
Use -resume to pick up polling from a previously-reported event cursor.
|
|
Use -count to stop polling after a certain number of events has been reported.
|
|
Use -batch to override the default request batch size.
|
|
Use -poll to override the default long-polling interval.
|
|
|
|
Options:
|
|
`, filepath.Base(os.Args[0]))
|
|
flag.PrintDefaults()
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
cli, err := rpcclient.New(*rpcAddr)
|
|
if err != nil {
|
|
log.Fatalf("RPC client: %v", err)
|
|
}
|
|
stream := eventstream.New(cli, *query, &eventstream.StreamOptions{
|
|
BatchSize: *batchSize,
|
|
ResumeFrom: *resumeFrom,
|
|
WaitTime: *waitTime,
|
|
})
|
|
|
|
// Shut down cleanly on SIGINT. Don't attempt clean shutdown for other
|
|
// fatal signals.
|
|
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
|
|
defer cancel()
|
|
|
|
var nr int
|
|
if err := stream.Run(ctx, func(itm *coretypes.EventItem) error {
|
|
nr++
|
|
bits, err := json.Marshal(itm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Println(string(bits))
|
|
if *numItems > 0 && nr >= *numItems {
|
|
return eventstream.ErrStopRunning
|
|
}
|
|
return nil
|
|
}); err != nil {
|
|
log.Fatalf("Stream failed: %v", err)
|
|
}
|
|
}
|