Code refactor

pkg/util/k8sutil/k8sutil.go

@@ -0,0 +1,66 @@
+package k8sutil
+
+import (
+	"k8s.io/client-go/kubernetes"
+	"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.bus.zalan.do/acid/postgres-operator/pkg/spec"
+	"github.bus.zalan.do/acid/postgres-operator/pkg/util/constants"
+)
+
+func RestConfig(kubeConfig string, outOfCluster bool) (config *rest.Config, err error) {
+	if outOfCluster {
+		/* out-of-cluster process */
+		rules := clientcmd.NewDefaultClientConfigLoadingRules()
+		overrides := &clientcmd.ConfigOverrides{}
+		rules.ExplicitPath = kubeConfig
+		config, err = clientcmd.NewNonInteractiveDeferredLoadingClientConfig(rules, overrides).ClientConfig()
+	} else {
+		/* in-cluster pod */
+		config, err = rest.InClusterConfig()
+	}
+
+	return
+}
+
+func KubernetesClient(config *rest.Config) (client *kubernetes.Clientset, err error) {
+	return kubernetes.NewForConfig(config)
+}
+
+func IsKubernetesResourceAlreadyExistError(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.APIPath = "/apis"
+	c.GroupVersion = &unversioned.GroupVersion{
+		Group:   constants.TPRVendor,
+		Version: constants.TPRApiVersion,
+	}
+	c.NegotiatedSerializer = serializer.DirectCodecFactory{CodecFactory: api.Codecs}
+
+	schemeBuilder := runtime.NewSchemeBuilder(
+		func(scheme *runtime.Scheme) error {
+			scheme.AddKnownTypes(
+				*c.GroupVersion,
+				&spec.Postgresql{},
+				&spec.PostgresqlList{},
+				&api.ListOptions{},
+				&api.DeleteOptions{},
+			)
+			return nil
+		})
+	schemeBuilder.AddToScheme(api.Scheme)
+
+	return rest.RESTClientFor(c)
+}

pkg/util/k8sutil/tpr_util.go

@@ -0,0 +1,28 @@
+package k8sutil
+
+import (
+	"fmt"
+	"time"
+
+	"k8s.io/client-go/rest"
+
+	"github.bus.zalan.do/acid/postgres-operator/pkg/util/constants"
+	"github.bus.zalan.do/acid/postgres-operator/pkg/util/retryutil"
+)
+
+func listClustersURI(ns string) string {
+	return fmt.Sprintf("/apis/%s/%s/namespaces/%s/%s", constants.TPRVendor, constants.TPRApiVersion, ns, constants.ResourceName)
+}
+
+func WaitTPRReady(restclient rest.Interface, interval, timeout time.Duration, ns string) error {
+	return retryutil.Retry(interval, int(timeout/interval), func() (bool, error) {
+		_, err := restclient.Get().RequestURI(listClustersURI(ns)).DoRaw()
+		if err != nil {
+			if ResourceNotFound(err) { // not set up yet. wait more.
+				return false, nil
+			}
+			return false, err
+		}
+		return true, nil
+	})
+}