2015-10-16 04:02:22 -04:00
|
|
|
require 'celluloid'
|
|
|
|
require 'active_support/callbacks'
|
|
|
|
|
2015-06-28 14:16:54 -04:00
|
|
|
module ActionCable
|
|
|
|
module Server
|
2015-07-07 17:13:00 -04:00
|
|
|
# Worker used by Server.send_async to do connection work in threads. Only for internal use.
|
2015-06-28 14:16:54 -04:00
|
|
|
class Worker
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
include Celluloid
|
|
|
|
|
2015-10-12 13:33:48 -04:00
|
|
|
attr_reader :connection
|
2015-06-28 14:16:54 -04:00
|
|
|
define_callbacks :work
|
2015-10-12 13:33:48 -04:00
|
|
|
include ActiveRecordConnectionManagement
|
2015-06-28 14:16:54 -04:00
|
|
|
|
|
|
|
def invoke(receiver, method, *args)
|
2015-10-12 13:33:48 -04:00
|
|
|
@connection = receiver
|
|
|
|
|
2015-06-28 14:16:54 -04:00
|
|
|
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)
|
|
|
|
end
|
|
|
|
|
|
|
|
def run_periodic_timer(channel, callback)
|
2015-10-12 13:33:48 -04:00
|
|
|
@connection = channel.connection
|
|
|
|
|
2015-06-28 14:16:54 -04:00
|
|
|
run_callbacks :work do
|
|
|
|
callback.respond_to?(:call) ? channel.instance_exec(&callback) : channel.send(callback)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
def logger
|
2015-07-06 13:19:18 -04:00
|
|
|
ActionCable.server.logger
|
2015-06-28 14:16:54 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2015-10-16 04:02:22 -04:00
|
|
|
end
|