com.redislabs.riot.stream.AvroToMapConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of riot-stream Show documentation

RIOT Core module

The newest version!

package com.redislabs.riot.stream;

import org.apache.avro.generic.GenericRecord;
import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.springframework.core.convert.converter.Converter;

import java.util.HashMap;
import java.util.Map;

public class AvroToMapConverter implements Converter, Map> {

    private final Converter, Map> flattener;

    public AvroToMapConverter(Converter, Map> flattener) {
        this.flattener = flattener;
    }

    @Override
    public Map convert(ConsumerRecord source) {
        GenericRecord record = (GenericRecord) source.value();
        Map map = new HashMap<>();
        record.getSchema().getFields().forEach(field -> map.put(field.name(), record.get(field.name())));
        return flattener.convert(map);
    }

}