connect_brotli.go

package brotli

import (
	"connectrpc.com/connect"
	"github.com/andybalholm/brotli"
)

const (
	BestSpeed          = brotli.BestSpeed
	BestCompression    = brotli.BestCompression
	DefaultCompression = brotli.DefaultCompression
)

const Name = "br"

// WithCompression returns client and handler options for the brotli compression method
// using the default compression level.
func WithCompression() connect.Option {
	return WithCompressionLevel(DefaultCompression)
}

// WithCompressionLevel returns client and handler options for the brotli compression
// method for your prefered compression level.
func WithCompressionLevel(level int) connect.Option {
	d, c := brComp(level)
	return compressorOption{
		ClientOption:  connect.WithAcceptCompression(Name, d, c),
		HandlerOption: connect.WithCompression(Name, d, c),
	}
}

func brComp(level int) (func() connect.Decompressor, func() connect.Compressor) {
	d := func() connect.Decompressor { return &brrWrapper{brotli.NewReader(nil)} }
	c := func() connect.Compressor { return brotli.NewWriterLevel(nil, level) }
	return d, c
}

type compressorOption struct {
	connect.ClientOption
	connect.HandlerOption
}

type brrWrapper struct{ *brotli.Reader }

func (b *brrWrapper) Close() error { return b.Reset(nil) }