grpcutil

Documentation

Index

• Variables
• func Chunk(data []byte, chunkSize int) [][]byte
• func ChunkReader(r io.Reader, f func([]byte) error) (int, error)
• func GetBuffer() []byte
• func NewStreamingBytesReader(streamingBytesClient StreamingBytesClient) io.Reader
• func NewStreamingBytesWriter(streamingBytesServer StreamingBytesServer) io.Writer
• func PutBuffer(buf []byte)
• func Serve(registerFunc func(*grpc.Server), options ServeOptions, serveEnv ServeEnv) (retErr error)
• func WriteFromStreamingBytesClient(streamingBytesClient StreamingBytesClient, writer io.Writer) error
• func WriteToStreamingBytesServer(reader io.Reader, streamingBytesServer StreamingBytesServer) error
• type Dialer
  • func NewDialer(opts ...grpc.DialOption) Dialer
• type Pool
  • func NewPool(kubeClient *kube.Client, namespace string, serviceName string, numWorkers int, ...) (*Pool, error)
  • func (p *Pool) Close() error
  • func (p *Pool) Get(ctx context.Context) (*grpc.ClientConn, error)
  • func (p *Pool) Put(conn *grpc.ClientConn) error
• type ServeEnv
• type ServeOptions
• type StreamingBytesClient
• type StreamingBytesServer

Variables

var (
	// ErrMustSpecifyRegisterFunc is used when a register func is nil.
	ErrMustSpecifyRegisterFunc = errors.New("must specify registerFunc")
)

var (
	// MaxMsgSize is used to define the GRPC frame size
	MaxMsgSize = 20 * 1024 * 1024
)

Functions

func Chunk

func Chunk(data []byte, chunkSize int) [][]byte

Chunk splits a piece of data up, this is useful for splitting up data that's bigger than MaxMsgSize

func ChunkReader

func ChunkReader(r io.Reader, f func([]byte) error) (int, error)

ChunkReader splits a reader into reasonably sized chunks for the purpose of transmitting the chunks over gRPC. For each chunk, it calls the given function.

func GetBuffer

func GetBuffer() []byte

GetBuffer returns a buffer. The buffer may or may not be freshly allocated, and it may or may not be zero-ed.

func NewStreamingBytesReader

func NewStreamingBytesReader(streamingBytesClient StreamingBytesClient) io.Reader

NewStreamingBytesReader returns an io.Reader for a StreamingBytesClient.

func NewStreamingBytesWriter

func NewStreamingBytesWriter(streamingBytesServer StreamingBytesServer) io.Writer

NewStreamingBytesWriter returns an io.Writer for a StreamingBytesServer.

func PutBuffer

func PutBuffer(buf []byte)

PutBuffer returns the buffer to the pool.

func Serve

func Serve(
	registerFunc func(*grpc.Server),
	options ServeOptions,
	serveEnv ServeEnv,
) (retErr error)

Serve serves stuff.

func WriteFromStreamingBytesClient

func WriteFromStreamingBytesClient(streamingBytesClient StreamingBytesClient, writer io.Writer) error

WriteFromStreamingBytesClient writes from the StreamingBytesClient to the io.Writer.

func WriteToStreamingBytesServer

func WriteToStreamingBytesServer(reader io.Reader, streamingBytesServer StreamingBytesServer) error

WriteToStreamingBytesServer writes the data from the io.Reader to the StreamingBytesServer.

Types

type Dialer

type Dialer interface {
	Dial(address string) (*grpc.ClientConn, error)
	CloseConns() error
}

Dialer defines a grpc.ClientConn connection dialer.

func NewDialer

func NewDialer(opts ...grpc.DialOption) Dialer

NewDialer creates a Dialer.

type Pool

type Pool struct {
	// contains filtered or unexported fields
}

Pool stores a pool of grpc connections to, it's useful in places where you would otherwise need to create several connections.

func NewPool

func NewPool(kubeClient *kube.Client, namespace string, serviceName string, numWorkers int, opts ...grpc.DialOption) (*Pool, error)

NewPool creates a new connection pool with connections to pods in the given service.

func (*Pool) Close

func (p *Pool) Close() error

Close closes all connections stored in the pool, it returns an error if any of the calls to Close error.

func (*Pool) Get

func (p *Pool) Get(ctx context.Context) (*grpc.ClientConn, error)

Get returns a new connection, unlike sync.Pool if it has a cached connection it will always return it. Otherwise it will create a new one. Get errors only when it needs to Dial a new connection and that process fails.

func (*Pool) Put

func (p *Pool) Put(conn *grpc.ClientConn) error

Put returns the connection to the pool. If there are more than `size` connections already cached in the pool the connection will be closed. Put errors only when it Closes a connection and that call errors.

type ServeEnv

type ServeEnv struct {
	// Default is 7070.
	GRPCPort uint16 `env:"GRPC_PORT,default=7070"`
}

ServeEnv are environment variables for serving.

type ServeOptions

type ServeOptions struct {
	Version    *versionpb.Version
	MaxMsgSize int
}

ServeOptions represent optional fields for serving.

type StreamingBytesClient

type StreamingBytesClient interface {
	Recv() (*types.BytesValue, error)
}

StreamingBytesClient represents a client for an rpc method of the form:

rpc Foo(Bar) returns (stream google.protobuf.BytesValue) {}

type StreamingBytesServer

type StreamingBytesServer interface {
	Send(bytesValue *types.BytesValue) error
}

StreamingBytesServer represents a server for an rpc method of the form:

rpc Foo(Bar) returns (stream google.protobuf.BytesValue) {}