1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2020 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.healthcheck
21 
22 import akka.util.Timeout
23 import org.apache.kafka.clients.admin.{AdminClient, DescribeClusterOptions}
24 import org.apache.kafka.clients.producer.ProducerConfig
25 import org.make.api.extensions.KafkaConfiguration
26 import org.make.api.technical.TimeSettings
27 import org.make.api.technical.healthcheck.HealthCheck.Status
28 
29 import java.util.Properties
30 import scala.concurrent.{ExecutionContext, Future}
31 import scala.util.{Failure, Success, Using}
32 
33 class KafkaHealthCheck(kafkaConfiguration: KafkaConfiguration) extends HealthCheck {
34 
35   override val techno: String = "kafka"
36 
37   val timeout: Timeout = TimeSettings.defaultTimeout
38 
39   private def createClient() = {
40     val properties = new Properties
41     properties.setProperty(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, kafkaConfiguration.connectionString)
42     AdminClient.create(properties)
43   }
44 
45   override def healthCheck()(implicit ctx: ExecutionContext): Future[Status] = {
46     Future {
47       Using(createClient()) { client =>
48         client
49           .describeCluster(new DescribeClusterOptions().timeoutMs(kafkaConfiguration.pollTimeout.toInt))
50           .clusterId()
51           .get(timeout.duration.length, timeout.duration.unit)
52       } match {
53         case Success(_) => Status.OK
54         case Failure(e) => Status.NOK(Some(e.getMessage))
55       }
56     }
57   }
58 }