1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# frozen_string_literal: true
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
attr_reader :executor
def initialize(max_size: 5)
@executor = Concurrent::ThreadPoolExecutor.new(
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
begin
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
end
private
def logger
ActionCable.server.logger
end
end
end
end
|
