Adding upstream version 1.34.4.

Signed-off-by: Daniel Baumann <daniel@debian.org>

plugins/inputs/ecs/ecs.go

@@ -0,0 +1,244 @@
+//go:generate ../../../tools/readme_config_includer/generator
+package ecs
+
+import (
+	_ "embed"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/influxdata/telegraf"
+	"github.com/influxdata/telegraf/config"
+	"github.com/influxdata/telegraf/filter"
+	"github.com/influxdata/telegraf/plugins/inputs"
+)
+
+//go:embed sample.conf
+var sampleConfig string
+
+const (
+	v2Endpoint = "http://169.254.170.2"
+)
+
+type Ecs struct {
+	EndpointURL string          `toml:"endpoint_url"`
+	Timeout     config.Duration `toml:"timeout"`
+
+	ContainerNameInclude []string `toml:"container_name_include"`
+	ContainerNameExclude []string `toml:"container_name_exclude"`
+
+	ContainerStatusInclude []string `toml:"container_status_include"`
+	ContainerStatusExclude []string `toml:"container_status_exclude"`
+
+	LabelInclude []string `toml:"ecs_label_include"`
+	LabelExclude []string `toml:"ecs_label_exclude"`
+
+	newClient func(timeout time.Duration, endpoint string, version int) (*ecsClient, error)
+
+	client              client
+	filtersCreated      bool
+	labelFilter         filter.Filter
+	containerNameFilter filter.Filter
+	statusFilter        filter.Filter
+	metadataVersion     int
+}
+
+func (*Ecs) SampleConfig() string {
+	return sampleConfig
+}
+
+func (ecs *Ecs) Gather(acc telegraf.Accumulator) error {
+	err := initSetup(ecs)
+	if err != nil {
+		return err
+	}
+
+	task, stats, err := pollSync(ecs.client)
+	if err != nil {
+		return err
+	}
+
+	mergeTaskStats(task, stats)
+
+	taskTags := map[string]string{
+		"cluster":  task.Cluster,
+		"task_arn": task.TaskARN,
+		"family":   task.Family,
+		"revision": task.Revision,
+	}
+
+	// accumulate metrics
+	accTask(task, taskTags, acc)
+	ecs.accContainers(task, taskTags, acc)
+
+	return nil
+}
+
+func initSetup(ecs *Ecs) error {
+	if ecs.client == nil {
+		resolveEndpoint(ecs)
+
+		c, err := ecs.newClient(time.Duration(ecs.Timeout), ecs.EndpointURL, ecs.metadataVersion)
+		if err != nil {
+			return err
+		}
+		ecs.client = c
+	}
+
+	// Create filters
+	if !ecs.filtersCreated {
+		err := ecs.createContainerNameFilters()
+		if err != nil {
+			return err
+		}
+		err = ecs.createContainerStatusFilters()
+		if err != nil {
+			return err
+		}
+		err = ecs.createLabelFilters()
+		if err != nil {
+			return err
+		}
+		ecs.filtersCreated = true
+	}
+
+	return nil
+}
+
+func resolveEndpoint(ecs *Ecs) {
+	if ecs.EndpointURL != "" {
+		// Use metadata v2 API since endpoint is set explicitly.
+		ecs.metadataVersion = 2
+		return
+	}
+
+	// Auto-detect metadata endpoint version.
+
+	// Use metadata v4 if available.
+	// https://docs.aws.amazon.com/AmazonECS/latest/developerguide/task-metadata-endpoint-v4.html
+	v4Endpoint := os.Getenv("ECS_CONTAINER_METADATA_URI_V4")
+	if v4Endpoint != "" {
+		ecs.EndpointURL = v4Endpoint
+		ecs.metadataVersion = 4
+		return
+	}
+
+	// Use metadata v3 if available.
+	// https://docs.aws.amazon.com/AmazonECS/latest/developerguide/task-metadata-endpoint-v3.html
+	v3Endpoint := os.Getenv("ECS_CONTAINER_METADATA_URI")
+	if v3Endpoint != "" {
+		ecs.EndpointURL = v3Endpoint
+		ecs.metadataVersion = 3
+		return
+	}
+
+	// Use v2 endpoint if nothing else is available.
+	ecs.EndpointURL = v2Endpoint
+	ecs.metadataVersion = 2
+}
+
+func accTask(task *ecsTask, tags map[string]string, acc telegraf.Accumulator) {
+	taskFields := map[string]interface{}{
+		"desired_status": task.DesiredStatus,
+		"known_status":   task.KnownStatus,
+		"limit_cpu":      task.Limits["CPU"],
+		"limit_mem":      task.Limits["Memory"],
+	}
+
+	acc.AddFields("ecs_task", taskFields, tags)
+}
+
+func (ecs *Ecs) accContainers(task *ecsTask, taskTags map[string]string, acc telegraf.Accumulator) {
+	for i := range task.Containers {
+		c := &task.Containers[i]
+		if !ecs.containerNameFilter.Match(c.Name) {
+			continue
+		}
+
+		if !ecs.statusFilter.Match(strings.ToUpper(c.KnownStatus)) {
+			continue
+		}
+
+		// add matching ECS container Labels
+		containerTags := map[string]string{
+			"id":   c.ID,
+			"name": c.Name,
+		}
+		for k, v := range c.Labels {
+			if ecs.labelFilter.Match(k) {
+				containerTags[k] = v
+			}
+		}
+		tags := mergeTags(taskTags, containerTags)
+
+		parseContainerStats(c, acc, tags)
+	}
+}
+
+// returns a new map with the same content values as the input map
+func copyTags(in map[string]string) map[string]string {
+	out := make(map[string]string)
+	for k, v := range in {
+		out[k] = v
+	}
+	return out
+}
+
+// returns a new map with the merged content values of the two input maps
+func mergeTags(a, b map[string]string) map[string]string {
+	c := copyTags(a)
+	for k, v := range b {
+		c[k] = v
+	}
+	return c
+}
+
+func (ecs *Ecs) createContainerNameFilters() error {
+	containerNameFilter, err := filter.NewIncludeExcludeFilter(ecs.ContainerNameInclude, ecs.ContainerNameExclude)
+	if err != nil {
+		return err
+	}
+	ecs.containerNameFilter = containerNameFilter
+	return nil
+}
+
+func (ecs *Ecs) createLabelFilters() error {
+	labelFilter, err := filter.NewIncludeExcludeFilter(ecs.LabelInclude, ecs.LabelExclude)
+	if err != nil {
+		return err
+	}
+	ecs.labelFilter = labelFilter
+	return nil
+}
+
+func (ecs *Ecs) createContainerStatusFilters() error {
+	if len(ecs.ContainerStatusInclude) == 0 && len(ecs.ContainerStatusExclude) == 0 {
+		ecs.ContainerStatusInclude = []string{"RUNNING"}
+	}
+
+	// ECS uses uppercase status names, normalizing for comparison.
+	for i, include := range ecs.ContainerStatusInclude {
+		ecs.ContainerStatusInclude[i] = strings.ToUpper(include)
+	}
+	for i, exclude := range ecs.ContainerStatusExclude {
+		ecs.ContainerStatusExclude[i] = strings.ToUpper(exclude)
+	}
+
+	statusFilter, err := filter.NewIncludeExcludeFilter(ecs.ContainerStatusInclude, ecs.ContainerStatusExclude)
+	if err != nil {
+		return err
+	}
+	ecs.statusFilter = statusFilter
+	return nil
+}
+
+func init() {
+	inputs.Add("ecs", func() telegraf.Input {
+		return &Ecs{
+			EndpointURL:    "",
+			Timeout:        config.Duration(5 * time.Second),
+			newClient:      newClient,
+			filtersCreated: false,
+		}
+	})
+}