-
Notifications
You must be signed in to change notification settings - Fork 0
/
backup_push.go
94 lines (77 loc) · 3.14 KB
/
backup_push.go
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
package gp
import (
"strconv"
"github.com/apecloud/dataprotection-wal-g/internal/databases/greenplum"
"github.com/apecloud/dataprotection-wal-g/internal"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"github.com/wal-g/tracelog"
)
const (
backupPushShortDescription = "Makes backup and uploads it to storage"
permanentFlag = "permanent"
fullBackupFlag = "full"
addUserDataFlag = "add-user-data"
deltaFromUserDataFlag = "delta-from-user-data"
deltaFromNameFlag = "delta-from-name"
permanentShorthand = "p"
fullBackupShorthand = "f"
)
var (
// backupPushCmd represents the backupPush command
backupPushCmd = &cobra.Command{
Use: "backup-push",
Short: backupPushShortDescription, // TODO : improve description
Args: cobra.NoArgs,
Run: func(cmd *cobra.Command, args []string) {
internal.ConfigureLimiters()
if deltaFromName == "" {
deltaFromName = viper.GetString(internal.DeltaFromNameSetting)
}
if deltaFromUserData == "" {
deltaFromUserData = viper.GetString(internal.DeltaFromUserDataSetting)
}
if userDataRaw == "" {
userDataRaw = viper.GetString(internal.SentinelUserDataSetting)
}
userData, err := internal.UnmarshalSentinelUserData(userDataRaw)
tracelog.ErrorLogger.FatalfOnError("Failed to unmarshal the provided UserData: %s", err)
deltaBaseSelector, err := internal.NewDeltaBaseSelector(
deltaFromName, deltaFromUserData, greenplum.NewGenericMetaFetcher())
tracelog.ErrorLogger.FatalfOnError("Failed to find the base for a delta backup: %s", err)
logsDir := viper.GetString(internal.GPLogsDirectory)
segPollInterval, err := internal.GetDurationSetting(internal.GPSegmentsPollInterval)
tracelog.ErrorLogger.FatalOnError(err)
segPollRetries := viper.GetInt(internal.GPSegmentsPollRetries)
arguments := greenplum.NewBackupArguments(permanent, fullBackup, userData, prepareSegmentFwdArgs(), logsDir,
segPollInterval, segPollRetries, deltaBaseSelector)
backupHandler, err := greenplum.NewBackupHandler(arguments)
tracelog.ErrorLogger.FatalOnError(err)
backupHandler.HandleBackupPush()
},
}
permanent = false
userDataRaw = ""
deltaFromName = ""
deltaFromUserData = ""
fullBackup = false
)
// prepare arguments that are going to be forwarded to segments
func prepareSegmentFwdArgs() []greenplum.SegmentFwdArg {
return []greenplum.SegmentFwdArg{
{Name: permanentFlag, Value: strconv.FormatBool(permanent)},
}
}
func init() {
cmd.AddCommand(backupPushCmd)
backupPushCmd.Flags().BoolVarP(&permanent, permanentFlag, permanentShorthand,
false, "Pushes permanent backup")
backupPushCmd.Flags().BoolVarP(&fullBackup, fullBackupFlag, fullBackupShorthand,
false, "Make full backup-push")
backupPushCmd.Flags().StringVar(&userDataRaw, addUserDataFlag,
"", "Write the provided user data to the backup sentinel and metadata files.")
backupPushCmd.Flags().StringVar(&deltaFromName, deltaFromNameFlag,
"", "Select the backup specified by name as the target for the delta backup")
backupPushCmd.Flags().StringVar(&deltaFromUserData, deltaFromUserDataFlag,
"", "Select the backup specified by UserData as the target for the delta backup")
}