Command exec inside the Pod method

cmd/main.go

@@ -55,6 +55,7 @@ func ControllerConfig() *controller.Config {
 	}
 
 	return &controller.Config{
+		RestConfig: restConfig,
 		KubeClient: client,
 		RestClient: restClient,
 	}

glide.lock

@@ -1,5 +1,5 @@
-hash: 5f99f20e4693c230cd709102e4bba350f87205bda86b2c6148ac79da05d7ae47
+hash: b2ea8a2133857a3860d2a63eec76a9eed71565f69723848ee17b77aacc49905b
-updated: 2017-05-10T11:51:19.667157509+02:00
+updated: 2017-05-10T14:44:44.907924292+02:00
 imports:
 - name: cloud.google.com/go
   version: 3b1ae45394a234c385be014e9a488f2bb6eef821
@@ -85,7 +85,7 @@ imports:
 - name: github.com/kr/text
   version: 7cafcd837844e784b526369c9bce262804aebc60
 - name: github.com/lib/pq
-  version: 0477eb88c5ca4009cb281f13c90633375b6a9987
+  version: 2704adc878c21e1329f46f6e56a1c387d788ff94
   subpackages:
   - oid
 - name: github.com/mailru/easyjson

glide.yaml

@@ -16,6 +16,7 @@ import:
 - package: k8s.io/apimachinery
   subpackages:
   - pkg/util/json
+  - pkg/util/remotecommand
 - package: k8s.io/client-go
   version: ^2.0.0
 - package: k8s.io/kubernetes

pkg/controller/controller.go

@@ -18,6 +18,7 @@ import (
 )
 
 type Config struct {
+	RestConfig          *rest.Config
 	KubeClient          *kubernetes.Clientset
 	RestClient          *rest.RESTClient
 	EtcdClient          etcdclient.KeysAPI

pkg/controller/exec.go

@@ -0,0 +1,61 @@
+package controller
+
+import (
+	"bytes"
+	"fmt"
+
+	"k8s.io/client-go/pkg/api"
+	"k8s.io/kubernetes/pkg/client/unversioned/remotecommand"
+	remotecommandconsts "k8s.io/apimachinery/pkg/util/remotecommand"
+
+	"github.bus.zalan.do/acid/postgres-operator/pkg/spec"
+)
+
+func (c *Controller) ExecCommand(podName spec.NamespacedName, command []string) (string, error) {
+	var (
+		execOut bytes.Buffer
+		execErr bytes.Buffer
+	)
+
+	pod, err := c.KubeClient.Pods(podName.Namespace).Get(podName.Name)
+	if err != nil {
+		return "", fmt.Errorf("Can't get Pod info: %s", err)
+	}
+
+	if len(pod.Spec.Containers) != 1 {
+		return "", fmt.Errorf("Can't determine which container to use")
+	}
+
+	req := c.RestClient.Post().
+		Resource("pods").
+		Name(podName.Name).
+		Namespace(podName.Namespace).
+		SubResource("exec")
+	req.VersionedParams(&api.PodExecOptions{
+		Container: pod.Spec.Containers[0].Name,
+		Command:   command,
+		Stdout:    true,
+		Stderr:    true,
+	}, api.ParameterCodec)
+
+	exec, err := remotecommand.NewExecutor(c.RestConfig, "POST", req.URL())
+	if err != nil {
+		return "", fmt.Errorf("Failed to init executor: %s", err)
+	}
+
+	err = exec.Stream(remotecommand.StreamOptions{
+		SupportedProtocols: remotecommandconsts.SupportedStreamingProtocols,
+		Stdout:             &execOut,
+		Stderr:             &execErr,
+	})
+
+	if err != nil {
+		return "", fmt.Errorf("Can't execute: %s", err)
+	}
+
+	if execErr.Len() > 0 {
+		return "", fmt.Errorf("Stderr: %s", execErr.String())
+	}
+
+	return execOut.String(), nil
+}