WIP

Author: ahammond

.projen/deps.json

@@ -1,19 +1,24 @@
 import { clickupCdk } from '@time-loop/clickup-projen';
+
+const name = 'cdk-lambda-eni-usage-metric-publisher';
 const project = new clickupCdk.ClickUpCdkConstructLibrary({
-  author: 'jose-clickup',
-  authorAddress: 'jamoroso@clickup.com',
-  cdkVersion: '2.99.0',
+  author: 'Andrew Hammond',
+  authorAddress: 'ahammond@clickup.com',
+  cdkVersion: '2.107.0',
   defaultReleaseBranch: 'main',
-  devDeps: ['@time-loop/clickup-projen', '@aws-cdk/integ-tests-alpha', 'aws-sdk-mock'],
+  devDeps: ['@types/aws-lambda', '@time-loop/clickup-projen', '@aws-cdk/integ-tests-alpha'],
   jsiiVersion: '~5.0.0',
-  name: 'cdk-lambda-eni-usage-metric-publisher',
+  name,
   projenrcTs: true,
-  repositoryUrl: 'https://github.com/time-loop/cdk-lambda-eni-usage-metric-publisher.git',
+  repositoryUrl: `https://github.com/time-loop/${name}.git`,
   gitignore: ['.vscode/**'],
-  bundledDeps: ['aws-sdk'],
+  bundledDeps: [
+    '@aws-lambda-powertools/metrics',
+    '@aws-lambda-powertools/parameters',
+    '@aws-sdk/client-ssm', // Might not need this. Will we be pulling from SSM in lambda?
+    '@aws-sdk/client-secrets-manager',
+    'ioredis',
+  ],
   peerDeps: ['multi-convention-namer'],
-  // deps: [] /* Runtime dependencies of this module. */,
-  // description: undefined,  /* The description is just a string that helps people understand the purpose of the package. */
-  // packageName: undefined,  /* The "name" in package.json. */
 });
 project.synth();

.projen/tasks.json

@@ -0,0 +1,87 @@
+import * as awsLambda from 'aws-lambda';
+
+
+import { getSecrets, getRedisConnection, getQueueDepth, publishMetrics } from './methods';
+
+
+export interface Result {
+  region: string;
+  vpcId: string;
+  count: number;
+}
+
+
+export const handler = async () => {
+  try {
+    const host = process.env.REDIS_ADDR;
+    if (!host) {
+      throw new Error('REDIS_ADDR environment variable not set');
+    }
+
+    const portStr = process.env.REDIS_PORT;
+    if (!portStr) {
+      throw new Error('REDIS_PORT environment variable not set');
+    }
+    const port = parseInt(portStr, 10);
+    if (isNaN(port)) {
+      throw new Error('REDIS_PORT environment variable does not parseInt');
+    }
+
+    // parse queues
+    const queuesBlob = process.env.QUEUES;
+    if (!queuesBlob) {
+      throw new Error('QUEUES environment variable not set');
+    }
+    const queues: string[] = JSON.parse(queuesBlob);
+
+    // get username/password
+    const redisSecretArn = process.env.REDIS_SECRET_ARN;
+    if (!redisSecretArn) {
+      throw new Error('REDIS_SECRET_ARN environment variable not set');
+    }
+
+    const redisSecretPasswordPath = process.env.REDIS_SECRET_PASSWORD_PATH;
+    if (!redisSecretPasswordPath) {
+      throw new Error('REDIS_SECRET_PASSWORD_PATH environment variable not set');
+    }
+
+    const redisSecretUsernamePath = process.env.REDIS_SECRET_USERNAME_PATH;
+    if (!redisSecretUsernamePath) {
+      throw new Error('REDIS_SECRET_USERNAME_PATH environment variable not set');
+    }
+
+    const { username, password } = await getSecrets({redisSecretArn, redisSecretPasswordPath, redisSecretUsernamePath });
+
+    // connect to redis instance, note this waits until the connection is 'ready'.
+    const redisConnection = await getRedisConnection({
+      connectionName: 'queueDepthMetricPublisher',
+      host,
+      port,
+      username,
+      password,
+      tls: {}, // always use tls, but be easy about it.
+    });
+
+    // capture the values for the queues
+    const results = await Promise.all(queues.map(async(queueName) => {
+      let depth = await getQueueDepth({redisConnection, queueName});
+      if(!depth) {
+        console.error(`Error fetching xlen for ${queueName}`);
+        depth = -1;
+      }
+      return {queueName, depth};
+    }));
+
+    if(results === undefined) {
+      throw new Error('Error fetching ALL queue depths');
+    }
+
+    // publish the metrics
+    publishMetrics(results);
+
+
+  } catch (error) {
+    console.error('Error publishing metric data', error);
+    throw error;
+  }
+};