如何解决如何优雅地关闭 Ruby 中的线程
过去一周我一直在 Ruby 中试验多线程概念。
为了练习,我正在设计一个文件下载器,它可以对一组 URL 进行并行请求。目前我需要在触发中断信号时安全地关闭线程。我已经阅读了多线程理论并在运行时捕获信号。然而,尽管有这些理论知识,我仍然不知道如何在实践中使用它们。
无论如何,我要把我的概念验证工作留在下面。
class MultiThread
attr_reader :limit,:threads,:queue
def initialize(limit)
@limit = limit
@threads = []
@queue = Queue.new
end
def add(*args,&block)
queue << [block,args]
end
def invoke
1.upto(limit).each { threads << spawn_thread }
threads.each(&:join)
end
private
def spawn_thread
Thread.new do
Thread.handle_interrupt(RuntimeError => :on_blocking) do
# nothing to do
end
until queue.empty?
block,args = queue.pop
block&.call(*args)
end
end
end
end
urls = %w[https://example.com]
thread = MultiThread.new(2)
urls.each do |url|
thread.add do
puts "Downloading #{url}..."
sleep 1
end
end
thread.invoke
解决方法
是的,handle_interrupt
的文档令人困惑。试试这个,我基于 connection_pool
gem 使用的,例如puma
。
$stdout.sync = true
threads = 3.times.map { |i|
Thread.new {
Thread.handle_interrupt(Exception => :never) do
begin
Thread.handle_interrupt(Exception => :immediate) do
puts "Thread #{i} doing work"
sleep 1000
end
ensure
puts "Thread #{i} cleaning up"
end
end
}
}
Signal.trap("INT") {
puts 'Exiting gracefully'
threads.each { |t|
puts 'killing thread'
t.kill
}
exit
}
threads.each { |t| t.join }
输出:
Thread 1 doing work
Thread 2 doing work
Thread 0 doing work
^CExiting gracefully
killing thread
killing thread
killing thread
Thread 0 cleaning up
Thread 1 cleaning up
Thread 2 cleaning up
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。