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
94
95
96
97
98
99
100
101
102
103
104
105
|
package praefect
import (
"fmt"
"strings"
"github.com/urfave/cli/v2"
"gitlab.com/gitlab-org/gitaly/v16/internal/log"
"gitlab.com/gitlab-org/gitaly/v16/proto/go/gitalypb"
)
const (
setReplicationFactorCmdName = "set-replication-factor"
paramReplicationFactor = "replication-factor"
)
func newSetReplicationFactorCommand() *cli.Command {
return &cli.Command{
Name: setReplicationFactorCmdName,
Usage: "set a replication factor for a repository",
Description: `Set a new replication factor for a repository.
By default, repositories are replicated to all physical storages managed by Praefect. Use the set-replication-factor
subcommand to change this behavior. You should rarely set replication factors above 3.
When a new replication factor is specified, the subcommand:
- Assigns physical storages to or unassigns physical storages from the repository to meet the new replication factor.
The assigned physical storages are displayed on stdout.
- Returns an error if the new replication factor is either:
- More than the number of physical storages in the virtual storage.
- Less than one.
The authoritative physical storage is never unassigned because it:
- Accepts writes.
- Is the first storage that is assigned when setting a replication factor for a repository.
Example: praefect --config praefect.config.toml set-replication-factor --virtual-storage default --repository <relative_path_on_the_virtual_storage> --replication-factor 3`,
HideHelpCommand: true,
Action: setReplicationFactorAction,
Flags: []cli.Flag{
&cli.StringFlag{
Name: paramVirtualStorage,
Usage: "name of the repository's virtual storage",
Required: true,
},
&cli.StringFlag{
Name: paramRelativePath,
Usage: "relative path on the virtual storage of the repository to set the replication factor for",
Required: true,
},
&cli.UintFlag{
Name: paramReplicationFactor,
Usage: "replication factor to set",
Required: true,
},
},
Before: func(ctx *cli.Context) error {
if ctx.Args().Present() {
_ = cli.ShowSubcommandHelp(ctx)
return cli.Exit(unexpectedPositionalArgsError{Command: ctx.Command.Name}, 1)
}
return nil
},
}
}
func setReplicationFactorAction(appCtx *cli.Context) error {
logger := log.Default()
conf, err := getConfig(logger, appCtx.String(configFlagName))
if err != nil {
return err
}
virtualStorage := appCtx.String(paramVirtualStorage)
relativePath := appCtx.String(paramRelativePath)
replicationFactor := appCtx.Uint(paramReplicationFactor)
nodeAddr, err := getNodeAddress(conf)
if err != nil {
return err
}
ctx := appCtx.Context
conn, err := subCmdDial(ctx, nodeAddr, conf.Auth.Token, defaultDialTimeout)
if err != nil {
return fmt.Errorf("error dialing: %w", err)
}
defer conn.Close()
client := gitalypb.NewPraefectInfoServiceClient(conn)
resp, err := client.SetReplicationFactor(ctx, &gitalypb.SetReplicationFactorRequest{
VirtualStorage: virtualStorage,
RelativePath: relativePath,
ReplicationFactor: int32(replicationFactor),
})
if err != nil {
return err
}
fmt.Fprintf(appCtx.App.Writer, "current assignments: %v\n", strings.Join(resp.Storages, ", "))
return nil
}
|
