Write and test JSON Schema to avro map

Signed-off-by: Francis Galiegue <[email protected]>

Author: fge

src/main/java/com/github/fge/jsonschema2avro/AvroWriterProcessor.java

@@ -12,6 +12,7 @@
 import com.github.fge.jsonschema.util.JsonLoader;
 import com.github.fge.jsonschema.util.ValueHolder;
 import com.github.fge.jsonschema2avro.writers.ArrayWriter;
+import com.github.fge.jsonschema2avro.writers.MapWriter;
 import com.github.fge.jsonschema2avro.writers.SimpleTypeWriter;
 import org.apache.avro.Schema;
 
@@ -29,6 +30,7 @@ public AvroWriterProcessor()
         processor = new ProcessorSelector<AvroPayload, ValueHolder<Schema>>()
             .when(simpleType()).then(SimpleTypeWriter.getInstance())
             .when(array()).then(ArrayWriter.getInstance())
+            .when(map()).then(MapWriter.getInstance())
             .getProcessor();
     }
 

src/main/java/com/github/fge/jsonschema2avro/writers/MapWriter.java

@@ -0,0 +1,37 @@
+package com.github.fge.jsonschema2avro.writers;
+
+import com.github.fge.jsonschema.exceptions.ProcessingException;
+import com.github.fge.jsonschema.jsonpointer.JsonPointer;
+import com.github.fge.jsonschema.processors.data.SchemaHolder;
+import com.github.fge.jsonschema.report.ProcessingReport;
+import com.github.fge.jsonschema.tree.SchemaTree;
+import com.github.fge.jsonschema2avro.AvroWriterProcessor;
+import org.apache.avro.Schema;
+
+public final class MapWriter
+    extends AvroWriter
+{
+    private static final JsonPointer POINTER
+        = JsonPointer.of("additionalProperties");
+
+    private static final AvroWriter INSTANCE = new MapWriter();
+
+    private MapWriter()
+    {
+    }
+
+    public static AvroWriter getInstance()
+    {
+        return INSTANCE;
+    }
+
+    @Override
+    protected Schema generate(final AvroWriterProcessor writer,
+        final ProcessingReport report, final SchemaTree tree)
+        throws ProcessingException
+    {
+        final SchemaHolder input = new SchemaHolder(tree.append(POINTER));
+        final Schema valueSchema = writer.process(report, input).getValue();
+        return Schema.createMap(valueSchema);
+    }
+}

src/test/java/com/github/fge/jsonschema2avro/MapTest.java

@@ -0,0 +1,13 @@
+package com.github.fge.jsonschema2avro;
+
+import java.io.IOException;
+
+public final class MapTest
+    extends AvroWriterProcessorTest
+{
+    public MapTest()
+        throws IOException
+    {
+        super("map");
+    }
+}

src/test/resources/jsonschema2avro/map.json

@@ -0,0 +1,25 @@
+[
+    {
+        "jsonSchema": {
+            "type": "object",
+            "additionalProperties": { "type": "string" }
+        },
+        "avroSchema": {
+            "type": "map",
+            "values": "string"
+        }
+    },
+    {
+        "jsonSchema": {
+            "type": "object",
+            "additionalProperties": {
+                "type": "array",
+                "items": { "type": "number" }
+            }
+        },
+        "avroSchema": {
+            "type": "map",
+            "values": { "type": "array", "items": "double" }
+        }
+    }
+]