com.github.fge.avro.translators.AvroTranslators Maven / Gradle / Ivy

Go to download
package com.github.fge.avro.translators;

import com.github.fge.jackson.NodeType;
import com.google.common.collect.ImmutableMap;
import org.apache.avro.Schema;

import java.util.Map;

public final class AvroTranslators
{
    private static final Map TRANSLATORS;

    private AvroTranslators()
    {
    }

    static {
        final ImmutableMap.Builder builder
            = ImmutableMap.builder();

        Schema.Type avroType;
        AvroTranslator translator;

        avroType = Schema.Type.NULL;
        translator = new SimpleTypeTranslator(NodeType.NULL);
        builder.put(avroType, translator);

        avroType = Schema.Type.BOOLEAN;
        translator = new SimpleTypeTranslator(NodeType.BOOLEAN);
        builder.put(avroType, translator);

        avroType = Schema.Type.STRING;
        translator = new SimpleTypeTranslator(NodeType.STRING);
        builder.put(avroType, translator);

        // Reuse for "bytes"
        avroType = Schema.Type.BYTES;
        translator = ByteTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.INT;
        translator = IntTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.LONG;
        translator = LongTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.FLOAT;
        translator = new SimpleTypeTranslator(NodeType.NUMBER);
        builder.put(avroType, translator);

        // Reuse for "double"
        avroType = Schema.Type.DOUBLE;
        builder.put(avroType, translator);

        avroType = Schema.Type.MAP;
        translator = MapTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.FIXED;
        translator = FixedTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.ENUM;
        translator = EnumTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.ARRAY;
        translator = ArrayTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.UNION;
        translator = UnionTranslator.getInstance();
        builder.put(avroType, translator);

        avroType = Schema.Type.RECORD;
        translator = RecordTranslator.getInstance();
        builder.put(avroType, translator);

        TRANSLATORS = builder.build();
    }

    public static AvroTranslator getTranslator(final Schema.Type avroType)
    {
        return TRANSLATORS.get(avroType);
    }
}