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
|
package repository
import (
"context"
"fmt"
"gitlab.com/gitlab-org/gitaly/v15/structerr"
"github.com/prometheus/client_golang/prometheus"
"gitlab.com/gitlab-org/gitaly/proto/v15/go/gitalypb"
"gitlab.com/gitlab-org/gitaly/v15/internal/git/housekeeping"
"gitlab.com/gitlab-org/gitaly/v15/internal/git/stats"
"gitlab.com/gitlab-org/gitaly/v15/internal/gitaly/service"
)
var repackCounter = prometheus.NewCounterVec(
prometheus.CounterOpts{
Name: "gitaly_repack_total",
Help: "Counter of Git repack operations",
},
[]string{"bitmap"},
)
func init() {
prometheus.MustRegister(repackCounter)
}
func (s *server) RepackFull(ctx context.Context, in *gitalypb.RepackFullRequest) (*gitalypb.RepackFullResponse, error) {
repository := in.GetRepository()
if err := service.ValidateRepository(repository); err != nil {
return nil, structerr.NewInvalidArgument("%w", err)
}
repo := s.localrepo(repository)
cfg := housekeeping.RepackObjectsConfig{
FullRepack: true,
WriteBitmap: in.GetCreateBitmap(),
}
repackCounter.WithLabelValues(fmt.Sprint(in.GetCreateBitmap())).Inc()
if err := housekeeping.RepackObjects(ctx, repo, cfg); err != nil {
return nil, structerr.NewInternal("repacking objects: %w", err)
}
writeCommitGraphCfg, err := housekeeping.WriteCommitGraphConfigForRepository(ctx, repo)
if err != nil {
return nil, structerr.NewInternal("getting commit-graph config: %w", err)
}
if err := housekeeping.WriteCommitGraph(ctx, repo, writeCommitGraphCfg); err != nil {
return nil, structerr.NewInternal("writing commit-graph: %w", err)
}
stats.LogRepositoryInfo(ctx, repo)
return &gitalypb.RepackFullResponse{}, nil
}
func (s *server) RepackIncremental(ctx context.Context, in *gitalypb.RepackIncrementalRequest) (*gitalypb.RepackIncrementalResponse, error) {
repository := in.GetRepository()
if err := service.ValidateRepository(repository); err != nil {
return nil, structerr.NewInvalidArgument("%w", err)
}
repo := s.localrepo(repository)
cfg := housekeeping.RepackObjectsConfig{
FullRepack: false,
WriteBitmap: false,
}
repackCounter.WithLabelValues(fmt.Sprint(false)).Inc()
if err := housekeeping.RepackObjects(ctx, repo, cfg); err != nil {
return nil, structerr.NewInternal("repacking objects: %w", err)
}
writeCommitGraphCfg, err := housekeeping.WriteCommitGraphConfigForRepository(ctx, repo)
if err != nil {
return nil, structerr.NewInternal("getting commit-graph config: %w", err)
}
if err := housekeeping.WriteCommitGraph(ctx, repo, writeCommitGraphCfg); err != nil {
return nil, structerr.NewInternal("writing commit-graph: %w", err)
}
stats.LogRepositoryInfo(ctx, repo)
return &gitalypb.RepackIncrementalResponse{}, nil
}
|
