109 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			109 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// +build linux
 | 
						|
 | 
						|
/*
 | 
						|
   Copyright The containerd Authors.
 | 
						|
 | 
						|
   Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
   you may not use this file except in compliance with the License.
 | 
						|
   You may obtain a copy of the License at
 | 
						|
 | 
						|
       http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 | 
						|
   Unless required by applicable law or agreed to in writing, software
 | 
						|
   distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
   See the License for the specific language governing permissions and
 | 
						|
   limitations under the License.
 | 
						|
*/
 | 
						|
 | 
						|
package tasks
 | 
						|
 | 
						|
import (
 | 
						|
	"encoding/json"
 | 
						|
	"errors"
 | 
						|
	"fmt"
 | 
						|
	"os"
 | 
						|
	"text/tabwriter"
 | 
						|
 | 
						|
	"github.com/containerd/cgroups"
 | 
						|
	"github.com/containerd/containerd/cmd/ctr/commands"
 | 
						|
	"github.com/containerd/typeurl"
 | 
						|
	"github.com/urfave/cli"
 | 
						|
)
 | 
						|
 | 
						|
func init() {
 | 
						|
	// metricsCommand is only added on Linux as github.com/containerd/cgroups
 | 
						|
	// does not compile on darwin or windows
 | 
						|
	Command.Subcommands = append(Command.Subcommands, metricsCommand)
 | 
						|
}
 | 
						|
 | 
						|
const (
 | 
						|
	formatFlag  = "format"
 | 
						|
	formatTable = "table"
 | 
						|
	formatJSON  = "json"
 | 
						|
)
 | 
						|
 | 
						|
var metricsCommand = cli.Command{
 | 
						|
	Name:      "metrics",
 | 
						|
	Usage:     "get a single data point of metrics for a task with the built-in Linux runtime",
 | 
						|
	ArgsUsage: "CONTAINER",
 | 
						|
	Aliases:   []string{"metric"},
 | 
						|
	Flags: []cli.Flag{
 | 
						|
		cli.StringFlag{
 | 
						|
			Name:  formatFlag,
 | 
						|
			Usage: `"table" or "json"`,
 | 
						|
			Value: formatTable,
 | 
						|
		},
 | 
						|
	},
 | 
						|
	Action: func(context *cli.Context) error {
 | 
						|
		client, ctx, cancel, err := commands.NewClient(context)
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		defer cancel()
 | 
						|
		container, err := client.LoadContainer(ctx, context.Args().First())
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		task, err := container.Task(ctx, nil)
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		metric, err := task.Metrics(ctx)
 | 
						|
		if err != nil {
 | 
						|
			return nil
 | 
						|
		}
 | 
						|
		anydata, err := typeurl.UnmarshalAny(metric.Data)
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		data, ok := anydata.(*cgroups.Metrics)
 | 
						|
		if !ok {
 | 
						|
			return errors.New("cannot convert metric data to cgroups.Metrics")
 | 
						|
		}
 | 
						|
 | 
						|
		switch context.String(formatFlag) {
 | 
						|
		case formatTable:
 | 
						|
			w := tabwriter.NewWriter(os.Stdout, 1, 8, 4, ' ', 0)
 | 
						|
			fmt.Fprintf(w, "ID\tTIMESTAMP\t\n")
 | 
						|
			fmt.Fprintf(w, "%s\t%s\t\n\n", metric.ID, metric.Timestamp)
 | 
						|
 | 
						|
			fmt.Fprintf(w, "METRIC\tVALUE\t\n")
 | 
						|
			fmt.Fprintf(w, "memory.usage_in_bytes\t%d\t\n", data.Memory.Usage.Usage)
 | 
						|
			fmt.Fprintf(w, "memory.stat.cache\t%d\t\n", data.Memory.TotalCache)
 | 
						|
			fmt.Fprintf(w, "cpuacct.usage\t%d\t\n", data.CPU.Usage.Total)
 | 
						|
			fmt.Fprintf(w, "cpuacct.usage_percpu\t%v\t\n", data.CPU.Usage.PerCPU)
 | 
						|
			return w.Flush()
 | 
						|
		case formatJSON:
 | 
						|
			marshaledJSON, err := json.MarshalIndent(data, "", "  ")
 | 
						|
			if err != nil {
 | 
						|
				return err
 | 
						|
			}
 | 
						|
			fmt.Println(string(marshaledJSON))
 | 
						|
			return nil
 | 
						|
		default:
 | 
						|
			return errors.New("format must be table or json")
 | 
						|
		}
 | 
						|
	},
 | 
						|
}
 |