107 lines
3.1 KiB
Go
107 lines
3.1 KiB
Go
package k8sutil
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"k8s.io/client-go/kubernetes"
|
|
v1beta1 "k8s.io/client-go/kubernetes/typed/apps/v1beta1"
|
|
v1core "k8s.io/client-go/kubernetes/typed/core/v1"
|
|
extensions "k8s.io/client-go/kubernetes/typed/extensions/v1beta1"
|
|
"k8s.io/client-go/pkg/api"
|
|
apierrors "k8s.io/client-go/pkg/api/errors"
|
|
"k8s.io/client-go/pkg/api/unversioned"
|
|
"k8s.io/client-go/pkg/runtime"
|
|
"k8s.io/client-go/pkg/runtime/serializer"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
|
|
"github.com/zalando-incubator/postgres-operator/pkg/spec"
|
|
"github.com/zalando-incubator/postgres-operator/pkg/util/constants"
|
|
"github.com/zalando-incubator/postgres-operator/pkg/util/retryutil"
|
|
)
|
|
|
|
type KubernetesClient struct {
|
|
v1core.SecretsGetter
|
|
v1core.ServicesGetter
|
|
v1core.EndpointsGetter
|
|
v1core.PodsGetter
|
|
v1core.PersistentVolumesGetter
|
|
v1core.PersistentVolumeClaimsGetter
|
|
v1core.ConfigMapsGetter
|
|
v1beta1.StatefulSetsGetter
|
|
extensions.ThirdPartyResourcesGetter
|
|
}
|
|
|
|
func NewFromKubernetesInterface(src kubernetes.Interface) (c KubernetesClient) {
|
|
c = KubernetesClient{}
|
|
c.PodsGetter = src.CoreV1()
|
|
c.ServicesGetter = src.CoreV1()
|
|
c.EndpointsGetter = src.CoreV1()
|
|
c.SecretsGetter = src.CoreV1()
|
|
c.ConfigMapsGetter = src.CoreV1()
|
|
c.PersistentVolumeClaimsGetter = src.CoreV1()
|
|
c.PersistentVolumesGetter = src.CoreV1()
|
|
c.StatefulSetsGetter = src.AppsV1beta1()
|
|
c.ThirdPartyResourcesGetter = src.ExtensionsV1beta1()
|
|
return
|
|
}
|
|
|
|
func RestConfig(kubeConfig string, outOfCluster bool) (*rest.Config, error) {
|
|
if outOfCluster {
|
|
return clientcmd.BuildConfigFromFlags("", kubeConfig)
|
|
}
|
|
return rest.InClusterConfig()
|
|
}
|
|
|
|
func ClientSet(config *rest.Config) (client *kubernetes.Clientset, err error) {
|
|
return kubernetes.NewForConfig(config)
|
|
}
|
|
|
|
func ResourceAlreadyExists(err error) bool {
|
|
return apierrors.IsAlreadyExists(err)
|
|
}
|
|
|
|
func ResourceNotFound(err error) bool {
|
|
return apierrors.IsNotFound(err)
|
|
}
|
|
|
|
func KubernetesRestClient(c *rest.Config) (*rest.RESTClient, error) {
|
|
c.GroupVersion = &unversioned.GroupVersion{Version: constants.K8sVersion}
|
|
c.APIPath = constants.K8sAPIPath
|
|
c.NegotiatedSerializer = serializer.DirectCodecFactory{CodecFactory: api.Codecs}
|
|
|
|
schemeBuilder := runtime.NewSchemeBuilder(
|
|
func(scheme *runtime.Scheme) error {
|
|
scheme.AddKnownTypes(
|
|
unversioned.GroupVersion{
|
|
Group: constants.TPRVendor,
|
|
Version: constants.TPRApiVersion,
|
|
},
|
|
&spec.Postgresql{},
|
|
&spec.PostgresqlList{},
|
|
&api.ListOptions{},
|
|
&api.DeleteOptions{},
|
|
)
|
|
return nil
|
|
})
|
|
if err := schemeBuilder.AddToScheme(api.Scheme); err != nil {
|
|
return nil, fmt.Errorf("could not apply functions to register PostgreSQL TPR type: %v", err)
|
|
}
|
|
|
|
return rest.RESTClientFor(c)
|
|
}
|
|
|
|
func WaitTPRReady(restclient rest.Interface, interval, timeout time.Duration, ns string) error {
|
|
return retryutil.Retry(interval, timeout, func() (bool, error) {
|
|
_, err := restclient.Get().RequestURI(fmt.Sprintf(constants.ListClustersURITemplate, ns)).DoRaw()
|
|
if err != nil {
|
|
if ResourceNotFound(err) { // not set up yet. wait more.
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
})
|
|
}
|