Improve PostgreSQL Lambda integration

- Update SQL functions to use JSONB casting instead of text for proper JSON handling
- Change Lambda invocation type from 'Event' to 'RequestResponse' for synchronous execution
- Add parameter group to RDS cluster for custom configuration
- Update Aurora PostgreSQL version to 17.4
- Fix code formatting and whitespace in Lambda functions

Author: kaiz-io

typescript/postgres-lambda/lambda/postgres-setup/index.js

@@ -40,26 +40,26 @@ exports.handler = async (event, context) => {
       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'
+        json_build_object('action', 'process', 'data', data)::JSONB,
+        'RequestResponse'
       );
       $$ LANGUAGE SQL;
 
       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'
+        json_build_object('action', 'transform', 'data', data)::JSONB,
+        'RequestResponse'
       );
       $$ LANGUAGE SQL;
 
       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'
+        json_build_object('action', 'validate', 'data', data)::JSONB,
+        'RequestResponse'
       );
       $$ LANGUAGE SQL;
     `;

typescript/postgres-lambda/lambda/postgres-to-lambda/index.js

@@ -1,9 +1,9 @@
 /**
  * Lambda function that is called by PostgreSQL
- * 
+ *
  * This function can be invoked from PostgreSQL using the aws_lambda extension
  * Example SQL:
- * 
+ *
  * SELECT * FROM aws_lambda.invoke(
  *   aws_commons.create_lambda_function_arn('PostgresFunction', 'us-east-1'),
  *   '{"action": "process", "data": {"id": 123, "value": "test"}}',
@@ -12,14 +12,14 @@
  */
 exports.handler = async (event) => {
   console.log('Event received from PostgreSQL:', JSON.stringify(event));
-  
+
   try {
     // Process the event data
     const action = event.action || 'default';
     const data = event.data || {};
-    
+
     let result;
-    
+
     // Perform different actions based on the event
     switch (action) {
       case 'process':
@@ -34,7 +34,7 @@ exports.handler = async (event) => {
       default:
         result = { status: 'success', message: 'Default action performed', data };
     }
-    
+
     return {
       statusCode: 200,
       body: result,

typescript/postgres-lambda/lib/postgres-lambda-stack.ts

@@ -16,6 +16,13 @@ export class PostgresLambdaStack extends cdk.Stack {
       maxAzs: 2,
       natGateways: 1,
     });
+    const parameterGroup = new rds.ParameterGroup(this, 'PGClusterParamGroup', {
+      engine: rds.DatabaseClusterEngine.auroraPostgres({
+        version: rds.AuroraPostgresEngineVersion.VER_17_4
+      }),
+      parameters: {
+      },
+    });
 
     // Create a PostgreSQL Aurora Serverless v2 cluster
     const dbCluster = new rds.DatabaseCluster(this, 'PostgresCluster', {
@@ -28,13 +35,16 @@ export class PostgresLambdaStack extends cdk.Stack {
       serverlessV2MaxCapacity: 1,
       defaultDatabaseName: 'demodb',
       credentials: rds.Credentials.fromGeneratedSecret('postgres'),
+      parameterGroup: parameterGroup,
     });
 
+
+
     const bundleCommand = [
-            'bash', '-c', [
-              'cp -r . /asset-output/',
-            ].join(' && ')
-          ]
+      'bash', '-c', [
+        'cp -r . /asset-output/',
+      ].join(' && ')
+    ]
 
     // Create a Lambda function that calls PostgreSQL with Docker bundling
     const lambdaToPostgres = new lambda.Function(this, 'LambdaToPostgres', {