samples.elasticsearch.BatchedKinesisMessageModelElasticsearchTransformer.java Source code

Introduction

Here is the source code for samples.elasticsearch.BatchedKinesisMessageModelElasticsearchTransformer.java

Source

/*
 * Copyright 2014 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Amazon Software License (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 * http://aws.amazon.com/asl/
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */
package samples.elasticsearch;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.util.Collection;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import samples.KinesisMessageModel;

import com.amazonaws.services.kinesis.connectors.elasticsearch.ElasticsearchObject;
import com.amazonaws.services.kinesis.connectors.interfaces.ICollectionTransformer;
import com.amazonaws.services.kinesis.model.Record;

/**
 * Extends KinesisMessageModelElasticsearchTransformer and implements ICollectionTransformer
 * to provide a toClass method to transform an Amazon Kinesis Record that contains multiple
 * instances of KinesisMessageModel.
 * 
 * To see how these records were batched, view {@class samples.BatchedStreamSource}.
 */
public class BatchedKinesisMessageModelElasticsearchTransformer extends KinesisMessageModelElasticsearchTransformer
        implements ICollectionTransformer<KinesisMessageModel, ElasticsearchObject> {
    private static final Log LOG = LogFactory.getLog(BatchedKinesisMessageModelElasticsearchTransformer.class);

    @SuppressWarnings("unchecked")
    @Override
    public Collection<KinesisMessageModel> toClass(Record record) throws IOException {

        try (ObjectInputStream ois = new ObjectInputStream(new ByteArrayInputStream(record.getData().array()))) {
            return (Collection<KinesisMessageModel>) ois.readObject();
        } catch (Exception e) {
            String message = "Error reading object from ObjectInputStream: " + new String(record.getData().array());
            LOG.error(message, e);
            throw new IOException(message, e);
        }
    }

}