rpc_util: export a SharedBufferPool

dialoptions.go

@@ -77,7 +77,7 @@ type dialOptions struct {
 	defaultServiceConfig        *ServiceConfig // defaultServiceConfig is parsed from defaultServiceConfigRawJSON.
 	defaultServiceConfigRawJSON *string
 	resolvers                   []resolver.Builder
-	sharedRecvBufferPool        sharedRecvBufferPool
+	sharedRecvBufferPool        SharedBufferPool
 }
 
 // DialOption configures how we set up the connection.
@@ -655,7 +655,7 @@ func WithResolvers(rs ...resolver.Builder) DialOption {
 //
 // Notice: This API is EXPERIMENTAL and may be changed or removed in a
 // later release.
-func WithSharedRecvBufferPool(bufferPool sharedRecvBufferPool) DialOption {
+func WithSharedRecvBufferPool(bufferPool SharedBufferPool) DialOption {
 	return newFuncDialOption(func(o *dialOptions) {
 		o.sharedRecvBufferPool = bufferPool
 	})

rpc_util.go

@@ -579,7 +579,7 @@ type parser struct {
 	header [5]byte
 
 	// sharedRecvBufferPool is the pool of shared receive buffers.
-	sharedRecvBufferPool sharedRecvBufferPool
+	sharedRecvBufferPool SharedBufferPool
 }
 
 // recvMsg reads a complete gRPC message from the stream.
@@ -966,8 +966,3 @@ const (
 )
 
 const grpcUA = "grpc-go/" + Version
-
-type sharedRecvBufferPool interface {
-	Get(length int) []byte
-	Put(*[]byte)
-}

server.go

@@ -175,7 +175,7 @@ type serverOptions struct {
 	maxHeaderListSize     *uint32
 	headerTableSize       *uint32
 	numServerWorkers      uint32
-	sharedRecvBufferPool  sharedRecvBufferPool
+	sharedRecvBufferPool  SharedBufferPool
 }
 
 var defaultServerOptions = serverOptions{
@@ -561,7 +561,7 @@ func NumStreamWorkers(numServerWorkers uint32) ServerOption {
 //
 // Notice: This API is EXPERIMENTAL and may be changed or removed in a
 // later release.
-func SharedRecvBufferPool(bufferPool sharedRecvBufferPool) ServerOption {
+func SharedRecvBufferPool(bufferPool SharedBufferPool) ServerOption {
 	return newFuncServerOption(func(o *serverOptions) {
 		o.sharedRecvBufferPool = bufferPool
 	})

shared_buffer_pool.go

@@ -0,0 +1,7 @@
+package grpc
+
+// SharedBufferPool is a pool of buffers that can be shared.
+type SharedBufferPool interface {
+	Get(length int) []byte
+	Put(*[]byte)
+}