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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
package log
import (
"bufio"
"context"
"fmt"
"io"
"gitlab.com/gitlab-org/gitaly/v16/internal/git"
"gitlab.com/gitlab-org/gitaly/v16/internal/git/catfile"
"gitlab.com/gitlab-org/gitaly/v16/proto/go/gitalypb"
)
// Parser holds necessary state for parsing a git log stream
type Parser struct {
scanner *bufio.Scanner
currentCommit *gitalypb.GitCommit
err error
objectReader catfile.ObjectContentReader
}
// NewParser returns a new Parser
func NewParser(ctx context.Context, catfileCache catfile.Cache, repo git.RepositoryExecutor, src io.Reader) (*Parser, func(), error) {
objectReader, cancel, err := catfileCache.ObjectReader(ctx, repo)
if err != nil {
return nil, nil, err
}
parser := &Parser{
scanner: bufio.NewScanner(src),
objectReader: objectReader,
}
return parser, cancel, nil
}
// Parse parses a single git log line. It returns true if successful, false if it finished
// parsing all logs or when it encounters an error, in which case use Parser.Err()
// to get the error.
func (parser *Parser) Parse(ctx context.Context) bool {
if !parser.scanner.Scan() || parser.err != nil {
return false
}
commitID := parser.scanner.Text()
commit, err := catfile.GetCommit(ctx, parser.objectReader, git.Revision(commitID))
if err != nil {
parser.err = err
return false
}
if commit == nil {
parser.err = fmt.Errorf("could not retrieve commit %q", commitID)
return false
}
parser.currentCommit = commit.GitCommit
return true
}
// Commit returns a successfully parsed git log line. It should be called only when Parser.Parse()
// returns true.
func (parser *Parser) Commit() *gitalypb.GitCommit {
return parser.currentCommit
}
// Err returns the error encountered (if any) when parsing the diff stream. It should be called only when Parser.Parse()
// returns false.
func (parser *Parser) Err() error {
if parser.err == nil {
parser.err = parser.scanner.Err()
}
return parser.err
}
|
