traces/vendor/github.com/grafana/gomemcache/memcache/line_reader.go

package memcache

import (
	"bufio"
	"bytes"
	"errors"
	"fmt"
	"io"
)

type lineReader interface {
	ReadLine(from io.Reader, lineLength int) ([]byte, error)
}

type allocatingLineReader struct {
	allocator Allocator
}

func (s allocatingLineReader) ReadLine(from io.Reader, lineLength int) ([]byte, error) {
	// Note that lineLength MUST account for the trailing \r\n.
	if lineLength < len(crlf) {
		return nil, errors.New("line length too small: must include CRLF")
	}

	// Get can return a larger buffer than requested, but never smaller.
	buff := s.allocator.Get(lineLength)

	destBuf := (*buff)[:lineLength]
	_, err := io.ReadFull(from, destBuf)
	if err != nil {
		s.allocator.Put(buff)
		return nil, fmt.Errorf("failed to read line: %w", err)
	}
	if !bytes.HasSuffix(destBuf, crlf) {
		s.allocator.Put(buff)
		return nil, fmt.Errorf("line is not followed by CRLF")
	}
	return destBuf[:lineLength-len(crlf)], nil
}

type noopLineReader struct{}

func (s noopLineReader) ReadLine(from io.Reader, lineLength int) ([]byte, error) {
	_, err := io.CopyN(io.Discard, from, int64(lineLength))
	if err != nil {
		return nil, fmt.Errorf("discarding line: %w", err)
	}
	return nil, nil
}

func tryDiscardLines(r *bufio.Reader) error {
	for {
		_, err := readLine(r, noopLineReader{})
		if errors.Is(err, io.EOF) {
			return nil
		} else if err != nil {
			return fmt.Errorf("memcache GetMulti: discarding cancelled response: %w", err)
		}
	}
}

func readLine[R lineReader](r *bufio.Reader, buff R) (*Item, error) {
	line, err := r.ReadSlice('\n')
	if err != nil {
		return nil, err
	}
	if bytes.Equal(line, resultEnd) {
		return nil, io.EOF
	}
	it := new(Item)
	size, err := scanGetResponseLine(line, it)
	if err != nil {
		return nil, err
	}

	// Expect the line to end with \r\n
	readSize := size + len(crlf)

	it.Value, err = buff.ReadLine(r, readSize)
	if err != nil {
		return nil, fmt.Errorf("memcache: corrupt get result: %w", err)
	}
	return it, nil
}