Use a switch expression in TypeDefinitionAwareCodec