1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
|
package test;
import scala.actors.TIMEOUT;
class MailBox {
private class LinkedList[a] {
var elem: a = _;
var next: LinkedList[a] = null;
}
private def insert[a](l: LinkedList[a], x: a): LinkedList[a] = {
l.next = new LinkedList[a];
l.next.elem = x;
l.next.next = l.next;
l
}
private abstract class Receiver {
def isDefined(msg: Any): Boolean;
var msg: Any = null;
}
private val sent = new LinkedList[Any];
private var lastSent = sent;
private val receivers = new LinkedList[Receiver];
private var lastReceiver = receivers;
def send(msg: Any): Unit = synchronized {
var r = receivers;
var r1 = r.next;
while (r1 != null && !r1.elem.isDefined(msg)) {
r = r1; r1 = r1.next;
}
if (r1 != null) {
r.next = r1.next; r1.elem.msg = msg; r1.elem.notify();
} else {
lastSent = insert(lastSent, msg);
}
}
def receive[a](f: PartialFunction[Any, a]): a = {
val msg: Any = synchronized {
var s = sent;
var s1 = s.next;
while (s1 != null && !f.isDefinedAt(s1.elem)) {
s = s1; s1 = s1.next
}
if (s1 != null) {
s.next = s1.next; s1.elem
} else {
val r = insert(lastReceiver, new Receiver {
def isDefined(msg: Any) = f.isDefinedAt(msg);
});
lastReceiver = r;
r.elem.wait();
r.elem.msg
}
}
f(msg)
}
def receiveWithin[a](msec: Long)(f: PartialFunction[Any, a]): a = {
val msg: Any = synchronized {
var s = sent;
var s1 = s.next;
while (s1 != null && !f.isDefinedAt(s1.elem)) {
s = s1; s1 = s1.next ;
}
if (s1 != null) {
s.next = s1.next; s1.elem
} else {
val r = insert(lastReceiver, new Receiver {
def isDefined(msg: Any) = f.isDefinedAt(msg);
});
lastReceiver = r;
r.elem.wait(msec);
if (r.elem.msg == null) r.elem.msg = TIMEOUT;
r.elem.msg
}
}
f(msg)
}
}
|