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.healthcheck
21 
22 import akka.persistence.cassandra.query.scaladsl.CassandraReadJournal
23 import org.make.api.ConfigComponent
24 import org.make.api.extensions.{KafkaConfigurationComponent, MakeSettingsComponent}
25 import org.make.api.technical.BaseReadJournalComponent
26 import org.make.api.technical.ExecutorServiceHelper.RichExecutorService
27 import org.make.api.technical.elasticsearch.{ElasticsearchClientComponent, ElasticsearchConfigurationComponent}
28 import org.make.api.technical.healthcheck.HealthCheck.HealthCheckResponse
29 import org.make.api.technical.storage.SwiftClientComponent
30 
31 import java.util.concurrent.Executors
32 import scala.concurrent.{ExecutionContext, Future}
33 
34 trait HealthCheckService {
35   def runAllHealthChecks(): Future[Seq[HealthCheckResponse]]
36 }
37 
38 trait HealthCheckServiceComponent {
39   def healthCheckService: HealthCheckService
40 }
41 
42 trait DefaultHealthCheckServiceComponent extends HealthCheckServiceComponent {
43   self: BaseReadJournalComponent[CassandraReadJournal]
44     with ConfigComponent
45     with ElasticsearchClientComponent
46     with ElasticsearchConfigurationComponent
47     with KafkaConfigurationComponent
48     with MakeSettingsComponent
49     with SwiftClientComponent =>
50 
51   override lazy val healthCheckService: HealthCheckService = new DefaultHealthCheckService
52 
53   class DefaultHealthCheckService extends HealthCheckService {
54     implicit val ctx: ExecutionContext =
55       Executors.newFixedThreadPool(10).instrument("healthchecks").toExecutionContext
56 
57     val healthChecks: Seq[(String, HealthCheck)] = Seq(
58       "avro-health-check" -> new AvroHealthCheck(kafkaConfiguration),
59       "cassandra-health-check" -> new CassandraHealthCheck(proposalJournal, config),
60       "cockroach-health-check" -> new CockroachHealthCheck(makeSettings.defaultAdmin.email),
61       "elasticsearch-health-check" -> new ElasticsearchHealthCheck(elasticsearchConfiguration, elasticsearchClient),
62       "kafka-health-check" -> new KafkaHealthCheck(kafkaConfiguration),
63       "swift-health-check" -> new SwiftHealthCheck(swiftClient),
64       "zookeeper-health-check" -> new ZookeeperHealthCheck(config)
65     )
66 
67     override def runAllHealthChecks(): Future[Seq[HealthCheckResponse]] = {
68       Future.traverse(healthChecks) {
69         case (name, hc) =>
70           hc.healthCheck().map(status => HealthCheckResponse(name, status.message)).recoverWith {
71             case e => Future.successful(HealthCheckResponse(name, e.getMessage))
72           }
73       }
74     }
75   }
76 }