SimpleLambda.py

import boto3
import json
import os

from opentelemetry import trace
from opentelemetry.sdk.resources import SERVICE_NAME, Resource, DEPLOYMENT_ENVIRONMENT
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
from opentelemetry.sdk.trace.export import BatchSpanProcessor

# Set up the OpenTelemetry tracer provider with a resource name.
resource = Resource(attributes={
 SERVICE_NAME: "python-lambda-example",
 DEPLOYMENT_ENVIRONMENT: "dev",
})
provider = TracerProvider(resource=resource)

span_processor = BatchSpanProcessor(OTLPSpanExporter(endpoint="http://localhost:4318/v1/traces"))
provider.add_span_processor(span_processor)

trace.set_tracer_provider(provider)

tracer = trace.get_tracer("python-lambda.tracer")

def echo(payload):
 return payload

def list_buckets(payload):
 client = boto3.client("s3")
 buckets = client.list_buckets()
 bucket_names = list(map(lambda b: b['Name'], buckets['Buckets']))

 return {
 "statusCode": 200,
 "headers": {
 "Content-Type": "application/json"
 },
 "body": json.dumps({
 "Buckets": bucket_names,
 "Region ": os.environ['AWS_REGION']
 })
 }

operations = {
 'echo': echo,
 'list_buckets': list_buckets,
}

def lambda_handler(event, context):
 '''Provide an event that contains the following keys:
 - operation: one of the operations in the operations dict below
 - payload: a JSON object containing parameters to pass to the
 operation being performed
 '''

 with tracer.start_as_current_span(f"op-{event['operation']}") as span:
 operation = event['operation']
 payload = event['payload']

 if operation in operations:
 return operations[operation](payload)
 else:
 raise ValueError(f'Unrecognized operation "{operation}"')