consumer-with-exception

consumer.py

import logging
import json

log = logging.getLogger(__name__)
log.setLevel(logging.DEBUG)


def lambda_handler(event, context):
    for record in event['Records']:
        message_id = record['messageId']
        attributes = record['attributes']
        approximateReceiveCount = attributes['ApproximateReceiveCount']

        body = record['body']
        msg = json.loads(body)

        if msg['no'] == 5:
            raise ValueError("Something wrong.")

        log.info("Consumed a message: no=[%s], messageId=[%s], ApproximateReceiveCount=[%s]",
                 msg['no'], message_id, approximateReceiveCount)

    log.info("Finish %d records", len(event['Records']))