mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
* lib/thread.rb (Queue#push, #pop, SizedQueue#push, #pop): remove
code that kicks waiting thread twice, which caused race and deadlock. [ruby-core:25537] * test/thread/test_queue.rb: added. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@27356 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
parent
7d26c313a1
commit
93c5002a7c
3 changed files with 50 additions and 20 deletions
|
@ -1,3 +1,11 @@
|
||||||
|
Fri Apr 16 20:05:24 2010 Yusuke Endoh <mame@tsg.ne.jp>
|
||||||
|
|
||||||
|
* lib/thread.rb (Queue#push, #pop, SizedQueue#push, #pop): remove
|
||||||
|
code that kicks waiting thread twice, which caused race and
|
||||||
|
deadlock. [ruby-core:25537]
|
||||||
|
|
||||||
|
* test/thread/test_queue.rb: added.
|
||||||
|
|
||||||
Fri Apr 16 20:01:47 2010 Yusuke Endoh <mame@tsg.ne.jp>
|
Fri Apr 16 20:01:47 2010 Yusuke Endoh <mame@tsg.ne.jp>
|
||||||
|
|
||||||
* .gitignore: updated.
|
* .gitignore: updated.
|
||||||
|
|
|
@ -150,7 +150,6 @@ class Queue
|
||||||
# Pushes +obj+ to the queue.
|
# Pushes +obj+ to the queue.
|
||||||
#
|
#
|
||||||
def push(obj)
|
def push(obj)
|
||||||
t = nil
|
|
||||||
@mutex.synchronize{
|
@mutex.synchronize{
|
||||||
@que.push obj
|
@que.push obj
|
||||||
begin
|
begin
|
||||||
|
@ -160,10 +159,6 @@ class Queue
|
||||||
retry
|
retry
|
||||||
end
|
end
|
||||||
}
|
}
|
||||||
begin
|
|
||||||
t.run if t
|
|
||||||
rescue ThreadError
|
|
||||||
end
|
|
||||||
end
|
end
|
||||||
|
|
||||||
#
|
#
|
||||||
|
@ -182,8 +177,8 @@ class Queue
|
||||||
# thread isn't suspended, and an exception is raised.
|
# thread isn't suspended, and an exception is raised.
|
||||||
#
|
#
|
||||||
def pop(non_block=false)
|
def pop(non_block=false)
|
||||||
while true
|
@mutex.synchronize{
|
||||||
@mutex.synchronize{
|
while true
|
||||||
if @que.empty?
|
if @que.empty?
|
||||||
raise ThreadError, "queue empty" if non_block
|
raise ThreadError, "queue empty" if non_block
|
||||||
@waiting.push Thread.current
|
@waiting.push Thread.current
|
||||||
|
@ -191,8 +186,8 @@ class Queue
|
||||||
else
|
else
|
||||||
return @que.shift
|
return @que.shift
|
||||||
end
|
end
|
||||||
}
|
end
|
||||||
end
|
}
|
||||||
end
|
end
|
||||||
|
|
||||||
#
|
#
|
||||||
|
@ -295,7 +290,6 @@ class SizedQueue < Queue
|
||||||
# until space becomes available.
|
# until space becomes available.
|
||||||
#
|
#
|
||||||
def push(obj)
|
def push(obj)
|
||||||
t = nil
|
|
||||||
@mutex.synchronize{
|
@mutex.synchronize{
|
||||||
while true
|
while true
|
||||||
break if @que.length < @max
|
break if @que.length < @max
|
||||||
|
@ -311,11 +305,6 @@ class SizedQueue < Queue
|
||||||
retry
|
retry
|
||||||
end
|
end
|
||||||
}
|
}
|
||||||
|
|
||||||
begin
|
|
||||||
t.run if t
|
|
||||||
rescue ThreadError
|
|
||||||
end
|
|
||||||
end
|
end
|
||||||
|
|
||||||
#
|
#
|
||||||
|
@ -333,7 +322,6 @@ class SizedQueue < Queue
|
||||||
#
|
#
|
||||||
def pop(*args)
|
def pop(*args)
|
||||||
retval = super
|
retval = super
|
||||||
t = nil
|
|
||||||
@mutex.synchronize {
|
@mutex.synchronize {
|
||||||
if @que.length < @max
|
if @que.length < @max
|
||||||
begin
|
begin
|
||||||
|
@ -344,10 +332,6 @@ class SizedQueue < Queue
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
}
|
}
|
||||||
begin
|
|
||||||
t.run if t
|
|
||||||
rescue ThreadError
|
|
||||||
end
|
|
||||||
retval
|
retval
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
38
test/thread/test_queue.rb
Normal file
38
test/thread/test_queue.rb
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
require 'test/unit'
|
||||||
|
require 'thread'
|
||||||
|
|
||||||
|
class TestQueue < Test::Unit::TestCase
|
||||||
|
def test_queue
|
||||||
|
grind(5, 1000, 15, Queue)
|
||||||
|
end
|
||||||
|
|
||||||
|
def test_sized_queue
|
||||||
|
grind(5, 1000, 15, SizedQueue, 1000)
|
||||||
|
end
|
||||||
|
|
||||||
|
def grind(num_threads, num_objects, num_iterations, klass, *args)
|
||||||
|
from_workers = klass.new(*args)
|
||||||
|
to_workers = klass.new(*args)
|
||||||
|
|
||||||
|
workers = (1..num_threads).map {
|
||||||
|
Thread.new {
|
||||||
|
while object = to_workers.pop
|
||||||
|
from_workers.push object
|
||||||
|
end
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Thread.new {
|
||||||
|
num_iterations.times {
|
||||||
|
num_objects.times { to_workers.push 99 }
|
||||||
|
num_objects.times { from_workers.pop }
|
||||||
|
}
|
||||||
|
}.join
|
||||||
|
|
||||||
|
num_threads.times { to_workers.push nil }
|
||||||
|
workers.each { |t| t.join }
|
||||||
|
|
||||||
|
assert 0, from_workers.size
|
||||||
|
assert 0, to_workers.size
|
||||||
|
end
|
||||||
|
end
|
Loading…
Add table
Reference in a new issue