mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
267e1d25dc
To assist in various debugging scenarios, such as [tracking which types of threads are using db connections][0]
Thread names will now be "ActionCable-worker-N" instead of "worker-N".
This is safe for all ruby versions. In versions which do not have thread names, [concurrent-ruby does not attempt to add the name][1]
In the environment I happened to be testing in:
```ruby
e = Concurrent::ThreadPoolExecutor.new(name: 'ActionCable', min_threads: 2, max_threads: 2)
e.post{sleep 100}
e.post{sleep 100}
Thread.list.map(&:name)
```
[0] https://github.com/puma/puma/issues/1512#issuecomment-756760388
[1] https://github.com/ruby-concurrency/concurrent-ruby/blob/master/lib/concurrent-ruby/concurrent/executor/ruby_thread_pool_executor.rb#L312
76 lines
2 KiB
Ruby
76 lines
2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "active_support/callbacks"
|
|
require "active_support/core_ext/module/attribute_accessors_per_thread"
|
|
require "action_cable/server/worker/active_record_connection_management"
|
|
require "concurrent"
|
|
|
|
module ActionCable
|
|
module Server
|
|
# Worker used by Server.send_async to do connection work in threads.
|
|
class Worker # :nodoc:
|
|
include ActiveSupport::Callbacks
|
|
|
|
thread_mattr_accessor :connection
|
|
define_callbacks :work
|
|
include ActiveRecordConnectionManagement
|
|
|
|
attr_reader :executor
|
|
|
|
def initialize(max_size: 5)
|
|
@executor = Concurrent::ThreadPoolExecutor.new(
|
|
name: "ActionCable",
|
|
min_threads: 1,
|
|
max_threads: max_size,
|
|
max_queue: 0,
|
|
)
|
|
end
|
|
|
|
# Stop processing work: any work that has not already started
|
|
# running will be discarded from the queue
|
|
def halt
|
|
@executor.shutdown
|
|
end
|
|
|
|
def stopping?
|
|
@executor.shuttingdown?
|
|
end
|
|
|
|
def work(connection)
|
|
self.connection = connection
|
|
|
|
run_callbacks :work do
|
|
yield
|
|
end
|
|
ensure
|
|
self.connection = nil
|
|
end
|
|
|
|
def async_exec(receiver, *args, connection:, &block)
|
|
async_invoke receiver, :instance_exec, *args, connection: connection, &block
|
|
end
|
|
|
|
def async_invoke(receiver, method, *args, connection: receiver, &block)
|
|
@executor.post do
|
|
invoke(receiver, method, *args, connection: connection, &block)
|
|
end
|
|
end
|
|
|
|
def invoke(receiver, method, *args, connection:, &block)
|
|
work(connection) do
|
|
receiver.send method, *args, &block
|
|
rescue Exception => e
|
|
logger.error "There was an exception - #{e.class}(#{e.message})"
|
|
logger.error e.backtrace.join("\n")
|
|
|
|
receiver.handle_exception if receiver.respond_to?(:handle_exception)
|
|
end
|
|
end
|
|
|
|
private
|
|
def logger
|
|
ActionCable.server.logger
|
|
end
|
|
end
|
|
end
|
|
end
|