added lambda code

Author: dmitchsplunk

workshop/lambda/README.md

@@ -0,0 +1,23 @@
+// Lambda Consumer App for Kinesis Stream
+
+// Lambda Consumer
+const consumer = async( event ) => {
+  try{
+	for( const record of event.Records ) {
+      const payload = record.kinesis;
+      const message = Buffer.from(payload.data, 'base64').toString();
+
+      console.log(
+        `Kinesis Message:
+        partition key: ${payload.partitionKey}
+        sequence number: ${payload.sequenceNumber}
+        kinesis schema version: ${payload.kinesisSchemaVersion}
+        data: ${message}`
+      );
+    }
+  } catch ( error ) {
+    console.log(error);
+  }
+};
+
+export { consumer };

workshop/lambda/auto/handler/consumer.mjs

@@ -0,0 +1,47 @@
+// Lambda Producer App for Kinesis Stream
+import { KinesisClient, PutRecordCommand } from "@aws-sdk/client-kinesis";
+
+
+// Lambda Producer
+const producer = async( event ) => {
+  const kinesis = new KinesisClient({});
+
+  let statusCode = 200;
+  let message;
+
+  if (!event.body) {
+    return {
+      statusCode: 400,
+      body: JSON.stringify({
+        message: "No message body found",
+      }),
+    };
+  }
+
+  const streamName = process.env.KINESIS_STREAM;
+
+  try {
+    await kinesis.send(
+      new PutRecordCommand({
+        StreamName: streamName,
+        PartitionKey: "1234",
+        Data: event.body,
+      }),
+
+      message = `Message placed in the Event Stream: ${streamName}`
+	)
+  } catch ( error ) {
+    console.log(error);
+    message = error;
+    statusCode = 500;
+  }
+
+  return {
+    statusCode,
+    body: JSON.stringify({
+      message,
+    }),
+  };
+};
+
+export { producer };

workshop/lambda/auto/handler/producer.mjs

@@ -0,0 +1,239 @@
+provider "aws" {
+  region = "us-east-1"
+
+  default_tags {
+    tags = {
+      o11y-workshop = "lambda-tracing"
+    }
+  }
+}
+
+
+# Get IAM Role
+data "aws_caller_identity" "current" {}
+resource "aws_iam_role" "lambda_kinesis" {
+  name = "lambda_kinesis"
+  assume_role_policy = jsonencode({
+    Version = "2012-10-17"
+    Statement = [
+      {
+        Action = "sts:AssumeRole"
+        Effect = "Allow"
+        Principal = {
+          Service = "lambda.amazonaws.com"
+        }
+      }
+    ]
+  })
+}
+resource "aws_iam_role_policy_attachment" "lambda_kinesis_execution" {
+  role       = aws_iam_role.lambda_kinesis.name
+  policy_arn = "arn:aws:iam::aws:policy/AmazonKinesisFullAccess"
+}
+
+# Create S3 Bucket, Ownership, ACL
+resource "aws_s3_bucket" "lambda_bucket" {
+  bucket = "${var.prefix}-lambda-code"
+}
+
+resource "aws_s3_bucket_ownership_controls" "lambda_bucket" {
+  bucket = aws_s3_bucket.lambda_bucket.id
+  rule {
+    object_ownership = "BucketOwnerPreferred"
+  }
+}
+
+resource "aws_s3_bucket_acl" "lambda_bucket" {
+  depends_on = [aws_s3_bucket_ownership_controls.lambda_bucket]
+
+  bucket = aws_s3_bucket.lambda_bucket.id
+  acl    = "private"
+}
+
+
+# Package Producer and Consumer Apps, Upload to S3 Bucket
+data "archive_file" "producer_app" {
+  type = "zip"
+
+  source_file  = "${path.module}/handler/producer.mjs"
+  output_path = "${path.module}/lambda_producer.zip"
+}
+
+resource "aws_s3_object" "producer_app" {
+  bucket = aws_s3_bucket.lambda_bucket.id
+
+  key    = "lambda_producer.zip"
+  source = data.archive_file.producer_app.output_path
+
+  etag = filemd5(data.archive_file.producer_app.output_path)
+}
+
+data "archive_file" "consumer_app" {
+  type = "zip"
+
+  source_file  = "${path.module}/handler/consumer.mjs"
+  output_path = "${path.module}/lambda_consumer.zip"
+}
+
+resource "aws_s3_object" "consumer_app" {
+  bucket = aws_s3_bucket.lambda_bucket.id
+
+  key    = "lambda_consumer.zip"
+  source = data.archive_file.consumer_app.output_path
+
+  etag = filemd5(data.archive_file.consumer_app.output_path)
+}
+
+
+# Create Lambda Functions
+resource "aws_lambda_function" "lambda_producer" {
+  function_name = "${var.prefix}-producer"
+
+  s3_bucket = aws_s3_bucket.lambda_bucket.id
+  s3_key    = aws_s3_object.producer_app.key
+
+  runtime = "nodejs20.x"
+  handler = "producer.producer"
+
+  source_code_hash = data.archive_file.producer_app.output_base64sha256
+
+  role = aws_iam_role.lambda_kinesis.arn
+
+  environment {
+    variables = {
+      SPLUNK_ACCESS_TOKEN = var.o11y_access_token
+      SPLUNK_REALM = var.o11y_realm
+      OTEL_SERVICE_NAME = "${var.prefix}-producer-lambda"
+      OTEL_RESOURCE_ATTRIBUTES = "deployment.environment=${var.prefix}-lambda-shop"
+      AWS_LAMBDA_EXEC_WRAPPER = "/opt/nodejs-otel-handler"
+      KINESIS_STREAM = aws_kinesis_stream.lambda_stream.name
+    }
+  }
+
+  layers = var.otel_lambda_layer
+
+  timeout = 60
+}
+
+resource "aws_lambda_function" "lambda_consumer" {
+  function_name = "${var.prefix}-consumer"
+
+  s3_bucket = aws_s3_bucket.lambda_bucket.id
+  s3_key    = aws_s3_object.consumer_app.key
+
+  runtime = "nodejs20.x"
+  handler = "consumer.consumer"
+
+  source_code_hash = data.archive_file.consumer_app.output_base64sha256
+
+  role = aws_iam_role.lambda_kinesis.arn
+
+  environment {
+    variables = {
+      SPLUNK_ACCESS_TOKEN = var.o11y_access_token
+      SPLUNK_REALM = var.o11y_realm
+      OTEL_SERVICE_NAME = "${var.prefix}-consumer-lambda"
+      OTEL_RESOURCE_ATTRIBUTES = "deployment.environment=${var.prefix}-lambda-shop"
+      AWS_LAMBDA_EXEC_WRAPPER = "/opt/nodejs-otel-handler"
+    }
+  }
+
+  layers = var.otel_lambda_layer
+
+  timeout = 60
+}
+
+# Add API Gateway API, Stage, Integration, Route and Permission Resources
+resource "aws_apigatewayv2_api" "lambda" {
+  name = "${var.prefix}-gateway"
+  protocol_type = "HTTP"
+}
+
+resource "aws_apigatewayv2_stage" "lambda" {
+  api_id = aws_apigatewayv2_api.lambda.id
+
+  name = "${var.prefix}-gw-stage"
+  auto_deploy = true
+
+  access_log_settings {
+    destination_arn = aws_cloudwatch_log_group.api_gw.arn
+
+    format = jsonencode({
+      requestId               = "$context.requestId"
+      sourceIp                = "$context.identity.sourceIp"
+      requestTime             = "$context.requestTime"
+      protocol                = "$context.protocol"
+      httpMethod              = "$context.httpMethod"
+      resourcePath            = "$context.resourcePath"
+      routeKey                = "$context.routeKey"
+      status                  = "$context.status"
+      responseLength          = "$context.responseLength"
+      integrationErrorMessage = "$context.integrationErrorMessage"
+      }
+    )
+  }
+}
+
+resource "aws_apigatewayv2_integration" "lambda" {
+  api_id = aws_apigatewayv2_api.lambda.id
+
+  integration_uri    = aws_lambda_function.lambda_producer.invoke_arn
+  integration_type   = "AWS_PROXY"
+  integration_method = "POST"
+}
+
+resource "aws_apigatewayv2_route" "lambda" {
+  api_id = aws_apigatewayv2_api.lambda.id
+
+  route_key = "POST /producer"
+  target    = "integrations/${aws_apigatewayv2_integration.lambda.id}"
+}
+
+# Lambda Permission for API Gateway
+resource "aws_lambda_permission" "api_gw" {
+  statement_id  = "AllowExecutionFromAPIGateway"
+  action        = "lambda:InvokeFunction"
+  function_name = aws_lambda_function.lambda_producer.function_name
+  principal     = "apigateway.amazonaws.com"
+
+  source_arn = "${aws_apigatewayv2_api.lambda.execution_arn}/*/*"
+}
+
+# CloudWatch Log Groups for Lambda and API Gateway
+resource "aws_cloudwatch_log_group" "lambda_producer" {
+  name = "/aws/lambda/${aws_lambda_function.lambda_producer.function_name}"
+
+  retention_in_days = 30
+}
+
+resource "aws_cloudwatch_log_group" "lambda_consumer" {
+  name = "/aws/lambda/${aws_lambda_function.lambda_consumer.function_name}"
+
+  retention_in_days = 30
+}
+
+resource "aws_cloudwatch_log_group" "api_gw" {
+  name = "/aws/api_gw/${aws_apigatewayv2_api.lambda.name}"
+
+  retention_in_days = 30
+}
+
+# Kinesis Data Stream
+resource "aws_kinesis_stream" "lambda_stream" {
+    name = "${var.prefix}-lambda_stream"
+    shard_count = 1
+    retention_period = 24
+    shard_level_metrics = [
+        "IncomingBytes",
+        "OutgoingBytes"
+    ]
+}
+
+# Source Mapping Lambda Consumer to Kinesis Stream
+resource "aws_lambda_event_source_mapping" "kinesis_lambda_event_mapping" {
+    batch_size = 100
+    event_source_arn = aws_kinesis_stream.lambda_stream.arn
+    enabled = true
+    function_name = aws_lambda_function.lambda_consumer.arn
+    starting_position = "LATEST"
+}

workshop/lambda/auto/main.tf

@@ -0,0 +1,47 @@
+# Output value definitions
+
+
+output "lambda_bucket_name" {
+  description = "S3 Bucket"
+  value = aws_s3_bucket.lambda_bucket.id
+}
+
+output "base_url" {
+  description = "Endpoint"
+  value = "${aws_apigatewayv2_stage.lambda.invoke_url}/producer"
+}
+
+output "producer_function_name" {
+  description = "Lambda Producer"
+  value = aws_lambda_function.lambda_producer.function_name
+}
+
+output "producer_log_group_name" {
+  description = "Log Group Name for the Lambda Producer"
+  value = aws_cloudwatch_log_group.lambda_producer.name
+}
+
+output "producer_log_group_arn" {
+  description = "Log Group ARN for the Lambda Producer"
+  value = aws_cloudwatch_log_group.lambda_producer.arn
+}
+
+output "consumer_function_name" {
+  description = "Lambda Consumer"
+  value = aws_lambda_function.lambda_consumer.function_name
+}
+
+output "consumer_log_group_name" {
+  description = "Log Group Name for the Lambda Consumer"
+  value = aws_cloudwatch_log_group.lambda_consumer.name
+}
+
+output "consumer_log_group_arn" {
+  description = "Log Group ARN for the Lambda Consumer"
+  value = aws_cloudwatch_log_group.lambda_consumer.arn
+}
+
+output "environment" {
+  description = "Deployment Environment"
+  value = "${var.prefix}-lambda-shop"
+}