Code refactor

pkg/cluster/cluster.go

@@ -0,0 +1,156 @@
+package cluster
+
+// Postgres ThirdPartyResource object i.e. Spilo
+
+import (
+	"fmt"
+
+	"github.com/Sirupsen/logrus"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/pkg/api/v1"
+	"k8s.io/client-go/rest"
+
+	"github.bus.zalan.do/acid/postgres-operator/pkg/spec"
+	"github.bus.zalan.do/acid/postgres-operator/pkg/util"
+	"github.bus.zalan.do/acid/postgres-operator/pkg/util/constants"
+)
+
+var patroniUsers = []string{"superuser", "replication", "admin"}
+
+//TODO: remove struct duplication
+type Config struct {
+	Namespace  string
+	KubeClient *kubernetes.Clientset //TODO: move clients to the better place?
+	RestClient *rest.RESTClient
+}
+
+type Cluster struct {
+	logger      *logrus.Entry
+	config      Config
+	etcdHost    string
+	dockerImage string
+	cluster     *spec.Postgresql
+	pgUsers     []pgUser
+}
+
+type pgUser struct {
+	username string
+	secretKey string
+	password []byte
+	flags []string
+}
+
+func New(cfg Config, spec *spec.Postgresql) *Cluster {
+	lg := logrus.WithField("pkg", "cluster").WithField("cluster-name", spec.Metadata.Name)
+
+	//TODO: check if image exist
+	dockerImage := fmt.Sprintf("registry.opensource.zalan.do/acid/spilo-%s", (*spec.Spec).PostgresqlParam.Version)
+
+	cluster := &Cluster{
+		config:      cfg,
+		cluster:     spec,
+		logger:      lg,
+		etcdHost:    constants.EtcdHost,
+		dockerImage: dockerImage,
+	}
+	cluster.init()
+
+	return cluster
+}
+
+func secretUserKey(userName string) string {
+	return fmt.Sprintf("%s-password", userName)
+}
+
+func (c *Cluster) init() {
+	for _, userName := range patroniUsers {
+		user := pgUser{
+			username: userName,
+			secretKey: secretUserKey(userName),
+			password: util.RandomPasswordBytes(constants.PasswordLength),
+		}
+		c.pgUsers = append(c.pgUsers, user)
+	}
+
+	for userName, userFlags := range (*c.cluster.Spec).Users {
+		user := pgUser{
+			username: userName,
+			secretKey: secretUserKey(userName),
+			password: util.RandomPasswordBytes(constants.PasswordLength),
+			flags: userFlags,
+		}
+	 	c.pgUsers = append(c.pgUsers, user)
+	}
+}
+
+func (c *Cluster) Create() error {
+	c.createEndPoint()
+	c.createService()
+	c.applySecrets()
+	c.createStatefulSet()
+
+	//TODO: wait for "spilo-role" label to appear on each pod
+
+	return nil
+}
+
+func (c *Cluster) Delete() error {
+	clusterName := (*c.cluster).Metadata.Name
+	nameSpace := c.config.Namespace
+	orphanDependents := false
+	deleteOptions := &v1.DeleteOptions{
+		OrphanDependents: &orphanDependents,
+	}
+
+	listOptions := v1.ListOptions{
+		LabelSelector: fmt.Sprintf("%s=%s", "spilo-cluster", clusterName),
+	}
+
+	kubeClient := c.config.KubeClient
+
+	podList, err := kubeClient.Pods(nameSpace).List(listOptions)
+	if err != nil {
+		c.logger.Errorf("Error: %+v", err)
+	}
+
+	err = kubeClient.StatefulSets(nameSpace).Delete(clusterName, deleteOptions)
+	if err != nil {
+		c.logger.Errorf("Error: %+v", err)
+	}
+	c.logger.Infof("StatefulSet %s.%s has been deleted\n", nameSpace, clusterName)
+
+	for _, pod := range podList.Items {
+		err = kubeClient.Pods(nameSpace).Delete(pod.Name, deleteOptions)
+		if err != nil {
+			c.logger.Errorf("Error while deleting Pod %s: %+v", pod.Name, err)
+			return err
+		}
+
+		c.logger.Infof("Pod %s.%s has been deleted\n", pod.Namespace, pod.Name)
+	}
+
+	serviceList, err := kubeClient.Services(nameSpace).List(listOptions)
+	if err != nil {
+		return err
+	}
+
+	for _, service := range serviceList.Items {
+		err = kubeClient.Services(nameSpace).Delete(service.Name, deleteOptions)
+		if err != nil {
+			c.logger.Errorf("Error while deleting Service %s: %+v", service.Name, err)
+
+			return err
+		}
+
+		c.logger.Infof("Service %s.%s has been deleted\n", service.Namespace, service.Name)
+	}
+
+	err = kubeClient.Secrets(nameSpace).Delete(clusterName, deleteOptions)
+	if err != nil {
+		c.logger.Errorf("Error while deleting Secret %s: %+v", clusterName, err)
+	}
+
+	//TODO: delete key from etcd
+
+	return nil
+}