AWS Step Function Activity Wrapper.js

const AWS = require("aws-sdk");
const proxy = require("proxy-agent");

const {
	getActivityTask,
	sendTaskFailure,
	sendTaskHeartbeat,
	sendTaskSuccess,
} = require("./awsHelper");
const { worker } = require("./worker");

AWS.config.update({
	region: "eu-west-1",
});

const activityRunner = async (heartbeatSecs = 890) => {
	const activityArn = process.env.SFN_ACTIVITY_ARN;

	let heartbeat;
	const task = await getActivityTask(activityArn, "<Activity Worker Name>");
	if (!task) return undefined;

	const { taskToken, input } = task;
	heartbeat = setInterval(() => {
		sendTaskHeartbeat(taskToken);
	}, heartbeatSecs * 1000);

	try {
		const result = await worker(input);
		await sendTaskSuccess(taskToken, result);
	} catch (err) {
		await sendTaskFailure(taskToken, err.stack);
	} finally {
		clearInterval(heartbeat);
	}
};

const main = async () => {
	while (true) {
		await activityRunner();
	}
};

main();

module.exports = {
	activityRunner,
};