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>
*/
object Pi {
val nrOfWorkers = 4
val nrOfMessages = 10000
val nrOfElements = 10000
// ====================
// ===== Messages =====
// ====================
sealed trait PiMessage
case class Calculate(nrOfMessages: Int, nrOfElements: Int) extends PiMessage
case class Work(arg: Int, fun: (Int) => Double) extends PiMessage
case object Calculate extends PiMessage
case class Work(arg: Int, nrOfElements: Int) extends PiMessage
case class Result(value: Double) extends PiMessage
// ==================
// ===== 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 = {
case Work(arg, fun) => self reply Result(fun(arg))
case Work(arg, nrOfElements) =>
self reply Result(calculatePiFor(arg, nrOfElements)) // perform the work
}
}
// ==================
// ===== Master =====
// ==================
class Master(latch: CountDownLatch) extends Actor {
class Master(nrOfWorkers: Int, nrOfMessages: Int, nrOfElements: Int, latch: CountDownLatch) extends Actor {
var pi: Double = _
var nrOfResults: Int = _
var start: Long = _
@ -71,17 +75,11 @@ object Pi {
// wrap them with a load-balancing router
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
def receive = {
case Calculate(nrOfMessages, nrOfElements) =>
case Calculate =>
// 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
router ! Broadcast(PoisonPill)
@ -106,21 +104,20 @@ object Pi {
// ===== Run it =====
// ==================
def calculate = {
val nrOfWorkers = 4
val nrOfMessages = 10000
val nrOfElements = 10000
// this latch is only plumbing to know when the calculation is completed
val latch = new CountDownLatch(1)
// create the master
val master = actorOf(new Master(latch)).start
val master = actorOf(new Master(nrOfWorkers, nrOfMessages, nrOfElements, latch)).start
// start the calculation
master ! Calculate(nrOfMessages, nrOfElements)
master ! Calculate
// wait for master to shut down
latch.await
}
}
// To be able to run it as a main application
object Main extends App {
Pi.calculate
}