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 akka.actor.typed.Behavior
23 import akka.actor.typed.scaladsl.Behaviors
24 import kamon.Kamon
25 import kamon.metric.Gauge
26 
27 import scala.concurrent.duration.DurationInt
28 
29 object MemoryMonitoringActor {
30   def apply(): Behavior[Monitor.type] = {
31     ActorSystemHelper.superviseWithBackoff(monitorMemory())
32   }
33 
34   private def monitorMemory(): Behavior[Monitor.type] = {
35     val memoryMax: Gauge = Kamon.gauge("jvm-memory-max").withoutTags()
36     val memoryFree: Gauge = Kamon.gauge("jvm-memory-free").withoutTags()
37     val memoryTotal: Gauge = Kamon.gauge("jvm-memory-total").withoutTags()
38 
39     Behaviors.withTimers { timers =>
40       timers.startTimerAtFixedRate(Monitor, 1.second)
41       Behaviors.receiveMessage {
42         case Monitor =>
43           memoryFree.update(Runtime.getRuntime.freeMemory().toDouble)
44           memoryMax.update(Runtime.getRuntime.maxMemory().toDouble)
45           memoryTotal.update(Runtime.getRuntime.totalMemory().toDouble)
46           Behaviors.same
47       }
48     }
49   }
50 
51   case object Monitor
52 
53   val name: String = "memory-monitor-backoff"
54 }