shithub: furgit

ref: 9053c85456bd9b4457b588610eeef1b8dfff2b89
dir: /format/pktline/chunk_writer.go/

package pktline

import "io"

// ChunkWriter packetizes arbitrary stream bytes into data pkt-lines.
// It never writes control packets automatically.
type ChunkWriter struct {
	enc *Encoder
}

// NewChunkWriter creates a chunking adapter over enc.
func NewChunkWriter(enc *Encoder) *ChunkWriter {
	return &ChunkWriter{enc: enc}
}

// Write splits p into data frames not larger than enc's maxData.
//
// It implements io.Writer.
func (cw *ChunkWriter) Write(p []byte) (int, error) {
	total := 0
	maxData := cw.enc.effectiveMaxData()

	for len(p) > 0 {
		n := min(len(p), maxData)

		err := cw.enc.WriteData(p[:n])
		if err != nil {
			return total, err
		}

		total += n
		p = p[n:]
	}

	return total, nil
}

// ReadFrom reads from r and writes pkt-line data frames to the encoder.
//
// It implements io.ReaderFrom.
func (cw *ChunkWriter) ReadFrom(r io.Reader) (int64, error) {
	buf := make([]byte, cw.enc.effectiveMaxData())

	var total int64

	for {
		n, err := r.Read(buf)
		if n > 0 {
			werr := cw.enc.WriteData(buf[:n])
			if werr != nil {
				return total, werr
			}

			total += int64(n)
		}

		if err != nil {
			if err == io.EOF {
				return total, nil
			}

			return total, err
		}
	}
}