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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
package diff
import (
"io"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/gitaly/diff"
"gitlab.com/gitlab-org/gitaly/v14/proto/go/gitalypb"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
var maxNumStatBatchSize = 1000
func (s *server) DiffStats(in *gitalypb.DiffStatsRequest, stream gitalypb.DiffService_DiffStatsServer) error {
if err := s.validateDiffStatsRequestParams(in); err != nil {
return err
}
var batch []*gitalypb.DiffStats
cmd, err := s.gitCmdFactory.New(stream.Context(), in.Repository, git.SubCmd{
Name: "diff",
Flags: []git.Option{git.Flag{Name: "--numstat"}, git.Flag{Name: "-z"}},
Args: []string{in.LeftCommitId, in.RightCommitId},
})
if err != nil {
if _, ok := status.FromError(err); ok {
return err
}
return status.Errorf(codes.Internal, "%s: cmd: %v", "DiffStats", err)
}
parser := diff.NewDiffNumStatParser(cmd)
for {
stat, err := parser.NextNumStat()
if err != nil {
if err == io.EOF {
break
}
return err
}
numStat := &gitalypb.DiffStats{
Additions: stat.Additions,
Deletions: stat.Deletions,
Path: stat.Path,
OldPath: stat.OldPath,
}
batch = append(batch, numStat)
if len(batch) == maxNumStatBatchSize {
if err := sendStats(batch, stream); err != nil {
return err
}
batch = nil
}
}
if err := cmd.Wait(); err != nil {
return status.Errorf(codes.Unavailable, "%s: %v", "DiffStats", err)
}
return sendStats(batch, stream)
}
func sendStats(batch []*gitalypb.DiffStats, stream gitalypb.DiffService_DiffStatsServer) error {
if len(batch) == 0 {
return nil
}
if err := stream.Send(&gitalypb.DiffStatsResponse{Stats: batch}); err != nil {
return status.Errorf(codes.Unavailable, "DiffStats: send: %v", err)
}
return nil
}
func (s *server) validateDiffStatsRequestParams(in *gitalypb.DiffStatsRequest) error {
repo := in.GetRepository()
if _, err := s.locator.GetRepoPath(repo); err != nil {
return err
}
if err := validateRequest(in); err != nil {
return status.Errorf(codes.InvalidArgument, "DiffStats: %v", err)
}
return nil
}
|