Add utility scripts and SQL setup file

Author: kaiz-io

typescript/postgres-lambda/setup-postgres-lambda.sql

@@ -0,0 +1,41 @@
+-- SQL script to set up PostgreSQL to call Lambda
+-- Replace <AWS_REGION> with your AWS region (e.g., us-east-1)
+-- Replace <POSTGRES_FUNCTION_NAME> with the name of your Lambda function
+
+-- Create the aws_lambda extension
+CREATE EXTENSION IF NOT EXISTS aws_lambda CASCADE;
+
+-- Create a function to process data using Lambda
+CREATE OR REPLACE FUNCTION process_data(data JSONB)
+RETURNS JSONB AS $$
+SELECT payload FROM aws_lambda.invoke(
+  aws_commons.create_lambda_function_arn('<POSTGRES_FUNCTION_NAME>', '<AWS_REGION>'),
+  json_build_object('action', 'process', 'data', data)::text,
+  'Event'
+);
+$$ LANGUAGE SQL;
+
+-- Create a function to transform data using Lambda
+CREATE OR REPLACE FUNCTION transform_data(data JSONB)
+RETURNS JSONB AS $$
+SELECT payload FROM aws_lambda.invoke(
+  aws_commons.create_lambda_function_arn('<POSTGRES_FUNCTION_NAME>', '<AWS_REGION>'),
+  json_build_object('action', 'transform', 'data', data)::text,
+  'Event'
+);
+$$ LANGUAGE SQL;
+
+-- Create a function to validate data using Lambda
+CREATE OR REPLACE FUNCTION validate_data(data JSONB)
+RETURNS JSONB AS $$
+SELECT payload FROM aws_lambda.invoke(
+  aws_commons.create_lambda_function_arn('<POSTGRES_FUNCTION_NAME>', '<AWS_REGION>'),
+  json_build_object('action', 'validate', 'data', data)::text,
+  'Event'
+);
+$$ LANGUAGE SQL;
+
+-- Test the functions
+SELECT process_data('{"id": 123, "value": "test"}'::JSONB);
+SELECT transform_data('{"id": 456, "value": "hello world"}'::JSONB);
+SELECT validate_data('{"id": 789, "value": "valid data"}'::JSONB);

typescript/postgres-lambda/test-lambda.sh

@@ -0,0 +1,75 @@
+#!/bin/bash
+
+# Test script for PostgreSQL Lambda integration
+
+# Check if jq is installed
+if ! command -v jq &> /dev/null; then
+    echo "Error: jq is required but not installed. Please install jq first."
+    exit 1
+fi
+
+# Function to display usage
+function display_usage {
+    echo "Usage: $0 [options]"
+    echo "Options:"
+    echo "  -f, --function-name NAME    Lambda function name to test"
+    echo "  -m, --message MESSAGE       Message to send to Lambda (default: 'Test message')"
+    echo "  -h, --help                  Display this help message"
+    exit 1
+}
+
+# Default values
+MESSAGE="Test message"
+
+# Parse command line arguments
+while [[ $# -gt 0 ]]; do
+    key="$1"
+    case $key in
+        -f|--function-name)
+            FUNCTION_NAME="$2"
+            shift
+            shift
+            ;;
+        -m|--message)
+            MESSAGE="$2"
+            shift
+            shift
+            ;;
+        -h|--help)
+            display_usage
+            ;;
+        *)
+            echo "Unknown option: $1"
+            display_usage
+            ;;
+    esac
+done
+
+# Check if function name is provided
+if [ -z "$FUNCTION_NAME" ]; then
+    echo "Error: Lambda function name is required."
+    display_usage
+fi
+
+# Create payload
+PAYLOAD=$(echo "{\"message\": \"$MESSAGE\"}" | jq -c .)
+
+# Create temporary file for response
+RESPONSE_FILE=$(mktemp)
+
+echo "Invoking Lambda function: $FUNCTION_NAME"
+echo "Payload: $PAYLOAD"
+
+# Invoke Lambda function
+aws lambda invoke \
+    --function-name "$FUNCTION_NAME" \
+    --payload "$PAYLOAD" \
+    --cli-binary-format raw-in-base64-out \
+    "$RESPONSE_FILE"
+
+# Display response
+echo "Response:"
+cat "$RESPONSE_FILE" | jq .
+
+# Clean up
+rm "$RESPONSE_FILE"