feat: improve lambda execution detection

Signed-off-by: Kavindu Dodanduwa <[email protected]>

Author: Kavindu-Dodan

receiver/awslambdareceiver/receiver.go

@@ -10,6 +10,7 @@ import (
 	"errors"
 	"fmt"
 	"os"
+	"strings"
 
 	"github.com/aws/aws-lambda-go/events"
 	"github.com/aws/aws-lambda-go/lambda"
@@ -95,8 +96,8 @@ func newMetricsReceiver(cfg *Config, set receiver.Settings, next consumer.Metric
 // when lambda is triggered
 func (a *awsLambdaReceiver) Start(ctx context.Context, host component.Host) error {
 	// Verify we're running in a Lambda environment
-	if os.Getenv("_LAMBDA_SERVER_PORT") == "" || os.Getenv("AWS_LAMBDA_RUNTIME_API") == "" {
-		return errors.New("receiver must be used in an AWS Lambda environment: required environment variables _LAMBDA_SERVER_PORT and AWS_LAMBDA_RUNTIME_API are not set")
+	if os.Getenv("AWS_EXECUTION_ENV") == "" || !strings.HasPrefix(os.Getenv("AWS_EXECUTION_ENV"), "AWS_Lambda_") {
+		return errors.New("awslambdareceiver must be used in an AWS Lambda environment: missing environment variable AWS_EXECUTION_ENV")
 	}
 
 	handler, err := a.newHandler(ctx, host, a.s3Provider)

receiver/awslambdareceiver/receiver_test.go

@@ -25,8 +25,7 @@ import (
 
 func TestCreateLogs(t *testing.T) {
 	// Set Lambda environment variables required by Start()
-	t.Setenv("_LAMBDA_SERVER_PORT", "9001")
-	t.Setenv("AWS_LAMBDA_RUNTIME_API", "localhost:9001")
+	t.Setenv("AWS_EXECUTION_ENV", "AWS_Lambda_python3.12")
 
 	// Create receiver using factory with S3 encoding config.
 	// Note: The S3Encoding value must match the component ID used when registering the extension.
@@ -95,8 +94,7 @@ func TestCreateLogs(t *testing.T) {
 
 func TestCreateMetrics(t *testing.T) {
 	// Set Lambda environment variables required by Start()
-	t.Setenv("_LAMBDA_SERVER_PORT", "9001")
-	t.Setenv("AWS_LAMBDA_RUNTIME_API", "localhost:9001")
+	t.Setenv("AWS_EXECUTION_ENV", "AWS_Lambda_python3.12")
 
 	// Create receiver using factory with a dummy encoding extension.
 	factory := NewFactory()
@@ -173,8 +171,7 @@ func TestCreateMetricsRequiresS3Encoding(t *testing.T) {
 
 func TestStartRequiresLambdaEnvironment(t *testing.T) {
 	// Ensure Lambda environment variables are not set
-	t.Setenv("_LAMBDA_SERVER_PORT", "")
-	t.Setenv("AWS_LAMBDA_RUNTIME_API", "")
+	t.Setenv("AWS_EXECUTION_ENV", "")
 
 	factory := NewFactory()
 	cfg := factory.CreateDefaultConfig().(*Config)