1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api.technical
21 
22 import com.sksamuel.avro4s.{Decoder, DefaultFieldMapper, FieldMapper, SchemaFor}
23 import io.confluent.kafka.schemaregistry.client.CachedSchemaRegistryClient
24 import io.confluent.kafka.serializers.KafkaAvroDeserializer
25 import org.apache.avro.Schema
26 import org.apache.avro.generic.GenericRecord
27 import org.apache.kafka.common.serialization.Deserializer
28 
29 import java.util
30 
31 class MakeKafkaAvroDeserializer[T: Decoder: SchemaFor](
32   registryUrl: String,
33   fieldMapper: FieldMapper = DefaultFieldMapper
34 ) extends Deserializer[T] {
35 
36   val schema: Schema = SchemaFor[T].schema(fieldMapper)
37   val decoder: Decoder[T] = Decoder[T]
38 
39   private val identityMapCapacity = 1000
40   private val delegate = new KafkaAvroDeserializer(new CachedSchemaRegistryClient(registryUrl, identityMapCapacity))
41 
42   override def configure(configs: util.Map[String, _], isKey: Boolean): Unit = {
43     delegate.configure(configs, isKey)
44   }
45 
46   override def close(): Unit = {
47     delegate.close()
48   }
49 
50   @SuppressWarnings(Array("org.wartremover.warts.AsInstanceOf"))
51   override def deserialize(topic: String, data: Array[Byte]): T = {
52     decoder.decode(delegate.deserialize(topic, data).asInstanceOf[GenericRecord], schema, fieldMapper)
53   }
54 }