1 ! Copyright (C) 2008 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: dlists kernel math concurrency.promises
4 concurrency.mailboxes accessors fry ;
5 IN: concurrency.count-downs
7 ! http://java.sun.com/j2se/1.5.0/docs/api/java/util/concurrent/CountDownLatch.html
9 TUPLE: count-down-tuple n promise ;
11 : count-down-check ( count-down -- )
12 dup n>> zero? [ t swap promise>> fulfill ] [ drop ] if ;
14 ERROR: invalid-count-down-count count ;
16 : <count-down> ( n -- count-down )
17 dup 0 < [ throw-invalid-count-down-count ] when
18 <promise> \ count-down-tuple boa
19 dup count-down-check ;
21 ERROR: count-down-already-done ;
23 : count-down ( count-down -- )
25 [ throw-count-down-already-done ]
26 [ 1 - >>n count-down-check ] if ;
28 : await-timeout ( count-down timeout -- )
29 [ promise>> ] dip ?promise-timeout ?linked t assert= ;
31 : await ( count-down -- )
34 : spawn-stage ( quot count-down -- )
35 [ '[ @ _ count-down ] ] keep
37 swap promise>> mailbox>> spawn-linked-to drop ;