mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-11-04 07:49:35 +00:00
119 lines
3.1 KiB
Go
119 lines
3.1 KiB
Go
/*
|
|
Copyright 2014 The Kubernetes Authors All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/client/portforward"
|
|
cmdutil "github.com/GoogleCloudPlatform/kubernetes/pkg/kubectl/cmd/util"
|
|
"github.com/golang/glog"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const (
|
|
portforward_example = `
|
|
// listens on ports 5000 and 6000 locally, forwarding data to/from ports 5000 and 6000 in the pod
|
|
$ kubectl port-forward -p mypod 5000 6000
|
|
|
|
// listens on port 8888 locally, forwarding to 5000 in the pod
|
|
$ kubectl port-forward -p mypod 8888:5000
|
|
|
|
// listens on a random port locally, forwarding to 5000 in the pod
|
|
$ kubectl port-forward -p mypod :5000
|
|
|
|
// listens on a random port locally, forwarding to 5000 in the pod
|
|
$ kubectl port-forward -p mypod 0:5000`
|
|
)
|
|
|
|
func NewCmdPortForward(f *cmdutil.Factory) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "port-forward -p POD [LOCAL_PORT:]REMOTE_PORT [...[LOCAL_PORT_N:]REMOTE_PORT_N]",
|
|
Short: "Forward one or more local ports to a pod.",
|
|
Long: "Forward one or more local ports to a pod.",
|
|
Example: portforward_example,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
err := RunPortForward(f, cmd, args)
|
|
cmdutil.CheckErr(err)
|
|
},
|
|
}
|
|
cmd.Flags().StringP("pod", "p", "", "Pod name")
|
|
cmd.MarkFlagRequired("pod")
|
|
// TODO support UID
|
|
return cmd
|
|
}
|
|
|
|
func RunPortForward(f *cmdutil.Factory, cmd *cobra.Command, args []string) error {
|
|
podName := cmdutil.GetFlagString(cmd, "pod")
|
|
if len(podName) == 0 {
|
|
return cmdutil.UsageError(cmd, "POD is required for exec")
|
|
}
|
|
|
|
if len(args) < 1 {
|
|
return cmdutil.UsageError(cmd, "at least 1 PORT is required for port-forward")
|
|
}
|
|
|
|
namespace, err := f.DefaultNamespace()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client, err := f.Client()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pod, err := client.Pods(namespace).Get(podName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if pod.Status.Phase != api.PodRunning {
|
|
glog.Fatalf("Unable to execute command because pod is not running. Current status=%v", pod.Status.Phase)
|
|
}
|
|
|
|
config, err := f.ClientConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
signals := make(chan os.Signal, 1)
|
|
signal.Notify(signals, os.Interrupt)
|
|
defer signal.Stop(signals)
|
|
|
|
stopCh := make(chan struct{}, 1)
|
|
go func() {
|
|
<-signals
|
|
close(stopCh)
|
|
}()
|
|
|
|
req := client.RESTClient.Get().
|
|
Prefix("proxy").
|
|
Resource("nodes").
|
|
Name(pod.Spec.Host).
|
|
Suffix("portForward", namespace, podName)
|
|
|
|
pf, err := portforward.New(req, config, args, stopCh)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return pf.ForwardPorts()
|
|
}
|