com.noleme.flow.connect.avro.transformer.AvroDeserializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of noleme-flow-connect-kafka Show documentation

The newest version!

package com.noleme.flow.connect.avro.transformer;

import com.noleme.flow.actor.transformer.TransformationException;
import com.noleme.flow.actor.transformer.Transformer;
import com.noleme.flow.connect.avro.AvroHelper;
import org.apache.avro.Schema;
import org.apache.avro.generic.GenericRecord;
import org.apache.avro.io.DatumReader;
import org.apache.avro.io.Decoder;
import org.apache.avro.io.DecoderFactory;
import org.apache.avro.specific.SpecificDatumReader;
import org.apache.avro.specific.SpecificRecordBase;

import java.io.IOException;

/**
 * @author Pierre Lecerf ([email protected])
 */
public class AvroDeserializer implements Transformer
{
    private final Schema schema;

    public AvroDeserializer(Class targetType)
    {
        this.schema = AvroHelper.getSchema(targetType);
    }

    @Override
    @SuppressWarnings("unchecked")
    public T transform(Byte[] bytes) throws TransformationException
    {
        if (bytes == null)
            return null;

        try {
            DatumReader datumReader = new SpecificDatumReader<>(this.schema);
            Decoder decoder = DecoderFactory.get().binaryDecoder(AvroHelper.toPrimitive(bytes), null);
            
            return (T) datumReader.read(null, decoder);
        }
        catch (IOException e) {
            throw new TransformationException("Unable to deserialize incoming bytes[]", e);
        }
    }
}