[processor/resourcedetection] Add GKE detector

processor/resourcedetectionprocessor/README.md

@@ -29,6 +29,12 @@ to read resource information from the [GCE metadata server](https://cloud.google
     * host.image.id
     * host.type
 
+* GKE: Google Kubernetes Engine
+
+    * cloud.provider ("gcp")
+    * cloud.infrastructure_service ("gcp_gke")
+    * k8s.cluster.name (name of the GKE cluster)
+
 * AWS EC2: Uses [AWS SDK for Go](https://docs.aws.amazon.com/sdk-for-go/api/aws/ec2metadata/) to read resource information from the [EC2 instance metadata API](https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/ec2-instance-metadata.html) to retrieve the following resource attributes:
 
     * cloud.provider ("aws")
@@ -93,7 +99,7 @@ ec2:
 ## Configuration
 
 ```yaml
-# a list of resource detectors to run, valid options are: "env", "system", "gce", "ec2", "ecs", "elastic_beanstalk", "azure"
+# a list of resource detectors to run, valid options are: "env", "system", "gce", "gke", "ec2", "ecs", "elastic_beanstalk", "azure"
 detectors: [ <string> ]
 # determines if existing resource attributes should be overridden or preserved, defaults to true
 override: <bool>

processor/resourcedetectionprocessor/factory.go

@@ -32,6 +32,7 @@ import (
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/azure"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/env"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/gcp/gce"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/gcp/gke"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/system"
 )
 
@@ -54,13 +55,14 @@ type factory struct {
 // NewFactory creates a new factory for ResourceDetection processor.
 func NewFactory() component.ProcessorFactory {
 	resourceProviderFactory := internal.NewProviderFactory(map[internal.DetectorType]internal.DetectorFactory{
-		env.TypeStr:              env.NewDetector,
-		system.TypeStr:           system.NewDetector,
-		gce.TypeStr:              gce.NewDetector,
+		azure.TypeStr:            azure.NewDetector,
 		ec2.TypeStr:              ec2.NewDetector,
 		ecs.TypeStr:              ecs.NewDetector,
 		elasticbeanstalk.TypeStr: elasticbeanstalk.NewDetector,
-		azure.TypeStr:            azure.NewDetector,
+		env.TypeStr:              env.NewDetector,
+		gce.TypeStr:              gce.NewDetector,
+		gke.TypeStr:              gke.NewDetector,
+		system.TypeStr:           system.NewDetector,
 	})
 
 	f := &factory{

processor/resourcedetectionprocessor/internal/gcp/gce/gce.go

@@ -25,6 +25,7 @@ import (
 	"go.opentelemetry.io/collector/translator/conventions"
 
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/gcp"
 )
 
 const (
@@ -34,11 +35,11 @@ const (
 var _ internal.Detector = (*Detector)(nil)
 
 type Detector struct {
-	metadata gceMetadata
+	metadata gcp.Metadata
 }
 
 func NewDetector(component.ProcessorCreateParams, internal.DetectorConfig) (internal.Detector, error) {
-	return &Detector{metadata: &gceMetadataImpl{}}, nil
+	return &Detector{metadata: &gcp.MetadataImpl{}}, nil
 }
 
 func (d *Detector) Detect(context.Context) (pdata.Resource, error) {

processor/resourcedetectionprocessor/internal/gcp/gce/gce_test.go

@@ -20,68 +20,30 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
 	"go.opentelemetry.io/collector/component"
 	"go.opentelemetry.io/collector/translator/conventions"
 	"go.uber.org/zap"
 
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/gcp"
 )
 
-type mockMetadata struct {
-	mock.Mock
-}
-
-func (m *mockMetadata) OnGCE() bool {
-	return m.MethodCalled("OnGCE").Bool(0)
-}
-
-func (m *mockMetadata) ProjectID() (string, error) {
-	args := m.MethodCalled("ProjectID")
-	return args.String(0), args.Error(1)
-}
-
-func (m *mockMetadata) Zone() (string, error) {
-	args := m.MethodCalled("Zone")
-	return args.String(0), args.Error(1)
-}
-
-func (m *mockMetadata) Hostname() (string, error) {
-	args := m.MethodCalled("Hostname")
-	return args.String(0), args.Error(1)
-}
-
-func (m *mockMetadata) InstanceID() (string, error) {
-	args := m.MethodCalled("InstanceID")
-	return args.String(0), args.Error(1)
-}
-
-func (m *mockMetadata) InstanceName() (string, error) {
-	args := m.MethodCalled("InstanceName")
-	return args.String(0), args.Error(1)
-}
-
-func (m *mockMetadata) Get(suffix string) (string, error) {
-	args := m.MethodCalled("Get")
-	return args.String(0), args.Error(1)
-}
-
 func TestNewDetector(t *testing.T) {
 	d, err := NewDetector(component.ProcessorCreateParams{Logger: zap.NewNop()}, nil)
 	assert.NotNil(t, d)
 	assert.NoError(t, err)
 }
 
 func TestDetectTrue(t *testing.T) {
-	md := &mockMetadata{}
+	md := &gcp.MockMetadata{}
 	md.On("OnGCE").Return(true)
 	md.On("ProjectID").Return("1", nil)
 	md.On("Zone").Return("zone", nil)
 	md.On("Hostname").Return("hostname", nil)
 	md.On("InstanceID").Return("2", nil)
 	md.On("InstanceName").Return("name", nil)
-	md.On("Get").Return("machine-type", nil)
+	md.On("Get", "").Return("machine-type", nil)
 
 	detector := &Detector{metadata: md}
 	res, err := detector.Detect(context.Background())
@@ -105,7 +67,7 @@ func TestDetectTrue(t *testing.T) {
 }
 
 func TestDetectFalse(t *testing.T) {
-	md := &mockMetadata{}
+	md := &gcp.MockMetadata{}
 	md.On("OnGCE").Return(false)
 
 	detector := &Detector{metadata: md}
@@ -116,7 +78,7 @@ func TestDetectFalse(t *testing.T) {
 }
 
 func TestDetectError(t *testing.T) {
-	md := &mockMetadata{}
+	md := &gcp.MockMetadata{}
 	md.On("OnGCE").Return(true)
 	md.On("ProjectID").Return("", errors.New("err1"))
 	md.On("Zone").Return("", errors.New("err2"))

processor/resourcedetectionprocessor/internal/gcp/gke/gke.go

@@ -0,0 +1,78 @@
+// 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 gke
+
+import (
+	"context"
+	"os"
+
+	"go.opentelemetry.io/collector/component"
+	"go.opentelemetry.io/collector/consumer/pdata"
+	"go.opentelemetry.io/collector/translator/conventions"
+	"go.uber.org/zap"
+
+	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/gcp"
+)
+
+const (
+	// TypeStr is type of detector.
+	TypeStr = "gke"
+
+	// GCE metadata attribute containing the GKE cluster name.
+	clusterNameAttribute = "cluster-name"
+
+	// Environment variable that is set when running on Kubernetes.
+	kubernetesServiceHostEnvVar = "KUBERNETES_SERVICE_HOST"
+)
+
+var _ internal.Detector = (*Detector)(nil)
+
+type Detector struct {
+	log      *zap.Logger
+	metadata gcp.Metadata
+}
+
+func NewDetector(params component.ProcessorCreateParams, _ internal.DetectorConfig) (internal.Detector, error) {
+	return &Detector{log: params.Logger, metadata: &gcp.MetadataImpl{}}, nil
+}
+
+// Detect detects associated resources when running in GKE environment.
+func (gke *Detector) Detect(ctx context.Context) (pdata.Resource, error) {
+	res := pdata.NewResource()
+
+	// Check if on GCP.
+	if !gke.metadata.OnGCE() {
+		return res, nil
+	}
+
+	attr := res.Attributes()
+	attr.InsertString(conventions.AttributeCloudProvider, conventions.AttributeCloudProviderGCP)
+
+	// Check if running on k8s.
+	if os.Getenv(kubernetesServiceHostEnvVar) == "" {
+		return res, nil
+	}
+
+	attr.InsertString(conventions.AttributeCloudInfrastructureService, conventions.AttributeCloudProviderGCPGKE)
+
+	if clusterName, err := gke.metadata.InstanceAttributeValue(clusterNameAttribute); err != nil {
+		gke.log.Warn("Unable to determine GKE cluster name", zap.Error(err))
+	} else if clusterName != "" {
+		attr.InsertString(conventions.AttributeK8sCluster, clusterName)
+	}
+
+	return res, nil
+}

processor/resourcedetectionprocessor/internal/gcp/gke/gke_test.go

@@ -0,0 +1,119 @@
+// 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 gke
+
+import (
+	"context"
+	"errors"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"go.opentelemetry.io/collector/component"
+	"go.uber.org/zap"
+
+	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourcedetectionprocessor/internal/gcp"
+)
+
+func TestNotGCE(t *testing.T) {
+	metadata := &gcp.MockMetadata{}
+	detector := &Detector{
+		log:      zap.NewNop(),
+		metadata: metadata,
+	}
+
+	metadata.On("OnGCE").Return(false)
+	res, err := detector.Detect(context.Background())
+	require.NoError(t, err)
+	assert.Equal(t, 0, res.Attributes().Len())
+
+	metadata.AssertExpectations(t)
+}
+
+func TestDetectWithoutCluster(t *testing.T) {
+	metadata := &gcp.MockMetadata{}
+	detector := &Detector{
+		log:      zap.NewNop(),
+		metadata: metadata,
+	}
+
+	metadata.On("OnGCE").Return(true)
+	metadata.On("InstanceAttributeValue", "cluster-name").Return("", errors.New("no cluster"))
+
+	require.NoError(t, os.Setenv("KUBERNETES_SERVICE_HOST", "localhost"))
+
+	res, err := detector.Detect(context.Background())
+	require.NoError(t, err)
+
+	assert.Equal(t, map[string]interface{}{
+		"cloud.provider":               "gcp",
+		"cloud.infrastructure_service": "gcp_gke",
+	}, internal.AttributesToMap(res.Attributes()))
+
+	metadata.AssertExpectations(t)
+}
+
+func TestDetectWithoutK8s(t *testing.T) {
+	metadata := &gcp.MockMetadata{}
+	detector := &Detector{
+		log:      zap.NewNop(),
+		metadata: metadata,
+	}
+
+	metadata.On("OnGCE").Return(true)
+
+	require.NoError(t, os.Unsetenv("KUBERNETES_SERVICE_HOST"))
+
+	res, err := detector.Detect(context.Background())
+	require.NoError(t, err)
+
+	assert.Equal(t, map[string]interface{}{
+		"cloud.provider": "gcp",
+	}, internal.AttributesToMap(res.Attributes()))
+
+	metadata.AssertExpectations(t)
+}
+
+func TestDetector_Detect(t *testing.T) {
+	metadata := &gcp.MockMetadata{}
+	detector := &Detector{
+		log:      zap.NewNop(),
+		metadata: metadata,
+	}
+
+	metadata.On("OnGCE").Return(true)
+	metadata.On("InstanceAttributeValue", "cluster-name").Return("cluster-a", nil)
+
+	require.NoError(t, os.Setenv("KUBERNETES_SERVICE_HOST", "localhost"))
+
+	res, err := detector.Detect(context.Background())
+	require.NoError(t, err)
+
+	assert.Equal(t, map[string]interface{}{
+		"cloud.provider":               "gcp",
+		"cloud.infrastructure_service": "gcp_gke",
+		"k8s.cluster.name":             "cluster-a",
+	}, internal.AttributesToMap(res.Attributes()))
+
+	metadata.AssertExpectations(t)
+}
+
+func TestNewDetector(t *testing.T) {
+	detector, err := NewDetector(component.ProcessorCreateParams{Logger: zap.NewNop()}, nil)
+	assert.NoError(t, err)
+	assert.NotNil(t, detector)
+}