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
|
package ref
import (
"bufio"
"fmt"
"strings"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/git/catfile"
"gitlab.com/gitlab-org/gitaly/internal/helper"
"gitlab.com/gitlab-org/gitaly/proto/go/gitalypb"
)
func (s *server) ListNewBlobs(in *gitalypb.ListNewBlobsRequest, stream gitalypb.RefService_ListNewBlobsServer) error {
oid := in.GetCommitId()
if err := git.ValidateCommitID(oid); err != nil {
return helper.ErrInvalidArgument(err)
}
if err := listNewBlobs(in, stream, oid); err != nil {
return helper.ErrInternal(err)
}
return nil
}
func listNewBlobs(in *gitalypb.ListNewBlobsRequest, stream gitalypb.RefService_ListNewBlobsServer, oid string) error {
ctx := stream.Context()
cmdFlags := []git.Option{git.Flag{Name: "--objects"}, git.Flag{Name: "--not"}, git.Flag{Name: "--all"}}
if in.GetLimit() > 0 {
cmdFlags = append(cmdFlags, git.ValueFlag{Name: "--max-count", Value: fmt.Sprint(in.GetLimit())})
}
// the added ^ is to negate the oid since there is a --not option that comes earlier in the arg list
revList, err := git.SafeCmd(ctx, in.GetRepository(), nil, git.SubCmd{Name: "rev-list", Flags: cmdFlags, Args: []string{"^" + oid}})
if err != nil {
return err
}
batch, err := catfile.New(ctx, in.GetRepository())
if err != nil {
return err
}
var newBlobs []*gitalypb.NewBlobObject
scanner := bufio.NewScanner(revList)
for scanner.Scan() {
line := scanner.Text()
parts := strings.SplitN(line, " ", 2)
if len(parts) != 2 {
continue
}
info, err := batch.Info(ctx, parts[0])
if err != nil {
return err
}
if !info.IsBlob() {
continue
}
newBlobs = append(newBlobs, &gitalypb.NewBlobObject{Oid: info.Oid, Size: info.Size, Path: []byte(parts[1])})
if len(newBlobs) >= 1000 {
response := &gitalypb.ListNewBlobsResponse{NewBlobObjects: newBlobs}
if err := stream.Send(response); err != nil {
return err
}
newBlobs = newBlobs[:0]
}
}
response := &gitalypb.ListNewBlobsResponse{NewBlobObjects: newBlobs}
if err := stream.Send(response); err != nil {
return err
}
return revList.Wait()
}
|
