mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
05088b6299
This PR checks all active Action Cable documentation for typos and other fixes. It aims to make sure that when Rails 5 is released, that the Action Cable docs are up to snuff with the other documentation included with Rails. [ci skip]
71 lines
1.8 KiB
Ruby
71 lines
1.8 KiB
Ruby
require 'active_support/callbacks'
|
|
require 'active_support/core_ext/module/attribute_accessors_per_thread'
|
|
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
|
|
|
|
def initialize(max_size: 5)
|
|
@pool = Concurrent::ThreadPoolExecutor.new(
|
|
min_threads: 1,
|
|
max_threads: max_size,
|
|
max_queue: 0,
|
|
)
|
|
end
|
|
|
|
def async_invoke(receiver, method, *args)
|
|
@pool.post do
|
|
invoke(receiver, method, *args)
|
|
end
|
|
end
|
|
|
|
def invoke(receiver, method, *args)
|
|
begin
|
|
self.connection = receiver
|
|
|
|
run_callbacks :work do
|
|
receiver.send method, *args
|
|
end
|
|
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)
|
|
ensure
|
|
self.connection = nil
|
|
end
|
|
end
|
|
|
|
def async_run_periodic_timer(channel, callback)
|
|
@pool.post do
|
|
run_periodic_timer(channel, callback)
|
|
end
|
|
end
|
|
|
|
def run_periodic_timer(channel, callback)
|
|
begin
|
|
self.connection = channel.connection
|
|
|
|
run_callbacks :work do
|
|
callback.respond_to?(:call) ? channel.instance_exec(&callback) : channel.send(callback)
|
|
end
|
|
ensure
|
|
self.connection = nil
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def logger
|
|
ActionCable.server.logger
|
|
end
|
|
end
|
|
end
|
|
end
|