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"
"os"
"gitlab.com/gitlab-org/gitaly/internal/git/housekeeping"
"gitlab.com/gitlab-org/gitaly/internal/git/localrepo"
"gitlab.com/gitlab-org/gitaly/internal/git/stats"
"gitlab.com/gitlab-org/gitaly/internal/helper"
"gitlab.com/gitlab-org/gitaly/proto/go/gitalypb"
)
// repackIfNoBitmap uses the bitmap index as a heuristic to determine whether the repository needs a
// full repack. So only if there is none will the full repack be started.
func (s *server) repackIfNoBitmap(ctx context.Context, repository *gitalypb.Repository) error {
repoPath, err := s.locator.GetRepoPath(repository)
if err != nil {
return err
}
hasBitmap, err := stats.HasBitmap(repoPath)
if err != nil {
return helper.ErrInternal(err)
}
if hasBitmap {
return nil
}
altFile, err := s.locator.InfoAlternatesPath(repository)
if err != nil {
return helper.ErrInternal(err)
}
// repositories with alternates should never have a bitmap, as Git will otherwise complain about
// multiple bitmaps being present in parent and alternate repository.
if _, err = os.Stat(altFile); !os.IsNotExist(err) {
return nil
}
if _, err = s.RepackFull(ctx, &gitalypb.RepackFullRequest{
Repository: repository,
CreateBitmap: true,
}); err != nil {
return err
}
return nil
}
func (s *server) optimizeRepository(ctx context.Context, repository *gitalypb.Repository) error {
if err := s.repackIfNoBitmap(ctx, repository); err != nil {
return fmt.Errorf("could not repack: %w", err)
}
repo := localrepo.New(s.gitCmdFactory, repository, s.cfg)
if err := housekeeping.Perform(ctx, repo); err != nil {
return fmt.Errorf("could not execute houskeeping: %w", err)
}
return nil
}
func (s *server) OptimizeRepository(ctx context.Context, in *gitalypb.OptimizeRepositoryRequest) (*gitalypb.OptimizeRepositoryResponse, error) {
if err := s.validateOptimizeRepositoryRequest(in); err != nil {
return nil, err
}
if err := s.optimizeRepository(ctx, in.GetRepository()); err != nil {
return nil, helper.ErrInternal(err)
}
return &gitalypb.OptimizeRepositoryResponse{}, nil
}
func (s *server) validateOptimizeRepositoryRequest(in *gitalypb.OptimizeRepositoryRequest) error {
if in.GetRepository() == nil {
return helper.ErrInvalidArgumentf("empty repository")
}
_, err := s.locator.GetRepoPath(in.GetRepository())
if err != nil {
return err
}
return nil
}
|
