EdgeNode/internal/compressions/reader_zstd.go

// Copyright 2021 GoEdge goedge.cdn@gmail.com. All rights reserved.

package compressions

import (
	"io"

	"github.com/klauspost/compress/zstd"
)

type ZSTDReader struct {
	BaseReader

	reader *zstd.Decoder
}

func NewZSTDReader(reader io.Reader) (Reader, error) {
	return sharedZSTDReaderPool.Get(reader)
}

func newZSTDReader(reader io.Reader) (Reader, error) {
	r, err := zstd.NewReader(reader, zstd.WithDecoderMaxWindow(256<<20))
	if err != nil {
		return nil, err
	}
	return &ZSTDReader{
		reader: r,
	}, nil
}

func (this *ZSTDReader) Read(p []byte) (n int, err error) {
	return this.reader.Read(p)
}

func (this *ZSTDReader) Reset(reader io.Reader) error {
	return this.reader.Reset(reader)
}

func (this *ZSTDReader) RawClose() error {
	this.reader.Close()
	return nil
}

func (this *ZSTDReader) Close() error {
	return this.Finish(this)
}