whatcontainer.go
No OneTemporary
Actions

Size

3 KB

Referenced Files

None

Subscribers

None

whatcontainer.go
View Options

	package main

	import (
	"bufio"
	"context"
	"devcentral.nasqueron.org/source/docker-processes/internal/stringutilities"
	"devcentral.nasqueron.org/source/docker-processes/pkg/dockerutils"
	"devcentral.nasqueron.org/source/docker-processes/pkg/process"
	"flag"
	"fmt"
	"github.com/docker/docker/api/types"
	"github.com/docker/docker/client"
	"log"
	"os"
	"strconv"
	"strings"
	)

	const DockerApiVersion = "1.37"

	type Config struct {
	Prepend bool
	Width int
	WithPosition bool
	Position int
	}

	func main() {
	config := parseArguments()
	scanner := bufio.NewScanner(os.Stdin)
	processesMap := getProcessesMap()

	for scanner.Scan() {
	line := addContainerName(scanner.Text(), processesMap, config)
	fmt.Println(line)
	}
	}

	func parseArguments() Config {
	config := Config{}

	positionPtr := flag.Int("p", -1, "the position of the field with the PID")
	widthPtr := flag.Int ("w", 20, "the amount of characters to use for container name")
	appendPtr := flag.Bool("append", false, "append the container name to the line (default behavior is to prepend)")

	flag.Parse()

	if *positionPtr > 0 {
	config.Position = *positionPtr
	config.WithPosition = true
	}

	config.Prepend = !*appendPtr
	config.Width = *widthPtr

	return config
	}

	func addContainerName (line string, processesMap map[int64]string, config Config) string {
	containerName := determineContainerName(line, processesMap, config)

	return formatLine(line, containerName, config)
	}

	func determineContainerName (line string, processesMap map[int64]string, config Config) string {
	fields := strings.Fields(line)

	for i, field := range fields {
	if !isValidFieldPosition(i + 1, config) {
	continue
	}

	pidCandidate, err := strconv.ParseInt(field, 10, 64)

	if err != nil {
	continue
	}

	if containerName, ok := processesMap[pidCandidate]; ok {
	return containerName
	}
	}

	return ""
	}

	func formatLine(line string, containerName string, config Config) string {
	if config.Prepend {
	name := stringutilities.PadField(containerName, config.Width)
	return fmt.Sprintf("%s %s", name, line)
	}

	if containerName == "" {
	return line
	}

	return fmt.Sprintf("%s %s", line, containerName)
	}

	func isValidFieldPosition(position int, config Config) bool {
	if config.WithPosition {
	return position == config.Position
	} else {
	// 1 for top and ps -ef
	// 2 for ps auxw
	return position < 3
	}
	}

	func getProcessesMap () map[int64]string {
	dockerClient, err := client.NewClientWithOpts(client.WithVersion(DockerApiVersion))
	if err != nil {
	log.Println("Can't connect to Docker engine.")
	os.Exit(1)
	}

	containers, err := dockerClient.ContainerList(context.Background(), types.ContainerListOptions{})
	if err != nil {
	panic(err)
	}

	processesMap := make(map[int64]string)

	psArgs := []string{"auxw"}
	for _, container := range containers {
	containerName := dockerutils.GetContainerName(container)

	response, err := dockerClient.ContainerTop(context.Background(), container.ID, psArgs)
	if err != nil {
	continue
	}
	for _, containerProcess := range response.Processes {
	processInfo := process.Parse(response.Titles, containerProcess)
	processesMap[processInfo.Pid] = containerName
	}
	}

	return processesMap
	}

File Metadata

Mime Type: text/x-c
Expires: Thu, Dec 26, 03:44 (20 h, 26 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 2310568
Default Alt Text: whatcontainer.go (3 KB)

whatcontainer.goNo OneTemporaryActions

whatcontainer.goView Options

File Metadata

Event Timeline

whatcontainer.go
No OneTemporary
Actions

whatcontainer.go
View Options