mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
Move "top" to daemon/top.go
This is part of an effort to break apart the deprecated server/ package Docker-DCO-1.1-Signed-off-by: Solomon Hykes <solomon@docker.com> (github: shykes)
This commit is contained in:
parent
49e61f62ce
commit
aa49632fef
4 changed files with 82 additions and 73 deletions
|
@ -158,6 +158,9 @@ func (daemon *Daemon) Install(eng *engine.Engine) error {
|
||||||
if err := eng.Register("container_changes", daemon.ContainerChanges); err != nil {
|
if err := eng.Register("container_changes", daemon.ContainerChanges); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if err := eng.Register("top", daemon.ContainerTop); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
79
daemon/top.go
Normal file
79
daemon/top.go
Normal file
|
@ -0,0 +1,79 @@
|
||||||
|
package daemon
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os/exec"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/docker/docker/engine"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (daemon *Daemon) ContainerTop(job *engine.Job) engine.Status {
|
||||||
|
if len(job.Args) != 1 && len(job.Args) != 2 {
|
||||||
|
return job.Errorf("Not enough arguments. Usage: %s CONTAINER [PS_ARGS]\n", job.Name)
|
||||||
|
}
|
||||||
|
var (
|
||||||
|
name = job.Args[0]
|
||||||
|
psArgs = "-ef"
|
||||||
|
)
|
||||||
|
|
||||||
|
if len(job.Args) == 2 && job.Args[1] != "" {
|
||||||
|
psArgs = job.Args[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
if container := daemon.Get(name); container != nil {
|
||||||
|
if !container.State.IsRunning() {
|
||||||
|
return job.Errorf("Container %s is not running", name)
|
||||||
|
}
|
||||||
|
pids, err := daemon.ExecutionDriver().GetPidsForContainer(container.ID)
|
||||||
|
if err != nil {
|
||||||
|
return job.Error(err)
|
||||||
|
}
|
||||||
|
output, err := exec.Command("ps", psArgs).Output()
|
||||||
|
if err != nil {
|
||||||
|
return job.Errorf("Error running ps: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
lines := strings.Split(string(output), "\n")
|
||||||
|
header := strings.Fields(lines[0])
|
||||||
|
out := &engine.Env{}
|
||||||
|
out.SetList("Titles", header)
|
||||||
|
|
||||||
|
pidIndex := -1
|
||||||
|
for i, name := range header {
|
||||||
|
if name == "PID" {
|
||||||
|
pidIndex = i
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if pidIndex == -1 {
|
||||||
|
return job.Errorf("Couldn't find PID field in ps output")
|
||||||
|
}
|
||||||
|
|
||||||
|
processes := [][]string{}
|
||||||
|
for _, line := range lines[1:] {
|
||||||
|
if len(line) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
fields := strings.Fields(line)
|
||||||
|
p, err := strconv.Atoi(fields[pidIndex])
|
||||||
|
if err != nil {
|
||||||
|
return job.Errorf("Unexpected pid '%s': %s", fields[pidIndex], err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, pid := range pids {
|
||||||
|
if pid == p {
|
||||||
|
// Make sure number of fields equals number of header titles
|
||||||
|
// merging "overhanging" fields
|
||||||
|
process := fields[:len(header)-1]
|
||||||
|
process = append(process, strings.Join(fields[len(header)-1:], " "))
|
||||||
|
processes = append(processes, process)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
out.SetJson("Processes", processes)
|
||||||
|
out.WriteTo(job.Stdout)
|
||||||
|
return engine.StatusOK
|
||||||
|
|
||||||
|
}
|
||||||
|
return job.Errorf("No such container: %s", name)
|
||||||
|
}
|
|
@ -7,8 +7,6 @@ package server
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os/exec"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/docker/docker/daemon"
|
"github.com/docker/docker/daemon"
|
||||||
|
@ -16,76 +14,6 @@ import (
|
||||||
"github.com/docker/docker/pkg/graphdb"
|
"github.com/docker/docker/pkg/graphdb"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (srv *Server) ContainerTop(job *engine.Job) engine.Status {
|
|
||||||
if len(job.Args) != 1 && len(job.Args) != 2 {
|
|
||||||
return job.Errorf("Not enough arguments. Usage: %s CONTAINER [PS_ARGS]\n", job.Name)
|
|
||||||
}
|
|
||||||
var (
|
|
||||||
name = job.Args[0]
|
|
||||||
psArgs = "-ef"
|
|
||||||
)
|
|
||||||
|
|
||||||
if len(job.Args) == 2 && job.Args[1] != "" {
|
|
||||||
psArgs = job.Args[1]
|
|
||||||
}
|
|
||||||
|
|
||||||
if container := srv.daemon.Get(name); container != nil {
|
|
||||||
if !container.State.IsRunning() {
|
|
||||||
return job.Errorf("Container %s is not running", name)
|
|
||||||
}
|
|
||||||
pids, err := srv.daemon.ExecutionDriver().GetPidsForContainer(container.ID)
|
|
||||||
if err != nil {
|
|
||||||
return job.Error(err)
|
|
||||||
}
|
|
||||||
output, err := exec.Command("ps", psArgs).Output()
|
|
||||||
if err != nil {
|
|
||||||
return job.Errorf("Error running ps: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
lines := strings.Split(string(output), "\n")
|
|
||||||
header := strings.Fields(lines[0])
|
|
||||||
out := &engine.Env{}
|
|
||||||
out.SetList("Titles", header)
|
|
||||||
|
|
||||||
pidIndex := -1
|
|
||||||
for i, name := range header {
|
|
||||||
if name == "PID" {
|
|
||||||
pidIndex = i
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if pidIndex == -1 {
|
|
||||||
return job.Errorf("Couldn't find PID field in ps output")
|
|
||||||
}
|
|
||||||
|
|
||||||
processes := [][]string{}
|
|
||||||
for _, line := range lines[1:] {
|
|
||||||
if len(line) == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
fields := strings.Fields(line)
|
|
||||||
p, err := strconv.Atoi(fields[pidIndex])
|
|
||||||
if err != nil {
|
|
||||||
return job.Errorf("Unexpected pid '%s': %s", fields[pidIndex], err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, pid := range pids {
|
|
||||||
if pid == p {
|
|
||||||
// Make sure number of fields equals number of header titles
|
|
||||||
// merging "overhanging" fields
|
|
||||||
process := fields[:len(header)-1]
|
|
||||||
process = append(process, strings.Join(fields[len(header)-1:], " "))
|
|
||||||
processes = append(processes, process)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
out.SetJson("Processes", processes)
|
|
||||||
out.WriteTo(job.Stdout)
|
|
||||||
return engine.StatusOK
|
|
||||||
|
|
||||||
}
|
|
||||||
return job.Errorf("No such container: %s", name)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (srv *Server) Containers(job *engine.Job) engine.Status {
|
func (srv *Server) Containers(job *engine.Job) engine.Status {
|
||||||
var (
|
var (
|
||||||
foundBefore bool
|
foundBefore bool
|
||||||
|
|
|
@ -93,7 +93,6 @@ func InitServer(job *engine.Job) engine.Status {
|
||||||
"history": srv.ImageHistory,
|
"history": srv.ImageHistory,
|
||||||
"viz": srv.ImagesViz,
|
"viz": srv.ImagesViz,
|
||||||
"log": srv.Log,
|
"log": srv.Log,
|
||||||
"top": srv.ContainerTop,
|
|
||||||
"load": srv.ImageLoad,
|
"load": srv.ImageLoad,
|
||||||
"build": srv.Build,
|
"build": srv.Build,
|
||||||
"pull": srv.ImagePull,
|
"pull": srv.ImagePull,
|
||||||
|
|
Loading…
Reference in a new issue