Add container id support to Resource

CHANGELOG.md

@@ -19,6 +19,7 @@ This project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.htm
   If the provided environment variables are invalid (negative), the default values would be used.
 - Rename the `gc` runtime name to `go` (#2560)
 - Log the Exporters configuration in the TracerProviders message. (#2578)
+- Add container id support to Resource. (#2418)
 
 ### Changed
 

sdk/resource/config.go

@@ -171,3 +171,16 @@ func WithProcessRuntimeVersion() Option {
 func WithProcessRuntimeDescription() Option {
 	return WithDetectors(processRuntimeDescriptionDetector{})
 }
+
+// WithContainer adds all the Container attributes to the configured Resource.
+// See individual WithContainer* functions to configure specific attributes.
+func WithContainer() Option {
+	return WithDetectors(
+		cgroupContainerIDDetector{},
+	)
+}
+
+// WithContainerID adds an attribute with the id of the container to the configured Resource.
+func WithContainerID() Option {
+	return WithDetectors(cgroupContainerIDDetector{})
+}

sdk/resource/container.go

@@ -0,0 +1,130 @@
+// Copyright The OpenTelemetry 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 resource // import "go.opentelemetry.io/otel/sdk/resource"
+
+import (
+	"bufio"
+	"context"
+	"errors"
+	"io"
+	"os"
+	"strings"
+
+	semconv "go.opentelemetry.io/otel/semconv/v1.7.0"
+)
+
+type containerIDProvider func() (string, error)
+
+var (
+	containerID containerIDProvider = getContainerIDFromCGroup
+)
+
+type cgroupContainerIDDetector struct{}
+
+const cgroupPath = "/proc/self/cgroup"
+
+// Detect returns a *Resource that describes the id of the container.
+// If no container id found, an empty resource will be returned.
+func (cgroupContainerIDDetector) Detect(ctx context.Context) (*Resource, error) {
+	containerID, err := containerID()
+	if err != nil {
+		return nil, err
+	}
+
+	if containerID == "" {
+		return Empty(), nil
+	}
+	return NewWithAttributes(semconv.SchemaURL, semconv.ContainerIDKey.String(containerID)), nil
+}
+
+// getContainerIDFromCGroup returns the id of the container from the cgroup file.
+// If no container id found, an empty string will be returned.
+func getContainerIDFromCGroup() (string, error) {
+	if _, err := os.Stat(cgroupPath); errors.Is(err, os.ErrNotExist) {
+		// File does not exists, skip
+		return "", nil
+	}
+
+	file, err := os.Open(cgroupPath)
+	if err != nil {
+		return "", err
+	}
+	defer file.Close()
+
+	containerID := getContainerIDFromReader(file)
+	if containerID == "" {
+		// Container ID not found
+		return "", nil
+	}
+	return containerID, nil
+}
+
+// getContainerIDFromReader returns the id of the container from reader.
+func getContainerIDFromReader(reader io.Reader) string {
+	scanner := bufio.NewScanner(reader)
+	for scanner.Scan() {
+		line := scanner.Text()
+
+		if id := getContainerIDFromLine(line); id != "" {
+			return id
+		}
+	}
+	return ""
+}
+
+// getContainerIDFromLine returns the id of the container from one string line.
+func getContainerIDFromLine(line string) string {
+	line = strings.TrimSpace(line)
+
+	lastSlashIndexOfLine := strings.LastIndexByte(line, '/')
+	if lastSlashIndexOfLine == -1 {
+		return ""
+	}
+
+	lastSection := line[lastSlashIndexOfLine+1:]
+	dashIndex := strings.IndexByte(lastSection, '-')
+	lastDotIndex := strings.LastIndexByte(lastSection, '.')
+
+	startIndex := 0
+	if dashIndex != -1 {
+		startIndex = dashIndex + 1
+	}
+
+	endIndex := len(lastSection)
+	if lastDotIndex != -1 {
+		endIndex = lastDotIndex
+	}
+
+	containerID := lastSection[startIndex:endIndex]
+	if !isHex(containerID) {
+		return ""
+	}
+	return containerID
+}
+
+// isHex returns true when input is a hex string.
+func isHex(h string) bool {
+	for _, r := range h {
+		switch {
+		case 'a' <= r && r <= 'f':
+			continue
+		case '0' <= r && r <= '9':
+			continue
+		default:
+			return false
+		}
+	}
+	return true
+}

sdk/resource/container_test.go

@@ -0,0 +1,114 @@
+// Copyright The OpenTelemetry 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 resource
+
+import (
+	"io"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func setDefaultContainerProviders() {
+	setContainerProviders(
+		getContainerIDFromCGroup,
+	)
+}
+
+func setContainerProviders(
+	idProvider containerIDProvider,
+) {
+	containerID = idProvider
+}
+
+func TestGetContainerIDFromLine(t *testing.T) {
+	testCases := []struct {
+		name                string
+		line                string
+		expectedContainerID string
+	}{
+		{
+			name:                "with suffix",
+			line:                "13:name=systemd:/podruntime/docker/kubepods/ac679f8a8319c8cf7d38e1adf263bc08d23.aaaa",
+			expectedContainerID: "ac679f8a8319c8cf7d38e1adf263bc08d23",
+		},
+		{
+			name:                "with prefix and suffix",
+			line:                "13:name=systemd:/podruntime/docker/kubepods/crio-dc679f8a8319c8cf7d38e1adf263bc08d23.stuff",
+			expectedContainerID: "dc679f8a8319c8cf7d38e1adf263bc08d23",
+		},
+		{
+			name:                "no prefix and suffix",
+			line:                "13:name=systemd:/pod/d86d75589bf6cc254f3e2cc29debdf85dde404998aa128997a819ff991827356",
+			expectedContainerID: "d86d75589bf6cc254f3e2cc29debdf85dde404998aa128997a819ff991827356",
+		},
+		{
+			name:                "with space",
+			line:                " 13:name=systemd:/pod/d86d75589bf6cc254f3e2cc29debdf85dde404998aa128997a819ff991827356 ",
+			expectedContainerID: "d86d75589bf6cc254f3e2cc29debdf85dde404998aa128997a819ff991827356",
+		},
+		{
+			name: "invalid hex string",
+			line: "13:name=systemd:/podruntime/docker/kubepods/ac679f8a8319c8cf7d38e1adf263bc08d23zzzz",
+		},
+		{
+			name: "no container id - 1",
+			line: "pids: /",
+		},
+		{
+			name: "no container id - 2",
+			line: "pids: ",
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			containerID := getContainerIDFromLine(tc.line)
+			assert.Equal(t, tc.expectedContainerID, containerID)
+		})
+	}
+}
+
+func TestGetContainerIDFromReader(t *testing.T) {
+	testCases := []struct {
+		name                string
+		reader              io.Reader
+		expectedContainerID string
+	}{
+		{
+			name: "multiple lines",
+			reader: strings.NewReader(`//
+1:name=systemd:/podruntime/docker/kubepods/docker-dc579f8a8319c8cf7d38e1adf263bc08d23
+1:name=systemd:/podruntime/docker/kubepods/docker-dc579f8a8319c8cf7d38e1adf263bc08d24
+`),
+			expectedContainerID: "dc579f8a8319c8cf7d38e1adf263bc08d23",
+		},
+		{
+			name: "no container id",
+			reader: strings.NewReader(`//
+1:name=systemd:/podruntime/docker
+`),
+			expectedContainerID: "",
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			containerID := getContainerIDFromReader(tc.reader)
+			assert.Equal(t, tc.expectedContainerID, containerID)
+		})
+	}
+}

sdk/resource/export_test.go

@@ -23,6 +23,8 @@ var (
 	SetUserProviders                = setUserProviders
 	SetDefaultOSDescriptionProvider = setDefaultOSDescriptionProvider
 	SetOSDescriptionProvider        = setOSDescriptionProvider
+	SetDefaultContainerProviders    = setDefaultContainerProviders
+	SetContainerProviders           = setContainerProviders
 )
 
 var (

sdk/resource/process_test.go

@@ -102,13 +102,14 @@ func restoreAttributesProviders() {
 	resource.SetDefaultRuntimeProviders()
 	resource.SetDefaultUserProviders()
 	resource.SetDefaultOSDescriptionProvider()
+	resource.SetDefaultContainerProviders()
 }
 
 func TestWithProcessFuncsErrors(t *testing.T) {
 	mockProcessAttributesProvidersWithErrors()
 
-	t.Run("WithPID", testWithProcessExecutablePathError)
-	t.Run("WithExecutableName", testWithProcessOwnerError)
+	t.Run("WithExecutablePath", testWithProcessExecutablePathError)
+	t.Run("WithOwner", testWithProcessOwnerError)
 
 	restoreAttributesProviders()
 }

sdk/resource/resource_test.go

@@ -649,3 +649,75 @@ func hostname() string {
 	}
 	return hn
 }
+
+func TestWithContainerID(t *testing.T) {
+	t.Cleanup(restoreAttributesProviders)
+
+	fakeContainerID := "fake-container-id"
+
+	testCases := []struct {
+		name                string
+		containerIDProvider func() (string, error)
+
+		expectedResource map[string]string
+		expectedErr      bool
+	}{
+		{
+			name: "get container id",
+			containerIDProvider: func() (string, error) {
+				return fakeContainerID, nil
+			},
+			expectedResource: map[string]string{
+				string(semconv.ContainerIDKey): fakeContainerID,
+			},
+		},
+		{
+			name: "no container id found",
+			containerIDProvider: func() (string, error) {
+				return "", nil
+			},
+			expectedResource: map[string]string{},
+		},
+		{
+			name: "error",
+			containerIDProvider: func() (string, error) {
+				return "", fmt.Errorf("unable to get container id")
+			},
+			expectedResource: map[string]string{},
+			expectedErr:      true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			resource.SetContainerProviders(tc.containerIDProvider)
+
+			res, err := resource.New(context.Background(),
+				resource.WithContainerID(),
+			)
+
+			if tc.expectedErr {
+				assert.Error(t, err)
+			}
+			assert.Equal(t, tc.expectedResource, toMap(res))
+		})
+	}
+}
+
+func TestWithContainer(t *testing.T) {
+	t.Cleanup(restoreAttributesProviders)
+
+	fakeContainerID := "fake-container-id"
+	resource.SetContainerProviders(func() (string, error) {
+		return fakeContainerID, nil
+	})
+
+	res, err := resource.New(context.Background(),
+		resource.WithContainer(),
+	)
+
+	assert.NoError(t, err)
+	assert.Equal(t, map[string]string{
+		string(semconv.ContainerIDKey): fakeContainerID,
+	}, toMap(res))
+}