1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2020 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.job
21 
22 import akka.actor.typed.receptionist.ServiceKey
23 import akka.actor.typed.{ActorRef, ActorSystem}
24 import akka.cluster.sharding.typed.scaladsl.{ClusterSharding, Entity, EntityTypeKey}
25 import org.make.api.technical.ShardingNoEnvelopeMessageExtractor
26 
27 import scala.concurrent.duration.Duration
28 
29 object JobCoordinator {
30   private val name = "job-coordinator"
31 
32   private val TypeKey: EntityTypeKey[JobActor.Protocol.Command] =
33     EntityTypeKey[JobActor.Protocol.Command]("job")
34 
35   val Key: ServiceKey[JobActor.Protocol.Command] = ServiceKey(name)
36 
37   def apply(system: ActorSystem[_], heartRate: Duration): ActorRef[JobActor.Protocol.Command] = {
38     ClusterSharding(system).init(
39       Entity(TypeKey)(_ => JobActor(heartRate))
40         .withMessageExtractor(ShardingNoEnvelopeMessageExtractor[JobActor.Protocol.Command](numberOfShards = 5))
41     )
42   }
43 }