samples.s3.KinesisMessageModelS3Transformer.java Source code

Introduction

Here is the source code for samples.s3.KinesisMessageModelS3Transformer.java

Source

/*
 * Copyright 2013-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.s3;

import com.amazonaws.services.dynamodbv2.model.AttributeValue;
import com.amazonaws.services.kinesis.connectors.BasicJsonTransformer;
import com.amazonaws.services.kinesis.connectors.dynamodb.DynamoDBTransformer;
import com.amazonaws.services.kinesis.connectors.impl.JsonToByteArrayTransformer;
import com.amazonaws.services.kinesis.model.Record;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import samples.KinesisMessageModel;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

/**
 * A custom transfomer for {@link samples.KinesisMessageModel} records in JSON format. The output is in a format
 * usable for insertions to Amazon DynamoDB.
 */
public class KinesisMessageModelS3Transformer extends JsonToByteArrayTransformer<KinesisMessageModel> {

    private static final Log LOG = LogFactory.getLog(KinesisMessageModelS3Transformer.class);

    /**
     * Creates a new KinesisMessageModelDynamoDBTransformer.
     */
    public KinesisMessageModelS3Transformer() {
        super(KinesisMessageModel.class);
    }

    @Override
    public KinesisMessageModel toClass(Record record) throws IOException {
        try {
            return KinesisMessageModel.newInstance(new String(record.getData().array(), "UTF-8"));
        } catch (Exception e) {
            String message = "Error parsing record from JSON: " + new String(record.getData().array());
            LOG.error(message, e);
            throw new IOException(message, e);
        }
    }
}