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 objectpool
import (
"context"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/helper"
)
// Link will write the relative path to the object pool from the repository that
// is to join the pool. This does not trigger deduplication, which is the
// responsibility of the caller.
func (o *ObjectPool) Link(ctx context.Context, repo *gitalypb.Repository) error {
altPath, err := git.AlternatesPath(repo)
if err != nil {
return err
}
repoPath, err := helper.GetRepoPath(repo)
if err != nil {
return err
}
relPath, err := filepath.Rel(filepath.Join(repoPath, "objects"), o.FullPath())
if err != nil {
return err
}
remoteName := repo.GetGlRepository()
for k, v := range map[string]string{
fmt.Sprintf("remote.%s.url", remoteName): relPath,
fmt.Sprintf("remote.%s.fetch", remoteName): fmt.Sprintf("+refs/*:refs/remotes/%s/*", remoteName),
fmt.Sprintf("remote.%s.tagOpt", remoteName): "--no-tags",
} {
if err := o.setConfig(ctx, k, v); err != nil {
return err
}
}
return ioutil.WriteFile(altPath, []byte(filepath.Join(relPath, "objects")), 0644)
}
// Unlink removes the alternates file, so Git won't look there anymore
// It removes the remote from the object pool too,
func (o *ObjectPool) Unlink(ctx context.Context, repo *gitalypb.Repository) error {
if !o.Exists() {
return nil
}
// We need to use removeRemote, and can't leverage `git config --remove-section`
// as the latter doesn't clean up refs
remoteName := repo.GetGlRepository()
if err := o.removeRemote(ctx, remoteName); err != nil {
if present, err2 := o.hasRemote(ctx, remoteName); err2 != nil || present {
return err
}
}
altPath, err := git.AlternatesPath(repo)
if err != nil {
return err
}
return os.RemoveAll(altPath)
}
// Config options setting will leak the key value pairs in the logs. This makes
// this function not suitable for general usage, and scoped to this package.
// To be corrected in: https://gitlab.com/gitlab-org/gitaly/issues/1430
func (o *ObjectPool) setConfig(ctx context.Context, key, value string) error {
cmd, err := git.Command(ctx, o, "config", key, value)
if err != nil {
return err
}
return cmd.Wait()
}
|