serverless-spy/lib/functions/onDisconnect.mjs

CDK-based library for writing elegant integration tests on AWS serverless architecture and an additional web console to monitor events in real time.

const __dirname = import.meta.dirname;
import { envVariableNames, init_envVariableNames } from "../src/common/envVariableNames.mjs";
import { DeleteItemCommand, DynamoDBClient } from "@aws-sdk/client-dynamodb";

//#region functions/onDisconnect.ts
init_envVariableNames();
const ddb = new DynamoDBClient({ region: process.env.AWS_REGION });
const handler = async (event) => {
	console.log("EVENT", JSON.stringify(event));
	const deleteParams = new DeleteItemCommand({
		TableName: process.env[envVariableNames.SSPY_WS_TABLE_NAME],
		Key: { connectionId: { S: event.requestContext.connectionId } }
	});
	try {
		await ddb.send(deleteParams);
	} catch (err) {
		console.error(err);
		return {
			statusCode: 500,
			body: "Failed to disconnect: " + JSON.stringify(err)
		};
	}
	return {
		statusCode: 200,
		body: "Disconnected."
	};
};

//#endregion
export { handler };
//# sourceMappingURL=onDisconnect.mjs.map