com.genexus.cloud.serverless.aws.handler.LambdaSQSHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gxawsserverless Show documentation
Core classes for the runtime used by Java and Android apps generated with GeneXus
There is a newer version: 4.6.8
package com.genexus.cloud.serverless.aws.handler;

import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.RequestHandler;
import com.amazonaws.services.lambda.runtime.events.SQSBatchResponse;
import com.amazonaws.services.lambda.runtime.events.SQSEvent;
import com.amazonaws.services.lambda.runtime.events.SQSEvent.SQSMessage;
import com.genexus.cloud.serverless.*;
import com.genexus.cloud.serverless.model.EventMessageProperty;
import com.genexus.cloud.serverless.model.EventMessage;
import com.genexus.cloud.serverless.model.EventMessageResponse;
import com.genexus.cloud.serverless.model.EventMessageSourceType;
import com.genexus.cloud.serverless.model.EventMessages;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;


public class LambdaSQSHandler extends LambdaBaseEventHandler implements RequestHandler {

	public LambdaSQSHandler() throws Exception {
		super();
	}

	public LambdaSQSHandler(String className) throws Exception {
		super(className);
	}

	@Override
	public SQSBatchResponse handleRequest(SQSEvent sqsEvent, Context context) {
		if (logger.isDebugEnabled()) {
			logger.debug("handleRequest started with #sqsItems: " + (int) sqsEvent.getRecords().size());
		}

		EventMessages msgs = new EventMessages();

		for (SQSMessage sqsItem : sqsEvent.getRecords()) {
			logger.debug(String.format("Processing sqsEvent Message: %s", sqsItem.getMessageId()));
			EventMessage msg = new EventMessage();
			msg.setMessageId(sqsItem.getMessageId());
			msg.setMessageSourceType(EventMessageSourceType.QUEUE_MESSAGE);
			msg.setMessageDate(new Date());
			msg.setMessageData(sqsItem.getBody());
			List msgAtts = msg.getMessageProperties();

			for (Map.Entry entry : sqsItem.getAttributes().entrySet()) {
				msgAtts.add(new EventMessageProperty(entry.getKey(), entry.getValue()));
			}
			for (Map.Entry entry : sqsItem.getMessageAttributes().entrySet()) {
				msgAtts.add(new EventMessageProperty(entry.getKey(), entry.getValue().getStringValue()));
			}
			msgAtts.add(new EventMessageProperty("eventSource", sqsItem.getEventSource()));
			msgAtts.add(new EventMessageProperty("eventSourceARN", sqsItem.getEventSourceArn()));
			msgAtts.add(new EventMessageProperty("awsRegion", sqsItem.getAwsRegion()));
			msgAtts.add(new EventMessageProperty("receiptHandle", sqsItem.getReceiptHandle()));
			msgAtts.add(new EventMessageProperty("md5Body", sqsItem.getMd5OfBody()));

			msgs.add(msg);
		}

		boolean wasHandled = false;
		String errorMessage;

		try {
			EventMessageResponse response = dispatchEvent(msgs, Helper.toJSONString(sqsEvent));
			wasHandled = !response.hasFailed();
			errorMessage = response.getErrorMessage();
		} catch (Exception e) {
			errorMessage = "HandleRequest execution error";
			logger.error(errorMessage, e);
		}

		if (!wasHandled) {
			logger.error(String.format("Messages were not handled. Marking all SQS Events as failed: %s", errorMessage));
			List failures = new ArrayList<>();
			//Assume all batch has failed.
			for (SQSMessage sqsItem : sqsEvent.getRecords()) {
				failures.add(new SQSBatchResponse.BatchItemFailure(sqsItem.getMessageId()));
			}
			return new SQSBatchResponse(failures);
		} else {
			logger.info("Event Message processed successfully");
		}

		return null;
	}
}