WIP, possibility to create cronjob

This commit is contained in:
2022-05-24 14:44:31 +02:00
parent 8cf3ff9f73
commit 5fd1e16f1a
8 changed files with 251 additions and 1 deletions

84
generator/crontabs.go Normal file
View File

@@ -0,0 +1,84 @@
package generator
import (
"fmt"
"katenary/helm"
"log"
"github.com/compose-spec/compose-go/types"
"gopkg.in/yaml.v3"
)
const (
cronMulti = `pods=$(kubectl get pods --selector=%s/component=%s,%s/resource=deployment -o jsonpath='{.items[*].metadata.name}')`
cronMultiCmd = `
for pod in $pods; do
kubectl exec -i $pod -c %s -- sh -c '%s'
done`
cronSingle = `pod=$(kubectl get pods --selector=%s/component=%s,%s/resource=deployment -o jsonpath='{.items[0].metadata.name}')`
cronCmd = `
kubectl exec -i $pod -c %s -- sh -c '%s'`
)
type CronDef struct {
Command string `yaml:"command"`
Schedule string `yaml:"schedule"`
Multi bool `yaml:"allPods,omitempty"`
}
func buildCrontab(deployName string, deployment *helm.Deployment, s *types.ServiceConfig, fileGeneratorChan HelmFileGenerator) {
// get the cron label from the service
var crondef string
var ok bool
if crondef, ok = s.Labels[helm.LABEL_CRON]; !ok {
return
}
// parse yaml
crons := []CronDef{}
err := yaml.Unmarshal([]byte(crondef), &crons)
if err != nil {
log.Fatalf("error: %v", err)
}
log.Println(crons)
// create a serviceAccount
sa := helm.NewServiceAccount(deployName)
// create a role
role := helm.NewCronRole(deployName)
// create a roleBinding
roleBinding := helm.NewRoleBinding(deployName, sa, role)
// make generation
fileGeneratorChan <- sa
fileGeneratorChan <- role
fileGeneratorChan <- roleBinding
// create crontabs
for _, cron := range crons {
var cmd, podget string
if cron.Multi {
podget = cronMulti
cmd = cronMultiCmd
} else {
podget = cronSingle
cmd = cronCmd
}
podget = fmt.Sprintf(podget, helm.K, deployName, helm.K)
cmd = fmt.Sprintf(cmd, s.Name, cron.Command)
cmd = podget + cmd
cronTab := helm.NewCrontab(
deployName,
"bitnami/kubectl",
cmd,
cron.Schedule,
sa,
)
// add crontab
fileGeneratorChan <- cronTab
}
return
}