Class: ActionCable::Server::Worker

Inherits:
Object
  • Object
show all
Includes:
ActiveRecordConnectionManagement, ActiveSupport::Callbacks
Defined in:
lib/action_cable/server/worker.rb,
lib/action_cable/server/worker/active_record_connection_management.rb

Overview

Worker used by Server.send_async to do connection work in threads.

Defined Under Namespace

Modules: ActiveRecordConnectionManagement

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from ActiveRecordConnectionManagement

#with_database_connections

Constructor Details

#initialize(max_size: 5) ⇒ Worker

Returns a new instance of Worker.



19
20
21
22
23
24
25
# File 'lib/action_cable/server/worker.rb', line 19

def initialize(max_size: 5)
  @executor = Concurrent::ThreadPoolExecutor.new(
    min_threads: 1,
    max_threads: max_size,
    max_queue: 0,
  )
end

Instance Attribute Details

#executorObject (readonly)

Returns the value of attribute executor.



17
18
19
# File 'lib/action_cable/server/worker.rb', line 17

def executor
  @executor
end

Instance Method Details

#async_exec(receiver, *args, connection:, &block) ⇒ Object



47
48
49
# File 'lib/action_cable/server/worker.rb', line 47

def async_exec(receiver, *args, connection:, &block)
  async_invoke receiver, :instance_exec, *args, connection: connection, &block
end

#async_invoke(receiver, method, *args, connection: receiver, &block) ⇒ Object



51
52
53
54
55
# File 'lib/action_cable/server/worker.rb', line 51

def async_invoke(receiver, method, *args, connection: receiver, &block)
  @executor.post do
    invoke(receiver, method, *args, connection: connection, &block)
  end
end

#haltObject

Stop processing work: any work that has not already started running will be discarded from the queue



29
30
31
# File 'lib/action_cable/server/worker.rb', line 29

def halt
  @executor.shutdown
end

#invoke(receiver, method, *args, connection:, &block) ⇒ Object



57
58
59
60
61
62
63
64
65
66
# File 'lib/action_cable/server/worker.rb', line 57

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

#stopping?Boolean

Returns:

  • (Boolean)


33
34
35
# File 'lib/action_cable/server/worker.rb', line 33

def stopping?
  @executor.shuttingdown?
end

#work(connection) ⇒ Object



37
38
39
40
41
42
43
44
45
# File 'lib/action_cable/server/worker.rb', line 37

def work(connection)
  self.connection = connection

  run_callbacks :work do
    yield
  end
ensure
  self.connection = nil
end