1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2021 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.{ActorRef, ActorSystem}
23 import akka.actor.typed.scaladsl.AskPattern._
24 import akka.util.Timeout
25 
26 import java.util.concurrent.TimeoutException
27 import scala.concurrent.{ExecutionContext, Future, Promise}
28 
29 object BetterLoggingActors {
30 
31   implicit class BetterLoggingTypedActorRef[Req](val ref: ActorRef[Req]) extends AnyVal {
32     def ??[Res](
33       replyTo: ActorRef[Res] => Req
34     )(implicit ec: ExecutionContext, system: ActorSystem[_], timeout: Timeout): Future[Res] = {
35       val message = Promise[Req]()
36       ref.ask { sender: ActorRef[Res] =>
37         val value = replyTo(sender)
38         message.success(value)
39         value
40       }.recoverWith {
41         case e: TimeoutException => message.future.flatMap(m => Future.failed(ActorTimeoutException(m, e)))
42         case other               => Future.failed(other)
43       }
44     }
45   }
46 
47 }