mirror of https://github.com/grpc/grpc-go.git
90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"math"
|
|
"net"
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
"sync"
|
|
"time"
|
|
|
|
"google.golang.org/grpc/benchmark"
|
|
"google.golang.org/grpc/benchmark/stats"
|
|
testpb "google.golang.org/grpc/interop/grpc_testing"
|
|
)
|
|
|
|
var (
|
|
server = flag.String("server", "", "The server address")
|
|
maxConcurrentRPCs = flag.Int("max_concurrent_rpcs", 1, "The max number of concurrent RPCs")
|
|
duration = flag.Int("duration", math.MaxInt32, "The duration in seconds to run the benchmark client")
|
|
)
|
|
|
|
func caller(client testpb.TestServiceClient) {
|
|
benchmark.DoUnaryCall(client, 1, 1)
|
|
}
|
|
|
|
func closeLoop() {
|
|
s := stats.NewStats(256)
|
|
conn := benchmark.NewClientConn(*server)
|
|
tc := testpb.NewTestServiceClient(conn)
|
|
// Warm up connection.
|
|
for i := 0; i < 100; i++ {
|
|
caller(tc)
|
|
}
|
|
ch := make(chan int, *maxConcurrentRPCs*4)
|
|
var (
|
|
mu sync.Mutex
|
|
wg sync.WaitGroup
|
|
)
|
|
wg.Add(*maxConcurrentRPCs)
|
|
// Distribute RPCs over maxConcurrentCalls workers.
|
|
for i := 0; i < *maxConcurrentRPCs; i++ {
|
|
go func() {
|
|
for _ = range ch {
|
|
start := time.Now()
|
|
caller(tc)
|
|
elapse := time.Since(start)
|
|
mu.Lock()
|
|
s.Add(elapse)
|
|
mu.Unlock()
|
|
}
|
|
wg.Done()
|
|
}()
|
|
}
|
|
// Stop the client when time is up.
|
|
done := make(chan struct{})
|
|
go func() {
|
|
<-time.After(time.Duration(*duration) * time.Second)
|
|
close(done)
|
|
}()
|
|
ok := true
|
|
for ok {
|
|
select {
|
|
case ch <- 0:
|
|
case <-done:
|
|
ok = false
|
|
}
|
|
}
|
|
close(ch)
|
|
wg.Wait()
|
|
conn.Close()
|
|
log.Println(s.String())
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
go func() {
|
|
lis, err := net.Listen("tcp", ":0")
|
|
if err != nil {
|
|
log.Fatalf("Failed to listen: %v", err)
|
|
}
|
|
log.Println("Client profiling address: ", lis.Addr().String())
|
|
if err := http.Serve(lis, nil); err != nil {
|
|
log.Fatalf("Failed to serve: %v", err)
|
|
}
|
|
}()
|
|
closeLoop()
|
|
}
|