forked from percona/percona-server-mongodb-operator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
job.go
114 lines (103 loc) · 2.88 KB
/
job.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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package backup
import (
"math/rand"
"time"
batchv1 "k8s.io/api/batch/v1"
batchv1b "k8s.io/api/batch/v1beta1"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
api "github.com/percona/percona-server-mongodb-operator/pkg/apis/psmdb/v1"
)
func init() {
rand.Seed(time.Now().UnixNano())
}
const genSymbols = "abcdefghijklmnopqrstuvwxyz1234567890"
func BackupCronJob(backup *api.BackupTaskSpec, crName, namespace string, backupSpec api.BackupSpec, imagePullSecrets []corev1.LocalObjectReference) *batchv1b.CronJob {
backupPod := corev1.PodSpec{
RestartPolicy: corev1.RestartPolicyNever,
ImagePullSecrets: imagePullSecrets,
ServiceAccountName: backupSpec.ServiceAccountName,
Containers: []corev1.Container{
{
Name: "backup",
Image: backupSpec.Image,
Command: []string{"sh"},
Env: []corev1.EnvVar{
{
Name: "psmdbCluster",
Value: crName,
},
{
Name: "suffix",
Value: genRandString(5),
},
},
Args: newBackupCronJobContainerArgs(backup),
SecurityContext: backupSpec.ContainerSecurityContext,
},
},
SecurityContext: backupSpec.PodSecurityContext,
}
return &batchv1b.CronJob{
TypeMeta: metav1.TypeMeta{
APIVersion: "batch/v1beta1",
Kind: "CronJob",
},
ObjectMeta: metav1.ObjectMeta{
Name: crName + "-backup-" + backup.Name,
Namespace: namespace,
Labels: NewBackupCronJobLabels(crName),
},
Spec: batchv1b.CronJobSpec{
Schedule: backup.Schedule,
ConcurrencyPolicy: batchv1b.ForbidConcurrent,
JobTemplate: batchv1b.JobTemplateSpec{
ObjectMeta: metav1.ObjectMeta{
Labels: NewBackupCronJobLabels(crName),
},
Spec: batchv1.JobSpec{
Template: corev1.PodTemplateSpec{
Spec: backupPod,
},
},
},
},
}
}
func NewBackupCronJobLabels(crName string) map[string]string {
return map[string]string{
"app.kubernetes.io/name": "percona-server-mongodb",
"app.kubernetes.io/instance": crName,
"app.kubernetes.io/replset": "general",
"app.kubernetes.io/managed-by": "percona-server-mongodb-operator",
"app.kubernetes.io/component": "backup-schedule",
"app.kubernetes.io/part-of": "percona-server-mongodb",
}
}
func newBackupCronJobContainerArgs(backup *api.BackupTaskSpec) []string {
return []string{
"-c",
`
cat <<-EOF | /usr/bin/kubectl apply -f -
apiVersion: psmdb.percona.com/v1
kind: PerconaServerMongoDBBackup
metadata:
name: "cron-${psmdbCluster:0:16}-$(date -u "+%Y%m%d%H%M%S")-${suffix}"
labels:
ancestor: "` + backup.Name + `"
cluster: "${psmdbCluster}"
type: "cron"
spec:
psmdbCluster: "${psmdbCluster}"
storageName: "` + backup.StorageName + `"
EOF
`,
}
}
func genRandString(ln int) string {
b := make([]byte, ln)
for i := range b {
b[i] = genSymbols[rand.Intn(len(genSymbols))]
}
return string(b)
}