85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
|
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
|
||
|
|
}
|