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.receptionist.{Receptionist, ServiceKey}
23 import akka.actor.typed.scaladsl.AskPattern.Askable
24 import akka.actor.typed.scaladsl.{ActorContext, Behaviors}
25 import akka.actor.typed.{ActorRef, ActorSystem, BackoffSupervisorStrategy, Behavior, Scheduler, SupervisorStrategy}
26 import akka.util.Timeout
27 
28 import scala.concurrent.duration.DurationInt
29 import scala.concurrent.Future
30 
31 object ActorSystemHelper {
32   implicit class RichActorSystem(val self: ActorSystem[_]) extends AnyVal {
33     import self.executionContext
34 
35     def findRefByKey[T](key: ServiceKey[T])(implicit timeout: Timeout, scheduler: Scheduler): Future[ActorRef[T]] = {
36       (self.receptionist ? Receptionist.Find(key)).flatMap {
37         case key.Listing(services) =>
38           services.find(_.path.address.hasLocalScope) match {
39             case Some(ref) => Future.successful(ref)
40             case None =>
41               Future.failed(new IllegalStateException(s"Receptionist unable to find actor ref for key: $key"))
42           }
43       }
44 
45     }
46   }
47 
48   private val maxRestarts = 50
49   val DefaultFallbackStrategy: BackoffSupervisorStrategy =
50     SupervisorStrategy
51       .restartWithBackoff(minBackoff = 3.seconds, maxBackoff = 30.seconds, randomFactor = 0.2)
52       .withMaxRestarts(maxRestarts)
53 
54   def superviseWithBackoff[T](behavior: Behavior[T]): Behavior[T] = {
55     Behaviors.supervise(behavior).onFailure(DefaultFallbackStrategy)
56   }
57 
58   implicit def contextToSystem(implicit context: ActorContext[_]): ActorSystem[_] = context.system
59 }