mirror of
https://github.com/unisonweb/unison.git
synced 2024-09-23 16:28:02 +03:00
125 lines
3.5 KiB
Plaintext
125 lines
3.5 KiB
Plaintext
|
|
use Universal == <
|
|
|
|
type Future a = Future ('{Remote} a)
|
|
|
|
-- A simple distributed computation ability
|
|
ability Remote where
|
|
|
|
-- Spawn a new node
|
|
spawn : {Remote} Node
|
|
|
|
-- Sequentially evaluate the given thunk on another node
|
|
-- then return to the current node when it completes
|
|
at : n -> '{Remote} a -> {Remote} a
|
|
|
|
-- Start a computation running, returning an `r` that can be forced to
|
|
-- await the result of the computation
|
|
fork : '{Remote} a ->{Remote} Future a
|
|
|
|
type Node = Node Nat -- more realistic would be perhaps a (Hostname, PublicKey) pair
|
|
|
|
force : Future a ->{Remote} a
|
|
force f = case f of Future.Future r -> !r
|
|
|
|
-- Let's test out this beast! do we need to deploy our code to some EC2 instances??
|
|
-- Gak, no not yet, we just want to test locally, let's write a handler
|
|
-- for the `Remote` ability that simulates everything locally!
|
|
|
|
Remote.runLocal : '{Remote} a -> a
|
|
Remote.runLocal r =
|
|
use Future Future
|
|
step nid r = case r of
|
|
{a} -> a
|
|
{Remote.fork t -> k} -> handle (step nid) in k (Future t)
|
|
{Remote.spawn -> k} -> handle (step (Node.increment nid)) in k nid
|
|
{Remote.at _ t -> k} -> handle (step nid) in k !t
|
|
handle (step (Node.Node 0)) in !r
|
|
|
|
Remote.forkAt : Node -> '{Remote} a ->{Remote} (Future a)
|
|
Remote.forkAt node r = Remote.fork '(Remote.at node r)
|
|
|
|
use Optional None Some
|
|
use Monoid Monoid
|
|
use Sequence ++
|
|
|
|
Sequence.map : (a ->{e} b) -> [a] ->{e} [b]
|
|
Sequence.map f as =
|
|
go f acc as i = case Sequence.at i as of
|
|
None -> acc
|
|
Some a -> go f (acc `snoc` f a) as (i + 1)
|
|
go f [] as 0
|
|
|
|
type Monoid a = Monoid (a -> a -> a) a
|
|
|
|
Monoid.zero m = case m of Monoid.Monoid op z -> z
|
|
Monoid.op m = case m of Monoid.Monoid op z -> op
|
|
|
|
Monoid.orElse m a = case a of
|
|
None -> Monoid.zero m
|
|
Some a -> a
|
|
|
|
merge : (a -> a -> Boolean) -> [a] -> [a] -> [a]
|
|
merge lte a b =
|
|
go acc a b = case Sequence.at 0 a of
|
|
None -> acc ++ b
|
|
Some hd1 -> case at 0 b of
|
|
None -> acc ++ a
|
|
Some hd2 ->
|
|
if hd1 `lte` hd2 then go (acc `snoc` hd1) (drop 1 a) b
|
|
else go (acc `snoc` hd2) a (drop 1 b)
|
|
go [] a b
|
|
|
|
dmap : (a ->{Remote} b) -> [a] ->{Remote} [b]
|
|
dmap f as =
|
|
bs = Sequence.map (a -> Remote.forkAt Remote.spawn '(f a)) as
|
|
Sequence.map force bs
|
|
|
|
dreduce : Monoid a -> [a] ->{Remote} a
|
|
dreduce m a =
|
|
if size a < 2 then Monoid.orElse m (Sequence.at 0 a)
|
|
else
|
|
l = Remote.forkAt Remote.spawn '(dreduce m (take (size a / 2) a))
|
|
r = Remote.forkAt Remote.spawn '(dreduce m (drop (size a / 2) a))
|
|
Monoid.op m (force l) (force r)
|
|
|
|
dmapReduce : (a ->{Remote} b) -> Monoid b -> [a] ->{Remote} b
|
|
dmapReduce f m as = dreduce m (Sequence.map f as)
|
|
|
|
dsort : (a -> a -> Boolean) -> [a] ->{Remote} [a]
|
|
dsort lte a =
|
|
dmapReduce (a -> [a]) (Monoid (merge lte) []) a
|
|
|
|
sort : (a -> a -> Boolean) -> [a] -> [a]
|
|
sort lte a =
|
|
if Sequence.size a < 2 then a
|
|
else
|
|
l = sort lte (take (size a / 2) a)
|
|
r = sort lte (drop (size a / 2) a)
|
|
merge lte l r
|
|
|
|
Node.increment : Node -> Node
|
|
Node.increment n =
|
|
use Node.Node -- the constructor
|
|
case n of Node n -> Node (n + 1)
|
|
|
|
> Remote.runLocal '(dsort (<) [3,2,1,1,2,3,9182,1,2,34,1,23])
|
|
|
|
halve : [a] -> Optional ([a], [a])
|
|
halve a =
|
|
if size a == 0 then None
|
|
else Some (take (size a / 2) a, drop (size a / 2) a)
|
|
|
|
foldMap : (a -> b) -> Monoid b -> [a] -> b
|
|
foldMap f m a =
|
|
base a = case Sequence.at 0 a of
|
|
None -> zero m
|
|
Some a -> f a
|
|
if size a < 2 then base a
|
|
else case halve a of
|
|
None -> zero m
|
|
Some (l, r) -> op m (foldMap f m l) (foldMap f m r)
|
|
|
|
> foldMap (x -> x) (Monoid (+) 0) [1]
|
|
> Remote.runLocal '(dmap (x -> x + 1) [1,2,3,4])
|