ApproximateReceiveCount limit on SqsPartialBatch

lambda.ts

// implementation poc for https://github.com/middyjs/middy/issues/1206

const approximateReceiveLimit = 3;

export const handler = middy<SQSEvent>
  // ...
  .use({
    /**
     * Filter out messages that have been retried 3 times
     *
     * @remarks must be placed before the `@middy/sqs-partial-batch-failure` middleware
     */
    after(request: middy.Request<SQSEvent, SQSBatchResponse>) {
      if (!request.response) {
        throw new Error("response is not set. It's a bug in the middleware chain");
      }
      const records = request.event.Records.reduce(
        (m, r) =>
          Object.assign(m, {
            [r.messageId]: r,
          }),
        {} as Record<string, SQSRecord>
      );

      request.response.batchItemFailures = request.response.batchItemFailures.filter(({ itemIdentifier }) => {
        const record = records[itemIdentifier]!;
        const shouldRetry = parseInt(record.attributes.ApproximateReceiveCount) < approximateReceiveLimit;
        if (!shouldRetry) {
          console.warn(`Failed to process message ${itemIdentifier} after ${approximateReceiveLimit} retries`, record);
        }
        return shouldRetry;
      });
    },
  })
  .use(sqsPartialBatchFailure())
  // ...