103 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
package main
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"os"
 | 
						|
	"strings"
 | 
						|
	"text/tabwriter"
 | 
						|
 | 
						|
	"github.com/containers/libpod/cmd/podman/libpodruntime"
 | 
						|
	"github.com/containers/libpod/cmd/podman/shared"
 | 
						|
	"github.com/containers/libpod/libpod"
 | 
						|
	"github.com/pkg/errors"
 | 
						|
	"github.com/urfave/cli"
 | 
						|
)
 | 
						|
 | 
						|
var (
 | 
						|
	podTopFlags = []cli.Flag{
 | 
						|
		LatestFlag,
 | 
						|
		cli.BoolFlag{
 | 
						|
			Name:   "list-descriptors",
 | 
						|
			Hidden: true,
 | 
						|
		},
 | 
						|
	}
 | 
						|
	podTopDescription = fmt.Sprintf(`Display the running processes containers in a pod.  Specify format descriptors
 | 
						|
to alter the output.  You may run "podman pod top -l pid pcpu seccomp" to print
 | 
						|
the process ID, the CPU percentage and the seccomp mode of each process of
 | 
						|
the latest pod.
 | 
						|
%s
 | 
						|
`, getDescriptorString())
 | 
						|
 | 
						|
	podTopCommand = cli.Command{
 | 
						|
		Name:           "top",
 | 
						|
		Usage:          "Display the running processes of containers in a pod",
 | 
						|
		Description:    podTopDescription,
 | 
						|
		Flags:          sortFlags(podTopFlags),
 | 
						|
		Action:         podTopCmd,
 | 
						|
		ArgsUsage:      "POD-NAME [format descriptors]",
 | 
						|
		SkipArgReorder: true,
 | 
						|
		OnUsageError:   usageErrorHandler,
 | 
						|
	}
 | 
						|
)
 | 
						|
 | 
						|
func podTopCmd(c *cli.Context) error {
 | 
						|
	var pod *libpod.Pod
 | 
						|
	var err error
 | 
						|
	args := c.Args()
 | 
						|
 | 
						|
	if c.Bool("list-descriptors") {
 | 
						|
		descriptors, err := libpod.GetContainerPidInformationDescriptors()
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		fmt.Println(strings.Join(descriptors, "\n"))
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	if len(args) < 1 && !c.Bool("latest") {
 | 
						|
		return errors.Errorf("you must provide the name or id of a running pod")
 | 
						|
	}
 | 
						|
	if err := validateFlags(c, podTopFlags); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	runtime, err := libpodruntime.GetRuntime(c)
 | 
						|
	if err != nil {
 | 
						|
		return errors.Wrapf(err, "error creating libpod runtime")
 | 
						|
	}
 | 
						|
	defer runtime.Shutdown(false)
 | 
						|
 | 
						|
	var descriptors []string
 | 
						|
	if c.Bool("latest") {
 | 
						|
		descriptors = args
 | 
						|
		pod, err = runtime.GetLatestPod()
 | 
						|
	} else {
 | 
						|
		descriptors = args[1:]
 | 
						|
		pod, err = runtime.LookupPod(args[0])
 | 
						|
	}
 | 
						|
 | 
						|
	if err != nil {
 | 
						|
		return errors.Wrapf(err, "unable to lookup requested container")
 | 
						|
	}
 | 
						|
 | 
						|
	podStatus, err := shared.GetPodStatus(pod)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	if podStatus != "Running" {
 | 
						|
		return errors.Errorf("pod top can only be used on pods with at least one running container")
 | 
						|
	}
 | 
						|
 | 
						|
	psOutput, err := pod.GetPodPidInformation(descriptors)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	w := tabwriter.NewWriter(os.Stdout, 5, 1, 3, ' ', 0)
 | 
						|
	for _, proc := range psOutput {
 | 
						|
		fmt.Fprintln(w, proc)
 | 
						|
	}
 | 
						|
	w.Flush()
 | 
						|
	return nil
 | 
						|
}
 |