blob: 6065f1cd1c38c6f26af5fa9e0494adfbdb696257 [file] [log] [blame]
// a simple command-line tool to run the benchmark client.
package main
import (
"flag"
"os"
"veyron/runtimes/google/ipc/benchmarks"
"veyron2/rt"
)
var (
server = flag.String("server", "", "object name of the server to connect to")
count = flag.Int("count", 1, "number of RPCs to send")
chunkCount = flag.Int("chunk_count", 0, "number of stream chunks to send")
payloadSize = flag.Int("payload_size", 32, "the size of the payload")
)
func main() {
r := rt.Init()
ctx := r.NewContext()
if *chunkCount == 0 {
benchmarks.CallEcho(ctx, *server, *count, *payloadSize, os.Stdout)
} else {
benchmarks.CallEchoStream(*server, *count, *chunkCount, *payloadSize, os.Stdout)
}
}