flate.go

package flate

import (
	"bytes"

	"github.com/3JoB/ulib/pool"
	"github.com/klauspost/compress/flate"

	"github.com/3JoB/anthropic-sdk-go/v2/pkg/compress"
)

type Flate struct{}

// Initialize a new Deflate object based on the Interface interface
func New() compress.Interface {
	return &Flate{}
}

// Encode compresses the given bytes using Deflate compression,
// returning the compressed data in a new bytes.Buffer.
func (f *Flate) Encode(v []byte) (*bytes.Buffer, error) {
	i := pool.NewBuffer()
	w, _ := flate.NewWriter(i, 9)
	if _, err := w.Write(v); err != nil {
		return i, err
	}
	if err := w.Close(); err != nil {
		return i, err
	}
	return i, nil
}

// The Decode method will first decode and then
// overwrite the data in the input *bytes.Buffer.
func (f *Flate) Decode(v *bytes.Buffer) {
	r := flate.NewReader(v)
	d := compress.Reader(r)
	_ = r.Close()
	v.Reset()
	_, _ = v.Write(d)
}