Added maps and nested maps to ReflectionSchemaBuilder

sksamuel · Apr 14, 2024 · 653b8e1 · 653b8e1
1 parent ff4fef4
commit 653b8e1
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/...on-avro/src/main/kotlin/com/sksamuel/centurion/avro/generation/ReflectionSchemaBuilder.kt b/...on-avro/src/main/kotlin/com/sksamuel/centurion/avro/generation/ReflectionSchemaBuilder.kt
@@ -36,6 +36,7 @@ class ReflectionSchemaBuilder {
          Float::class -> typeBuilder.floatType()
          Set::class -> typeBuilder.array().items(schemaFor(type.arguments.first().type!!))
          List::class -> typeBuilder.array().items(schemaFor(type.arguments.first().type!!))
+         Map::class -> typeBuilder.map().values(schemaFor(type.arguments[1].type!!))
          is KClass<*> -> if (classifier.java.isEnum)
             typeBuilder
                .enumeration(classifier.java.name)

diff --git a/...vro/src/test/kotlin/com/sksamuel/centurion/avro/generation/ReflectionSchemaBuilderTest.kt b/...vro/src/test/kotlin/com/sksamuel/centurion/avro/generation/ReflectionSchemaBuilderTest.kt
@@ -11,6 +11,8 @@ data class Foo2(val a: String?, val b: Boolean, val c: Long)
 data class Foo3(val set1: Set<Int>, val set2: Set<Int?>)
 data class Foo4(val list1: List<Int>, val list2: List<Int?>)
 data class Foo5(val wine: Wine)
+data class Foo6(val map: Map<String, Boolean>)
+data class Foo7(val map: Map<String, Map<String, String>>)
 
 class ReflectionSchemaBuilderTest : FunSpec({
 
@@ -60,4 +62,20 @@ class ReflectionSchemaBuilderTest : FunSpec({
          .endRecord()
       ReflectionSchemaBuilder().schema(Foo5::class) shouldBe expected
    }
+
+   test("maps") {
+      val expected = SchemaBuilder.record("Foo6").namespace(Foo6::class.java.packageName)
+         .fields()
+         .name("map").type().map().values().booleanType().noDefault()
+         .endRecord()
+      ReflectionSchemaBuilder().schema(Foo6::class) shouldBe expected
+   }
+
+   test("maps of maps") {
+      val expected = SchemaBuilder.record("Foo7").namespace(Foo7::class.java.packageName)
+         .fields()
+         .name("map").type().map().values(SchemaBuilder.builder().map().values().stringType()).noDefault()
+         .endRecord()
+      ReflectionSchemaBuilder().schema(Foo7::class) shouldBe expected
+   }
 })