104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package generator
|
|
|
|
import (
|
|
"fmt"
|
|
"katenary/helm"
|
|
"katenary/logger"
|
|
"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"`
|
|
Image string `yaml:"image"`
|
|
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)
|
|
}
|
|
|
|
// 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
|
|
logger.Magenta(ICON_RBAC, "Generating ServiceAccount, Role and RoleBinding for cron jobs", deployName)
|
|
fileGeneratorChan <- sa
|
|
fileGeneratorChan <- role
|
|
fileGeneratorChan <- roleBinding
|
|
|
|
index := len(crons) - 1 // will be 0 when there is only one cron - made to name crons
|
|
|
|
// 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
|
|
|
|
if cron.Image == "" {
|
|
cron.Image = "bitnami/kubectl"
|
|
}
|
|
|
|
name := deployName
|
|
if index > 0 {
|
|
name = fmt.Sprintf("%s-%d", deployName, index)
|
|
index++
|
|
}
|
|
|
|
cronTab := helm.NewCrontab(
|
|
name,
|
|
cron.Image,
|
|
cmd,
|
|
cron.Schedule,
|
|
sa,
|
|
)
|
|
// add crontab
|
|
suffix := ""
|
|
if index > 0 {
|
|
suffix = fmt.Sprintf("%d", index)
|
|
}
|
|
logger.Magenta(ICON_CRON, "Generating crontab", deployName, suffix)
|
|
fileGeneratorChan <- cronTab
|
|
}
|
|
|
|
return
|
|
}
|