benchmark: update proper benchmark binary to use larger buffers (#6537)

grpc · Aug 14, 2023 · 53d1f23 · 53d1f23
1 parent fbff2ab
commit 53d1f23
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/benchmark/worker/benchmark_client.go b/benchmark/worker/benchmark_client.go
@@ -116,7 +116,10 @@ func setupClientEnv(config *testpb.ClientConfig) {
 // It returns the connections and corresponding function to close them.
 // It returns non-nil error if there is anything wrong.
 func createConns(config *testpb.ClientConfig) ([]*grpc.ClientConn, func(), error) {
-	var opts []grpc.DialOption
+	opts := []grpc.DialOption{
+		grpc.WithWriteBufferSize(128 * 1024),
+		grpc.WithReadBufferSize(128 * 1024),
+	}
 
 	// Sanity check for client type.
 	switch config.ClientType {

diff --git a/benchmark/worker/benchmark_server.go b/benchmark/worker/benchmark_server.go
@@ -80,7 +80,10 @@ func startBenchmarkServer(config *testpb.ServerConfig, serverPort int) (*benchma
 	}
 	runtime.GOMAXPROCS(numOfCores)
 
-	var opts []grpc.ServerOption
+	opts := []grpc.ServerOption{
+		grpc.WriteBufferSize(128 * 1024),
+		grpc.ReadBufferSize(128 * 1024),
+	}
 
 	// Sanity check for server type.
 	switch config.ServerType {