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.extensions
21 
22 import akka.actor.typed.{Behavior, SupervisorStrategy}
23 import akka.actor.typed.scaladsl.Behaviors
24 import kamon.Kamon
25 import kamon.metric.Gauge
26 import org.make.api.technical.MonitorableExecutionContext
27 
28 import scala.concurrent.duration.DurationInt
29 
30 object ThreadPoolMonitoringActor {
31   def apply(): Behavior[Protocol] = {
32     Behaviors.supervise(monitorThreadPools()).onFailure(SupervisorStrategy.resume)
33   }
34 
35   @SuppressWarnings(Array("org.wartremover.warts.Recursion"))
36   private def monitorThreadPools(executors: Map[String, ExecutorWithGauges] = Map.empty): Behavior[Protocol] = {
37     Behaviors.withTimers { timers =>
38       timers.startTimerAtFixedRate(Monitor, 1.second)
39       Behaviors.receiveMessage {
40         case Monitor =>
41           executors.foreachEntry {
42             case (_, executorAndGauges) =>
43               val executor = executorAndGauges.executor
44               executorAndGauges.activeTasks.update(executor.activeTasks)
45               executorAndGauges.currentTasks.update(executor.currentTasks)
46               executorAndGauges.maxTasks.update(executor.maxTasks)
47               executorAndGauges.waitingTasks.update(executor.waitingTasks)
48           }
49           Behaviors.same
50         case MonitorThreadPool(newExecutor, name) =>
51           val executorWithGauges = ExecutorWithGauges(
52             newExecutor,
53             Kamon.gauge("executors-active-threads").withTag("name", name),
54             Kamon.gauge("executors-core-size").withTag("name", name),
55             Kamon.gauge("executors-max-threads").withTag("name", name),
56             Kamon.gauge("executors-waiting").withTag("name", name)
57           )
58           monitorThreadPools(executors + (name -> executorWithGauges))
59       }
60     }
61   }
62   sealed trait Protocol
63   case object Monitor extends Protocol
64   final case class MonitorThreadPool(pool: MonitorableExecutionContext, name: String) extends Protocol
65 
66   final case class ExecutorWithGauges(
67     executor: MonitorableExecutionContext,
68     activeTasks: Gauge,
69     currentTasks: Gauge,
70     maxTasks: Gauge,
71     waitingTasks: Gauge
72   )
73 
74   val name: String = "ThreadPoolMonitoringActor"
75 
76 }