mirror of
https://github.com/rancher/rke.git
synced 2025-04-27 11:21:08 +00:00
46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
package addons
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"k8s.io/client-go/transport"
|
|
|
|
"github.com/rancher/rke/k8s"
|
|
"github.com/rancher/rke/templates"
|
|
"github.com/sirupsen/logrus"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func GetAddonsExecuteJob(addonName, nodeName, image string) (string, error) {
|
|
return getAddonJob(addonName, nodeName, image, false)
|
|
}
|
|
|
|
func GetAddonsDeleteJob(addonName, nodeName, image string) (string, error) {
|
|
return getAddonJob(addonName, nodeName, image, true)
|
|
}
|
|
|
|
func getAddonJob(addonName, nodeName, image string, isDelete bool) (string, error) {
|
|
jobConfig := map[string]string{
|
|
"AddonName": addonName,
|
|
"NodeName": nodeName,
|
|
"Image": image,
|
|
"DeleteJob": strconv.FormatBool(isDelete),
|
|
}
|
|
template, err := templates.CompileTemplateFromMap(templates.AddonJobTemplate, jobConfig)
|
|
logrus.Tracef("template for [%s] is: [%s]", addonName, template)
|
|
return template, err
|
|
}
|
|
|
|
func AddonJobExists(addonJobName, kubeConfigPath string, k8sWrapTransport transport.WrapperFunc) (bool, error) {
|
|
k8sClient, err := k8s.NewClient(kubeConfigPath, k8sWrapTransport)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
addonJobStatus, err := k8s.GetK8sJobStatus(k8sClient, addonJobName, metav1.NamespaceSystem)
|
|
if err != nil {
|
|
return false, fmt.Errorf("Failed to get job [%s] status: %v", addonJobName, err)
|
|
}
|
|
return addonJobStatus.Created, nil
|
|
}
|