benchmark: restore old buffer size values for published benchmarks (#…

…6516)
grpc · Aug 8, 2023 · 07609e1 · 07609e1
1 parent 2059c6e
commit 07609e1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/benchmark/client/main.go b/benchmark/client/main.go
@@ -136,7 +136,12 @@ func main() {
 func buildConnections(ctx context.Context) []*grpc.ClientConn {
 	ccs := make([]*grpc.ClientConn, *numConn)
 	for i := range ccs {
-		ccs[i] = benchmark.NewClientConnWithContext(ctx, "localhost:"+*port, grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithBlock())
+		ccs[i] = benchmark.NewClientConnWithContext(ctx, "localhost:"+*port,
+			grpc.WithTransportCredentials(insecure.NewCredentials()),
+			grpc.WithBlock(),
+			grpc.WithWriteBufferSize(128*1024),
+			grpc.WithReadBufferSize(128*1024),
+		)
 	}
 	return ccs
 }

diff --git a/benchmark/server/main.go b/benchmark/server/main.go
@@ -39,6 +39,7 @@ import (
 	"runtime/pprof"
 	"time"
 
+	"google.golang.org/grpc"
 	"google.golang.org/grpc/benchmark"
 	"google.golang.org/grpc/grpclog"
 	"google.golang.org/grpc/internal/syscall"
@@ -70,7 +71,10 @@ func main() {
 	pprof.StartCPUProfile(cf)
 	cpuBeg := syscall.GetCPUTime()
 	// Launch server in a separate goroutine.
-	stop := benchmark.StartServer(benchmark.ServerInfo{Type: "protobuf", Listener: lis})
+	stop := benchmark.StartServer(benchmark.ServerInfo{Type: "protobuf", Listener: lis},
+		grpc.WriteBufferSize(128*1024),
+		grpc.ReadBufferSize(128*1024),
+	)
 	// Wait on OS terminate signal.
 	ch := make(chan os.Signal, 1)
 	signal.Notify(ch, os.Interrupt)