rancher/rke
1
0
Fork 0
mirror of https://github.com/rancher/rke.git synced 2025-09-18 16:36:41 +00:00
Code Issues Releases Wiki Activity

Create a general retry function for k8s requests

Browse Source
This commit is contained in:
moelsayed
2017-12-21 01:11:50 +02:00
parent e6c3f50a4d
commit 7fdad4bcf9
6 changed files with 97 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
k8s/clusterrole.go
View File

@@ -1,61 +1,48 @@
package k8s
import (
"bytes"
"time"
rbacv1 "k8s.io/api/rbac/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
yamlutil "k8s.io/apimachinery/pkg/util/yaml"
"k8s.io/client-go/kubernetes"
)
func UpdateClusterRoleBindingFromYaml(k8sClient *kubernetes.Clientset, clusterRoleBindingYaml string) error {
clusterRoleBinding := rbacv1.ClusterRoleBinding{}
decoder := yamlutil.NewYAMLToJSONDecoder(bytes.NewReader([]byte(clusterRoleBindingYaml)))
err := decoder.Decode(&clusterRoleBinding)
if err != nil {
if err := decodeYamlResource(&clusterRoleBinding, clusterRoleBindingYaml); err != nil {
return err
}
for retries := 0; retries <= 5; retries++ {
if _, err = k8sClient.RbacV1().ClusterRoleBindings().Create(&clusterRoleBinding); err != nil {
if apierrors.IsAlreadyExists(err) {
if _, err = k8sClient.RbacV1().ClusterRoleBindings().Update(&clusterRoleBinding); err == nil {
return nil
}
}
} else {
return nil
}
time.Sleep(time.Second * 5)
}
return err
return retryTo(updateClusterRoleBinding, k8sClient, clusterRoleBinding, DefaultRetries, DefaultSleepSeconds)
}
func UpdateClusterRoleFromYaml(k8sClient *kubernetes.Clientset, clusterRoleYaml string) error {
clusterRole := rbacv1.ClusterRole{}
err := decodeYamlResource(&clusterRole, clusterRoleYaml)
if err != nil {
return err
}
for retries := 0; retries <= 5; retries++ {
if err = updateClusterRole(k8sClient, clusterRole); err != nil {
time.Sleep(time.Second * 5)
continue
}
return nil
}
return err
}
func updateClusterRole(k8sClient *kubernetes.Clientset, cr rbacv1.ClusterRole) error {
if _, err := k8sClient.RbacV1().ClusterRoles().Create(&cr); err != nil {
func updateClusterRoleBinding(k8sClient *kubernetes.Clientset, crb interface{}) error {
clusterRoleBinding := crb.(rbacv1.ClusterRoleBinding)
if _, err := k8sClient.RbacV1().ClusterRoleBindings().Create(&clusterRoleBinding); err != nil {
if !apierrors.IsAlreadyExists(err) {
return err
}
if _, err := k8sClient.RbacV1().ClusterRoles().Update(&cr); err != nil {
if _, err := k8sClient.RbacV1().ClusterRoleBindings().Update(&clusterRoleBinding); err != nil {
return err
}
}
return nil
}
func UpdateClusterRoleFromYaml(k8sClient *kubernetes.Clientset, clusterRoleYaml string) error {
clusterRole := rbacv1.ClusterRole{}
if err := decodeYamlResource(&clusterRole, clusterRoleYaml); err != nil {
return err
}
return retryTo(updateClusterRole, k8sClient, clusterRole, DefaultRetries, DefaultSleepSeconds)
}
func updateClusterRole(k8sClient *kubernetes.Clientset, cr interface{}) error {
clusterRole := cr.(rbacv1.ClusterRole)
if _, err := k8sClient.RbacV1().ClusterRoles().Create(&clusterRole); err != nil {
if !apierrors.IsAlreadyExists(err) {
return err
}
if _, err := k8sClient.RbacV1().ClusterRoles().Update(&clusterRole); err != nil {
return err
}
}