-
Notifications
You must be signed in to change notification settings - Fork 46
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Updated RecordDecoderGenerator to run from KClass
- Loading branch information
Showing
2 changed files
with
109 additions
and
43 deletions.
There are no files selected for viewing
80 changes: 56 additions & 24 deletions
80
...ion-avro/src/main/kotlin/com/sksamuel/centurion/avro/generation/RecordDecoderGenerator.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,50 +1,82 @@ | ||
package com.sksamuel.centurion.avro.generation | ||
|
||
import kotlin.reflect.KClass | ||
import kotlin.reflect.KProperty1 | ||
import kotlin.reflect.KType | ||
import kotlin.reflect.full.declaredMemberProperties | ||
|
||
/** | ||
* A code generator that outputs Kotlin that will deserialize a generic record for a given | ||
* type into an instance of that data class. | ||
*/ | ||
class RecordDecoderGenerator { | ||
fun generate(ds: DataClass): String { | ||
|
||
fun generate(kclass: KClass<*>): String { | ||
return buildString { | ||
appendLine("package ${ds.packageName}") | ||
appendLine("package ${kclass.java.packageName}") | ||
appendLine() | ||
appendLine("import com.sksamuel.centurion.avro.decoders.*") | ||
appendLine("import org.apache.avro.Schema") | ||
appendLine("import org.apache.avro.generic.GenericData") | ||
appendLine("import org.apache.avro.generic.GenericRecord") | ||
appendLine() | ||
appendLine("/**") | ||
appendLine(" * This is a generated [Decoder] that deserializes Avro [GenericRecord]s to [${ds.className}]s") | ||
appendLine(" * This is a generated [Decoder] that deserializes Avro [GenericRecord]s to [${kclass.java.simpleName}]s") | ||
appendLine(" */") | ||
appendLine("object ${ds.className}Decoder : Decoder<${ds.className}> {") | ||
appendLine(" override fun decode(schema: Schema, value: Any?): ${ds.className} {") | ||
appendLine(" require(value is GenericRecord)") | ||
appendLine(" return ${ds.className}(") | ||
ds.members.forEach { member -> | ||
appendLine(" ${member.name} = ${decoderFor(member)},") | ||
appendLine("object ${kclass.java.simpleName}Decoder : Decoder<${kclass.java.simpleName}> {") | ||
appendLine() | ||
appendLine(" override fun decode(schema: Schema): (Any?) -> ${kclass.java.simpleName} {") | ||
appendLine() | ||
kclass.declaredMemberProperties.forEach { property -> | ||
appendLine(" val ${property.name}Schema = schema.getField(\"${property.name}\").schema()") | ||
appendLine(" val ${property.name}Pos = schema.getField(\"${property.name}\").pos()") | ||
appendLine(" val ${property.name}Decode = ${decode(property)}") | ||
} | ||
appendLine(" )") | ||
appendLine() | ||
appendLine(" return { record ->") | ||
appendLine(" require(record is GenericRecord)") | ||
appendLine(" ${kclass.java.simpleName}(") | ||
kclass.declaredMemberProperties.forEach { property -> | ||
appendLine(" ${property.name} = ${property.name}Decode(record[${property.name}Pos]),") | ||
} | ||
appendLine(" )") | ||
appendLine(" }") | ||
appendLine(" }") | ||
appendLine("}") | ||
} | ||
} | ||
|
||
private fun decode(property: KProperty1<out Any, *>): String { | ||
val baseDecoder = decoderFor(property.returnType) | ||
val wrapped = if (property.returnType.isMarkedNullable) "NullDecoder($baseDecoder)" else baseDecoder | ||
return "$wrapped.decode(${property.name}Schema)" | ||
} | ||
|
||
private fun decoderFor(type: KType): String { | ||
return when (val classifier = type.classifier) { | ||
Boolean::class -> "BooleanDecoder" | ||
Double::class -> "DoubleDecoder" | ||
Float::class -> "FloatDecoder" | ||
Int::class -> "IntDecoder" | ||
Long::class -> "LongDecoder" | ||
String::class -> "StringDecoder" | ||
Set::class -> { | ||
val elementDecoder = decoderFor(type.arguments.first().type!!) | ||
"SetDecoder($elementDecoder)" | ||
} | ||
|
||
List::class -> { | ||
val elementDecoder = decoderFor(type.arguments.first().type!!) | ||
"ListDecoder($elementDecoder)" | ||
} | ||
|
||
Map::class -> { | ||
val valueDecoder = decoderFor(type.arguments[1].type!!) | ||
"MapDecoder($valueDecoder)" | ||
} | ||
|
||
private fun decoderFor(member: Member): String { | ||
val getSchema = "schema.getField(\"${member.name}\").schema()" | ||
val getValue = "value.get(\"${member.name}\")" | ||
return when (member.type) { | ||
Type.BooleanType -> "BooleanDecoder.decode($getSchema, $getValue)" | ||
Type.DoubleType -> "DoubleDecoder.decode($getSchema, $getValue)" | ||
Type.FloatType -> "FloatDecoder.decode($getSchema, $getValue)" | ||
Type.IntType -> "IntDecoder.decode($getSchema, $getValue)" | ||
Type.LongType -> "LongDecoder.decode($getSchema, $getValue)" | ||
is Type.Nullable -> TODO() | ||
is Type.RecordType -> TODO() | ||
Type.StringType -> "StringDecoder.decode($getSchema, $getValue)" | ||
is Type.ArrayType -> "ListDecoder.decode($getSchema, $getValue)" | ||
is Type.MapType -> "MapDecoder.decode($getSchema, $getValue)" | ||
is KClass<*> -> if (classifier.java.isEnum) "EnumDecoder<${classifier.java.name}>()" else error("Unsupported type: $type") | ||
else -> error("Unsupported type: $type") | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters