import AWS from "aws-sdk";

const dynamoDb = new AWS.DynamoDB.DocumentClient();

export async function main() {

const getParams = {

// Get the table name from the environment variable

TableName: process.env.tableName,

// Get the row where the counter is called "clicks"

Key: {

counter: "clicks",

},

};

const results = await dynamoDb.get(getParams).promise();

// If there is a row, then get the value of the

// column called "tally"

let count = results.Item ? results.Item.tally : 0;

const putParams = {

TableName: process.env.tableName,

Key: {

counter: "clicks",

},

// Update the "tally" column

UpdateExpression: "SET tally = :count",

ExpressionAttributeValues: {

// Increase the count

":count": ++count,

},

};

await dynamoDb.update(putParams).promise();

return {

statusCode: 200,

body: count,

};

}