io.quarkus.funqy.lambda.event.kinesis.KinesisEventHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-funqy-amazon-lambda Show documentation

Amazon Lambda Binding for Quarkus Funqy framework

The newest version!

package io.quarkus.funqy.lambda.event.kinesis;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.List;
import java.util.function.Supplier;
import java.util.stream.Stream;

import com.amazonaws.services.lambda.runtime.events.KinesisEvent;
import com.amazonaws.services.lambda.runtime.events.KinesisEvent.KinesisEventRecord;
import com.amazonaws.services.lambda.runtime.events.StreamsEventResponse;
import com.amazonaws.services.lambda.runtime.events.StreamsEventResponse.BatchItemFailure;
import com.fasterxml.jackson.databind.util.ByteBufferBackedInputStream;

import io.quarkus.funqy.lambda.config.FunqyAmazonConfig;
import io.quarkus.funqy.lambda.event.EventHandler;

public class KinesisEventHandler implements EventHandler {

    @Override
    public Stream streamEvent(KinesisEvent event, FunqyAmazonConfig amazonConfig) {
        if (event == null) {
            return Stream.empty();
        }
        return event.getRecords().stream().map(KinesisEventRecord::getKinesis);
    }

    @Override
    public String getIdentifier(KinesisEvent.Record message, FunqyAmazonConfig amazonConfig) {
        return message.getSequenceNumber();
    }

    @Override
    public Supplier getBody(KinesisEvent.Record message, FunqyAmazonConfig amazonConfig) {
        if (message.getData() == null) {
            return ByteArrayInputStream::nullInputStream;
        }
        return () -> new ByteBufferBackedInputStream(message.getData());
    }

    @Override
    public StreamsEventResponse createResponse(List failures, FunqyAmazonConfig amazonConfig) {
        if (!amazonConfig.advancedEventHandling().kinesis().reportBatchItemFailures()) {
            return null;
        }
        return StreamsEventResponse.builder().withBatchItemFailures(
                failures.stream().map(id -> BatchItemFailure.builder().withItemIdentifier(id).build()).toList()).build();
    }

    @Override
    public Class getMessageClass() {
        return KinesisEvent.Record.class;
    }
}