Port ctr to use client

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
This commit is contained in:
Michael Crosby
2017-06-05 16:17:37 -07:00
parent bdf9f5f738
commit 4c1af8fdd8
26 changed files with 526 additions and 1179 deletions

View File

@@ -1,7 +1,6 @@
package main
import (
"github.com/containerd/containerd/api/services/execution"
"github.com/pkg/errors"
"github.com/urfave/cli"
)
@@ -11,8 +10,9 @@ var killCommand = cli.Command{
Usage: "signal a container (default: SIGTERM)",
Flags: []cli.Flag{
cli.StringFlag{
Name: "id",
Usage: "id of the container",
Name: "signal, s",
Value: "SIGTERM",
Usage: "signal to send to the container",
},
cli.IntFlag{
Name: "pid",
@@ -26,53 +26,36 @@ var killCommand = cli.Command{
},
Action: func(context *cli.Context) error {
var (
id = context.String("id")
id = context.Args().First()
ctx, cancel = appContext(context)
)
defer cancel()
if id == "" {
return errors.New("container id must be provided")
}
sigstr := context.Args().First()
if sigstr == "" {
sigstr = "SIGTERM"
}
signal, err := parseSignal(sigstr)
signal, err := parseSignal(context.String("signal"))
if err != nil {
return err
}
pid := context.Int("pid")
all := context.Bool("all")
var (
pid = context.Int("pid")
all = context.Bool("all")
)
if pid > 0 && all {
return errors.New("enter a pid or all; not both")
}
killRequest := &execution.KillRequest{
ContainerID: id,
Signal: uint32(signal),
PidOrAll: &execution.KillRequest_Pid{
Pid: uint32(pid),
},
}
if all {
killRequest.PidOrAll = &execution.KillRequest_All{
All: true,
}
}
tasks, err := getTasksService(context)
client, err := newClient(context)
if err != nil {
return err
}
_, err = tasks.Kill(ctx, killRequest)
container, err := client.LoadContainer(ctx, id)
if err != nil {
return err
}
return nil
task, err := container.Task(ctx, nil)
if err != nil {
return err
}
return task.Kill(ctx, signal)
},
}