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
21 
22 import com.sksamuel.avro4s.{Encoder, SchemaFor}
23 import org.apache.kafka.clients.producer.{KafkaProducer, ProducerConfig}
24 import org.apache.kafka.common.serialization.{Serializer, StringSerializer}
25 import org.make.api.docker.DockerKafkaService
26 
27 import java.util.Properties
28 
29 trait KafkaTest extends MakeUnitTest with DockerKafkaService {
30   def createProducer[T: SchemaFor: Encoder]: KafkaProducer[String, T] = {
31     val registryUrl = s"http://localhost:$registryExposedPort"
32     val props = new Properties()
33     props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, s"localhost:$kafkaExposedPort")
34     props.put(ProducerConfig.ACKS_CONFIG, "all")
35     props.put(ProducerConfig.RETRIES_CONFIG, "3")
36     props.put(ProducerConfig.BATCH_SIZE_CONFIG, "16384")
37     props.put(ProducerConfig.LINGER_MS_CONFIG, "1")
38     props.put(ProducerConfig.BUFFER_MEMORY_CONFIG, "33554432")
39     val serializer: Serializer[T] = createSerializer[T](registryUrl)
40     new KafkaProducer(props, new StringSerializer(), serializer)
41   }
42 
43   def createSerializer[T: SchemaFor: Encoder](registryUrl: String): Serializer[T]
44 }