… | |
… | |
31 | |
31 | |
32 | =cut |
32 | =cut |
33 | |
33 | |
34 | package Coro::Semaphore; |
34 | package Coro::Semaphore; |
35 | |
35 | |
36 | BEGIN { eval { require warnings } && warnings->unimport ("uninitialized") } |
36 | no warnings; |
37 | |
37 | |
38 | use Coro (); |
38 | use Coro (); |
39 | |
39 | |
40 | $VERSION = 1.9; |
40 | $VERSION = 5.0; |
41 | |
41 | |
42 | =item new [inital count] |
42 | =item new [inital count] |
43 | |
43 | |
44 | Creates a new sempahore object with the given initial lock count. The |
44 | Creates a new sempahore object with the given initial lock count. The |
45 | default lock count is 1, which means it is unlocked by default. Zero (or |
45 | default lock count is 1, which means it is unlocked by default. Zero (or |
46 | negative values) are also allowed, in which case the semaphore is locked |
46 | negative values) are also allowed, in which case the semaphore is locked |
47 | by default. |
47 | by default. |
48 | |
48 | |
49 | =cut |
49 | =item $sem->count |
50 | |
50 | |
51 | sub new { |
51 | Returns the current semaphore count. |
52 | bless [defined $_[1] ? $_[1] : 1], $_[0]; |
52 | |
53 | } |
53 | =item $sem->adjust ($diff) |
|
|
54 | |
|
|
55 | Atomically adds the amount given to the current semaphore count. If the |
|
|
56 | count becomes positive, wakes up any waiters. Does not block if the count |
|
|
57 | becomes negative, however. |
54 | |
58 | |
55 | =item $sem->down |
59 | =item $sem->down |
56 | |
60 | |
57 | Decrement the counter, therefore "locking" the semaphore. This method |
61 | Decrement the counter, therefore "locking" the semaphore. This method |
58 | waits until the semaphore is available if the counter is zero. |
62 | waits until the semaphore is available if the counter is zero. |
59 | |
63 | |
60 | =item $status = $sem->timed_down($timeout) |
|
|
61 | |
|
|
62 | Like C<down>, but returns false if semaphore couldn't be acquired within |
|
|
63 | $timeout seconds, otherwise true. |
|
|
64 | |
|
|
65 | =cut |
64 | =cut |
66 | |
65 | |
67 | sub down { |
66 | #=item $status = $sem->timed_down ($timeout) |
68 | while ($_[0][0] <= 0) { |
67 | # |
69 | push @{$_[0][1]}, $Coro::current; |
68 | #Like C<down>, but returns false if semaphore couldn't be acquired within |
70 | Coro::schedule; |
69 | #$timeout seconds, otherwise true. |
71 | } |
|
|
72 | --$_[0][0]; |
|
|
73 | } |
|
|
74 | |
70 | |
75 | sub timed_down { |
71 | #sub timed_down { |
76 | require Coro::Timer; |
72 | # require Coro::Timer; |
77 | my $timeout = Coro::Timer::timeout($_[1]); |
73 | # my $timeout = Coro::Timer::timeout ($_[1]); |
78 | |
74 | # |
79 | while ($_[0][0] <= 0) { |
75 | # while ($_[0][0] <= 0) { |
80 | push @{$_[0][1]}, $Coro::current; |
76 | # push @{$_[0][1]}, $Coro::current; |
81 | Coro::schedule; |
77 | # &Coro::schedule; |
82 | if ($timeout) { |
78 | # if ($timeout) { |
83 | # ugly as hell. slow, too, btw! |
79 | # # ugly as hell. slow, too, btw! |
84 | for (0..$#{$_[0][1]}) { |
80 | # for (0..$#{$_[0][1]}) { |
85 | if ($_[0][1][$_] == $Coro::current) { |
81 | # if ($_[0][1][$_] == $Coro::current) { |
86 | splice @{$_[0][1]}, $_, 1; |
82 | # splice @{$_[0][1]}, $_, 1; |
87 | return; |
83 | # return; |
88 | } |
84 | # } |
89 | } |
85 | # } |
90 | die; |
86 | # die; |
91 | } |
87 | # } |
92 | } |
88 | # } |
93 | |
89 | # |
94 | --$_[0][0]; |
90 | # --$_[0][0]; |
95 | return 1; |
91 | # return 1; |
96 | } |
92 | #} |
97 | |
93 | |
98 | =item $sem->up |
94 | =item $sem->up |
99 | |
95 | |
100 | Unlock the semaphore again. |
96 | Unlock the semaphore again. |
101 | |
|
|
102 | =cut |
|
|
103 | |
|
|
104 | sub up { |
|
|
105 | if (++$_[0][0] > 0) { |
|
|
106 | (shift @{$_[0][1]})->ready if @{$_[0][1]}; |
|
|
107 | } |
|
|
108 | } |
|
|
109 | |
97 | |
110 | =item $sem->try |
98 | =item $sem->try |
111 | |
99 | |
112 | Try to C<down> the semaphore. Returns true when this was possible, |
100 | Try to C<down> the semaphore. Returns true when this was possible, |
113 | otherwise return false and leave the semaphore unchanged. |
101 | otherwise return false and leave the semaphore unchanged. |
114 | |
102 | |
115 | =cut |
|
|
116 | |
|
|
117 | sub try { |
|
|
118 | if ($_[0][0] > 0) { |
|
|
119 | --$_[0][0]; |
|
|
120 | return 1; |
|
|
121 | } else { |
|
|
122 | return 0; |
|
|
123 | } |
|
|
124 | } |
|
|
125 | |
|
|
126 | =item $sem->waiters |
103 | =item $sem->waiters |
127 | |
104 | |
128 | In scalar context, returns the number of coroutines waiting for this |
105 | In scalar context, returns the number of coroutines waiting for this |
129 | semaphore. |
106 | semaphore. |
130 | |
107 | |
131 | =cut |
|
|
132 | |
|
|
133 | sub waiters { |
|
|
134 | @{$_[0][1]}; |
|
|
135 | } |
|
|
136 | |
|
|
137 | =item $guard = $sem->guard |
108 | =item $guard = $sem->guard |
138 | |
109 | |
139 | This method calls C<down> and then creates a guard object. When the guard |
110 | This method calls C<down> and then creates a guard object. When the guard |
140 | object is destroyed it automatically calls C<up>. |
111 | object is destroyed it automatically calls C<up>. |
141 | |
|
|
142 | =item $guard = $sem->timed_guard($timeout) |
|
|
143 | |
|
|
144 | Like C<guard>, but returns undef if semaphore couldn't be acquired within |
|
|
145 | $timeout seconds, otherwise the guard object. |
|
|
146 | |
112 | |
147 | =cut |
113 | =cut |
148 | |
114 | |
149 | sub guard { |
115 | sub guard { |
150 | &down; |
116 | &down; |
151 | # double indirection because bless works on the referenced |
117 | # double indirection because bless works on the referenced |
152 | # object, not (only) on the reference itself. |
118 | # object, not (only) on the reference itself. |
153 | bless \\$_[0], Coro::Semaphore::guard::; |
119 | bless \\$_[0], Coro::Semaphore::guard::; |
154 | } |
120 | } |
155 | |
121 | |
|
|
122 | #=item $guard = $sem->timed_guard ($timeout) |
|
|
123 | # |
|
|
124 | #Like C<guard>, but returns undef if semaphore couldn't be acquired within |
|
|
125 | #$timeout seconds, otherwise the guard object. |
|
|
126 | |
156 | sub timed_guard { |
127 | #sub timed_guard { |
157 | &timed_down |
128 | # &timed_down |
158 | ? bless \\$_[0], Coro::Semaphore::guard:: |
129 | # ? bless \\$_[0], Coro::Semaphore::guard:: |
159 | : (); |
130 | # : (); |
160 | } |
131 | #} |
161 | |
132 | |
162 | sub Coro::Semaphore::guard::DESTROY { |
133 | sub Coro::Semaphore::guard::DESTROY { |
163 | &up(${${$_[0]}}); |
134 | &up(${${$_[0]}}); |
164 | } |
135 | } |
165 | |
136 | |