somod-chat-service/build/serverless/functions/wsOnConnect.js

Serverless Chat Service from SOMOD

import { UserProviderMiddlewareKey } from "../../lib";
import { DynamoDBClient, PutItemCommand } from "@aws-sdk/client-dynamodb";
import { marshall } from "@aws-sdk/util-dynamodb";
const dynamodb = new DynamoDBClient();
const WebsocketOnConnect = async (event) => {
    const userId = event.somodMiddlewareContext.get(UserProviderMiddlewareKey);
    const connectionId = event.requestContext.connectionId;
    await dynamodb.send(new PutItemCommand({
        TableName: process.env.CONNECTIONS_TABLE_NAME + "",
        Item: marshall({ id: connectionId, userId: userId })
    }));
    return { statusCode: 200, body: "Connected." };
};
export default WebsocketOnConnect;