Class: Aikido::Zen::DetachedAgent::Agent

Inherits:
Object
  • Object
show all
Defined in:
lib/aikido/zen/detached_agent/agent.rb

Overview

Agent that runs in forked processes. It communicates with the parent process to dRB calls. It's in charge of schedule and send heartbeats to the parent process, to be later pushed.

heartbeat & polling interval are configured to 10s , because they are connecting with parent process. We want to have the freshest data.

It's possible to use extend Forwardable here for one-line forward calls to the created at runtime by DRbObject, which leads to an ugly warning about private methods after the delegator is bound.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(config: Aikido::Zen.config, worker: Aikido::Zen::Worker.new(config: config), heartbeat_interval: 10, polling_interval: 10, collector: Aikido::Zen.collector) ⇒ Agent

Returns a new instance of Agent.



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/aikido/zen/detached_agent/agent.rb', line 23

def initialize(
  config: Aikido::Zen.config,
  worker: Aikido::Zen::Worker.new(config: config),
  heartbeat_interval: 10,
  polling_interval: 10,
  collector: Aikido::Zen.collector
)
  @config = config
  @worker = worker
  @heartbeat_interval = heartbeat_interval
  @polling_interval = polling_interval

  @collector = collector

  @front_object = DRbObject.new_with_uri(config.expanded_detached_agent_socket_uri)

  @has_forked = false
  schedule_tasks
end

Instance Attribute Details

#worker ⇒ Object (readonly)

Returns the value of attribute worker.



21
22
23
 
# File 'lib/aikido/zen/detached_agent/agent.rb', line 21

def worker
  @worker
end

Instance Method Details

#calculate_rate_limits(request) ⇒ Object 



48
49
50
 
# File 'lib/aikido/zen/detached_agent/agent.rb', line 48

def calculate_rate_limits(request)
  @front_object.calculate_rate_limits(request.route.as_json, request.client_ip, request.actor.as_json)
end

#handle_fork ⇒ Object

Every time a fork occurs (a new child process is created), we need to start a DRb service in a background thread within the child process. This service will manage the connection and handle resource cleanup.



55
56
57
58
59
60
61
62
 
# File 'lib/aikido/zen/detached_agent/agent.rb', line 55

def handle_fork
  @has_forked = true
  DRb.start_service
  # we need to ensure that there are not more jobs in the queue, but
  # we reuse the same object
  @worker.restart
  schedule_tasks
end

#send_collector_events ⇒ Object 



43
44
45
46
 
# File 'lib/aikido/zen/detached_agent/agent.rb', line 43

def send_collector_events
  events_data = @collector.flush_events.map(&:as_json)
  @front_object.send_collector_events(events_data)
end