forked from openshift/oc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
exec_remote.go
70 lines (63 loc) · 1.82 KB
/
exec_remote.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package rsync
import (
"io"
"strings"
"k8s.io/klog"
"k8s.io/cli-runtime/pkg/genericclioptions"
"k8s.io/client-go/kubernetes"
restclient "k8s.io/client-go/rest"
kexec "k8s.io/kubernetes/pkg/kubectl/cmd/exec"
)
// remoteExecutor will execute commands on a given pod/container by using the kube Exec command
type remoteExecutor struct {
Namespace string
PodName string
ContainerName string
SuggestedCmdUsage string
Client kubernetes.Interface
Config *restclient.Config
}
// Ensure it implements the executor interface
var _ executor = &remoteExecutor{}
// Execute will run a command in a pod
func (e *remoteExecutor) Execute(command []string, in io.Reader, out, errOut io.Writer) error {
klog.V(3).Infof("Remote executor running command: %s", strings.Join(command, " "))
execOptions := &kexec.ExecOptions{
StreamOptions: kexec.StreamOptions{
Namespace: e.Namespace,
PodName: e.PodName,
ContainerName: e.ContainerName,
IOStreams: genericclioptions.IOStreams{
In: in,
Out: out,
ErrOut: errOut,
},
Stdin: in != nil,
},
SuggestedCmdUsage: e.SuggestedCmdUsage,
Executor: &kexec.DefaultRemoteExecutor{},
PodClient: e.Client.CoreV1(),
Config: e.Config,
Command: command,
}
err := execOptions.Validate()
if err != nil {
klog.V(4).Infof("Error from remote command validation: %v", err)
return err
}
err = execOptions.Run()
if err != nil {
klog.V(4).Infof("Error from remote execution: %v", err)
}
return err
}
func newRemoteExecutor(o *RsyncOptions) executor {
return &remoteExecutor{
Namespace: o.Namespace,
PodName: o.PodName(),
ContainerName: o.ContainerName,
SuggestedCmdUsage: o.SuggestedCmdUsage,
Config: o.Config,
Client: o.Client,
}
}