com.github.imrafaelmerino.kafkacli.AvroSchemas Maven / Gradle / Ivy

Go to download
package com.github.imrafaelmerino.kafkacli;

import jsonvalues.JsObj;
import jsonvalues.JsObjPair;
import org.apache.avro.Schema;
import org.apache.avro.Schema.Parser;

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

class AvroSchemas {

    Parser avroParser = new Parser();
    Map keySchemasPerChannel = new HashMap<>();

    Map valueSchemasPerChannel = new HashMap<>();


    public AvroSchemas(final JsObj conf) {

        for (JsObjPair pair : conf.getObj("channels")) {
            String channel = pair.key();
            JsObj channelConf = pair.value()
                                    .toJsObj();
            String keySchema = channelConf.getStr(ConfigurationFields.KEY_SCHEMA);
            String valueSchema = channelConf.getStr(ConfigurationFields.VALUE_SCHEMA);
            if (valueSchema != null) {
                valueSchemasPerChannel.put(channel,
                                           avroParser.parse(valueSchema));
            }
            if (keySchema != null) {
                keySchemasPerChannel.put(channel,
                                         avroParser.parse(keySchema));
            }
        }

    }


}