aboutsummaryrefslogtreecommitdiff
path: root/format/packfile/ingest/fill.go
blob: eca4e4d662570d3cd57fa71a0d37ae3e5ad8bea5 (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package ingest

import (
	"errors"
	"fmt"
	"io"
)

// fill ensures at least min unread bytes are available in receiver's buffer.
func (scanner *streamScanner) fill(minLen int) error {
	if minLen <= 0 {
		return nil
	}

	if minLen > len(scanner.buf) {
		return fmt.Errorf("packfile/ingest: fill(%d) exceeds scanner buffer", minLen)
	}

	for scanner.n-scanner.off < minLen {
		err := scanner.flushConsumedPrefix()
		if err != nil {
			return err
		}

		readN, err := scanner.src.Read(scanner.buf[scanner.n:])
		if readN > 0 {
			scanner.n += readN
		}

		if err != nil {
			if errors.Is(err, io.EOF) && scanner.n-scanner.off >= minLen {
				return nil
			}

			return err
		}

		if readN == 0 {
			return io.ErrNoProgress
		}
	}

	return nil
}