forked from docker/docs
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: Zhang Wei <[email protected]>
- Loading branch information
1 parent
07a7c06
commit eceb862
Showing
7 changed files
with
139 additions
and
121 deletions.
There are no files selected for viewing
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,130 @@ | ||
| package container | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "io" | ||
| "net/http/httputil" | ||
|
|
||
| "golang.org/x/net/context" | ||
|
|
||
| "github.com/Sirupsen/logrus" | ||
| "github.com/docker/docker/api/client" | ||
| "github.com/docker/docker/cli" | ||
| "github.com/docker/docker/pkg/signal" | ||
| "github.com/docker/engine-api/types" | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| type attachOptions struct { | ||
| noStdin bool | ||
| proxy bool | ||
| detachKeys string | ||
|
|
||
| container string | ||
| } | ||
|
|
||
| // NewAttachCommand creats a new cobra.Command for `docker attach` | ||
| func NewAttachCommand(dockerCli *client.DockerCli) *cobra.Command { | ||
| var opts attachOptions | ||
|
|
||
| cmd := &cobra.Command{ | ||
| Use: "attach [OPTIONS] CONTAINER", | ||
| Short: "Attach to a running container", | ||
| Args: cli.ExactArgs(1), | ||
| RunE: func(cmd *cobra.Command, args []string) error { | ||
| opts.container = args[0] | ||
| return runAttach(dockerCli, &opts) | ||
| }, | ||
| } | ||
| cmd.SetFlagErrorFunc(flagErrorFunc) | ||
|
|
||
| flags := cmd.Flags() | ||
| flags.BoolVar(&opts.noStdin, "no-stdin", false, "Do not attach STDIN") | ||
| flags.BoolVar(&opts.proxy, "sig-proxy", true, "Proxy all received signals to the process") | ||
| flags.StringVar(&opts.detachKeys, "detach-keys", "", "Override the key sequence for detaching a container") | ||
| return cmd | ||
| } | ||
|
|
||
| func runAttach(dockerCli *client.DockerCli, opts *attachOptions) error { | ||
| ctx := context.Background() | ||
|
|
||
| c, err := dockerCli.Client().ContainerInspect(ctx, opts.container) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if !c.State.Running { | ||
| return fmt.Errorf("You cannot attach to a stopped container, start it first") | ||
| } | ||
|
|
||
| if c.State.Paused { | ||
| return fmt.Errorf("You cannot attach to a paused container, unpause it first") | ||
| } | ||
|
|
||
| if err := dockerCli.CheckTtyInput(!opts.noStdin, c.Config.Tty); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if opts.detachKeys != "" { | ||
| dockerCli.ConfigFile().DetachKeys = opts.detachKeys | ||
| } | ||
|
|
||
| options := types.ContainerAttachOptions{ | ||
| Stream: true, | ||
| Stdin: !opts.noStdin && c.Config.OpenStdin, | ||
| Stdout: true, | ||
| Stderr: true, | ||
| DetachKeys: dockerCli.ConfigFile().DetachKeys, | ||
| } | ||
|
|
||
| var in io.ReadCloser | ||
| if options.Stdin { | ||
| in = dockerCli.In() | ||
| } | ||
|
|
||
| if opts.proxy && !c.Config.Tty { | ||
| sigc := dockerCli.ForwardAllSignals(ctx, opts.container) | ||
| defer signal.StopCatch(sigc) | ||
| } | ||
|
|
||
| resp, errAttach := dockerCli.Client().ContainerAttach(ctx, opts.container, options) | ||
| if errAttach != nil && errAttach != httputil.ErrPersistEOF { | ||
| // ContainerAttach returns an ErrPersistEOF (connection closed) | ||
| // means server met an error and put it in Hijacked connection | ||
| // keep the error and read detailed error message from hijacked connection later | ||
| return errAttach | ||
| } | ||
| defer resp.Close() | ||
|
|
||
| if c.Config.Tty && dockerCli.IsTerminalOut() { | ||
| height, width := dockerCli.GetTtySize() | ||
| // To handle the case where a user repeatedly attaches/detaches without resizing their | ||
| // terminal, the only way to get the shell prompt to display for attaches 2+ is to artificially | ||
| // resize it, then go back to normal. Without this, every attach after the first will | ||
| // require the user to manually resize or hit enter. | ||
| dockerCli.ResizeTtyTo(ctx, opts.container, height+1, width+1, false) | ||
|
|
||
| // After the above resizing occurs, the call to MonitorTtySize below will handle resetting back | ||
| // to the actual size. | ||
| if err := dockerCli.MonitorTtySize(ctx, opts.container, false); err != nil { | ||
| logrus.Debugf("Error monitoring TTY size: %s", err) | ||
| } | ||
| } | ||
| if err := dockerCli.HoldHijackedConnection(ctx, c.Config.Tty, in, dockerCli.Out(), dockerCli.Err(), resp); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if errAttach != nil { | ||
| return errAttach | ||
| } | ||
|
|
||
| _, status, err := dockerCli.GetExitCode(ctx, opts.container) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if status != 0 { | ||
| return cli.StatusError{StatusCode: status} | ||
| } | ||
|
|
||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters