All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • io.wizzie.normalizer.funcs.impl.ArrayFlattenMapper Maven / Gradle / Ivy

    package io.wizzie.normalizer.funcs.impl;

import io.wizzie.metrics.MetricsManager;
import io.wizzie.normalizer.funcs.FlatMapperFunction;
import org.apache.kafka.streams.KeyValue;

import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class ArrayFlattenMapper extends FlatMapperFunction {

    String flatDimension;

    @Override
    public void prepare(Map properties, MetricsManager metricsManager) {
        flatDimension = (String) properties.get("flat_dimension");
    }

    @Override
    public Iterable>> process(String key, Map value) {
        List>> results = Collections.singletonList(new KeyValue<>(key, value));

        if (value != null && flatDimension != null) {

            if (value.containsKey(flatDimension)) {
                List array = (List) value.remove(flatDimension);

                if (array != null) {
                    results = array.stream().map(val -> {

                        Map newValue = new HashMap<>();
                        newValue.putAll(value);

                        if (val instanceof Map) {
                            newValue.putAll((Map) val);
                        } else {
                            newValue.put(flatDimension, val);
                        }

                        KeyValue> kv = new KeyValue<>(key, newValue);

                        return kv;

                    }).collect(Collectors.toList());
                }

            }
        }
        return results;
    }

    @Override
    public String toString() {
        StringBuilder builder = new StringBuilder();
        builder.append(" {")
                .append("flat_dimension: ").append(flatDimension).append("} ");

        return builder.toString();
    }

    @Override
    public void stop() {

    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy