raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Simplified the Master/Worker interaction in first tutorial

Browse source
This commit is contained in:
Jonas Bonér 2011-04-02 14:01:36 +02:00
parent 5303f9f25a
commit 64d620c7b1
1 changed files with 20 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

43
akka-tutorials/akka-tutorial-pi-sbt/src/main/scala/Pi.scala
View file

@ -32,31 +32,35 @@ import java.util.concurrent.CountDownLatch
* @author <a href="http://jonasboner.com">Jonas Bon&#233;r</a> * @author <a href="http://jonasboner.com">Jonas Bon&#233;r</a>
*/ */
object Pi { object Pi {
val nrOfWorkers = 4
val nrOfMessages = 10000
val nrOfElements = 10000
// ==================== // ====================
// ===== Messages ===== // ===== Messages =====
// ==================== // ====================
sealed trait PiMessage sealed trait PiMessage
case class Calculate(nrOfMessages: Int, nrOfElements: Int) extends PiMessage case object Calculate extends PiMessage
case class Work(arg: Int, fun: (Int) => Double) extends PiMessage case class Work(arg: Int, nrOfElements: Int) extends PiMessage
case class Result(value: Double) extends PiMessage case class Result(value: Double) extends PiMessage
// ================== // ==================
// ===== Worker ===== // ===== Worker =====
// ================== // ==================
class Worker extends Actor { class Worker() extends Actor {
// define the work
val calculatePiFor = (arg: Int, nrOfElements: Int) => {
val range = (arg * nrOfElements) to ((arg + 1) * nrOfElements - 1)
range map (j => 4 * math.pow(-1, j) / (2 * j + 1)) sum
}
def receive = { def receive = {
case Work(arg, fun) => self reply Result(fun(arg)) case Work(arg, nrOfElements) =>
self reply Result(calculatePiFor(arg, nrOfElements)) // perform the work
} }
} }
// ================== // ==================
// ===== Master ===== // ===== Master =====
// ================== // ==================
class Master(latch: CountDownLatch) extends Actor { class Master(nrOfWorkers: Int, nrOfMessages: Int, nrOfElements: Int, latch: CountDownLatch) extends Actor {
var pi: Double = _ var pi: Double = _
var nrOfResults: Int = _ var nrOfResults: Int = _
var start: Long = _ var start: Long = _
@ -71,17 +75,11 @@ object Pi {
// wrap them with a load-balancing router // wrap them with a load-balancing router
val router = Routing.loadBalancerActor(CyclicIterator(workers)).start val router = Routing.loadBalancerActor(CyclicIterator(workers)).start
// define the work
val algorithm = (i: Int) => {
val range = (i * nrOfElements) to ((i + 1) * nrOfElements - 1)
range map (j => 4 * math.pow(-1, j) / (2 * j + 1)) sum
}
// message handler // message handler
def receive = { def receive = {
case Calculate(nrOfMessages, nrOfElements) => case Calculate =>
// schedule work // schedule work
for (arg <- 0 until nrOfMessages) router ! Work(arg, algorithm) for (arg <- 0 until nrOfMessages) router ! Work(arg, nrOfElements)
// send a PoisonPill to all workers telling them to shut down themselves // send a PoisonPill to all workers telling them to shut down themselves
router ! Broadcast(PoisonPill) router ! Broadcast(PoisonPill)
@ -106,21 +104,20 @@ object Pi {
// ===== Run it ===== // ===== Run it =====
// ================== // ==================
def calculate = { def calculate = {
val nrOfWorkers = 4
val nrOfMessages = 10000
val nrOfElements = 10000
// this latch is only plumbing to know when the calculation is completed // this latch is only plumbing to know when the calculation is completed
val latch = new CountDownLatch(1) val latch = new CountDownLatch(1)
// create the master // create the master
val master = actorOf(new Master(latch)).start val master = actorOf(new Master(nrOfWorkers, nrOfMessages, nrOfElements, latch)).start
// start the calculation // start the calculation
master ! Calculate(nrOfMessages, nrOfElements) master ! Calculate
// wait for master to shut down // wait for master to shut down
latch.await latch.await
} }
} }
// To be able to run it as a main application
object Main extends App {
Pi.calculate
}