Class: Actions::RemoteExecution::RunHostsJob

Inherits:
ActionWithSubPlans
  • Object
show all
Extended by:
ApipieDSL::Class
Includes:
ObservableAction, Actions::RecurringAction, Dynflow::Action::WithBulkSubPlans, Dynflow::Action::WithPollingSubPlans
Defined in:
app/lib/actions/remote_execution/run_hosts_job.rb

Defined Under Namespace

Classes: CheckOnProxyActions, Jail

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.event_names ⇒ Object 



160
161
162
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 160

def self.event_names
  super + [event_name_base + '_' + event_name_suffix('running')]
end

Instance Method Details

#batch(from, size) ⇒ Object 



107
108
109
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 107

def batch(from, size)
  hosts.offset(from).limit(size)
end

#create_sub_plans ⇒ Object 



45
46
47
48
49
50
51
52
53
54
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 45

def create_sub_plans
  proxy_selector = RemoteExecutionProxySelector.new

  current_batch.map do |host|
    # composer creates just "pattern" for template_invocations because target is evaluated
    # during actual run (here) so we build template invocations from these patterns
    template_invocation = job_invocation.pattern_template_invocation_for_host(host).deep_clone
    trigger(RunHostJob, job_invocation, host, template_invocation, proxy_selector, { :use_concurrency_control => uses_concurrency_control })
  end
end

#delay(delay_options, job_invocation) ⇒ Object 



24
25
26
27
28
29
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 24

def delay(delay_options, job_invocation)
  task.add_missing_task_groups(job_invocation.task_group)
  job_invocation.targeting.resolve_hosts! if job_invocation.targeting.static? && !job_invocation.targeting.resolved?
  input.update :job_invocation => job_invocation.to_action_input
  super delay_options, job_invocation
end

#emit_event_running(plan) ⇒ Object 



164
165
166
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 164

def emit_event_running(plan)
  emit_event(plan, :running)
end

#finalize ⇒ Object 



75
76
77
78
79
80
81
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 75

def finalize
  job_invocation.password = job_invocation.key_passphrase = job_invocation.effective_user_password = nil
  job_invocation.save!

  Rails.logger.debug "cleaning cache for keys that begin with 'job_invocation_#{job_invocation.id}'"
  Rails.cache.delete_matched(cache_deletion_query(job_invocation.id))
end

#hosts ⇒ Object 



124
125
126
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 124

def hosts
  job_invocation.targeting.hosts.order("#{TargetingHost.table_name}.id")
end

#humanized_input ⇒ Object 



148
149
150
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 148

def humanized_input
  input.fetch(:job_invocation, {}).fetch(:description, '')
end

#humanized_name ⇒ Object 



152
153
154
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 152

def humanized_name
  '%s:' % _(super)
end

#initiate ⇒ Object 



111
112
113
114
115
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 111

def initiate
  output[:host_count] = total_count
  output[:remote_triggered_count] = 0
  super
end

#job_invocation ⇒ Object 



103
104
105
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 103

def job_invocation
  @job_invocation ||= JobInvocation.find(job_invocation_id)
end

#job_invocation_id ⇒ Object 



99
100
101
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 99

def job_invocation_id
  input[:job_invocation_id] || input.fetch(:job_invocation, {})[:id]
end

#notify_on_failure(_plan) ⇒ Object 



91
92
93
94
95
96
97
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 91

def notify_on_failure(_plan)
  job_invocation.build_notification.deliver!

  if [RexMailNotification::FAILED_JOBS, RexMailNotification::ALL_JOBS].include?(mail_notification_preference&.interval)
    RexJobMailer.job_finished(job_invocation, subject: _("REX job has failed - %s") % job_invocation.to_s).deliver_now
  end
end

#notify_on_success(_plan) ⇒ Object 



83
84
85
86
87
88
89
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 83

def notify_on_success(_plan)
  job_invocation.build_notification.deliver!

  if [RexMailNotification::SUCCEEDED_JOBS, RexMailNotification::ALL_JOBS].include?(mail_notification_preference&.interval)
    RexJobMailer.job_finished(job_invocation, subject: _("REX job has succeeded - %s") % job_invocation.to_s).deliver_now
  end
end

#on_planning_finished ⇒ Object 



70
71
72
73
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 70

def on_planning_finished
  plan_event(Actions::TriggerProxyBatch::TriggerLastBatch, nil, step_id: input[:trigger_run_step_id])
  super
end

#plan(job_invocation) ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 31

def plan(job_invocation)
  job_invocation.task_group.save! if job_invocation.task_group.try(:new_record?)
  task.add_missing_task_groups(job_invocation.task_group) if job_invocation.task_group
  action_subject(job_invocation)
  job_invocation.targeting.resolve_hosts! if job_invocation.targeting.dynamic? || !job_invocation.targeting.resolved?
  set_up_concurrency_control job_invocation
  input.update(:job_category => job_invocation.job_category)
  plan_self(:job_invocation_id => job_invocation.id)
  provider = job_invocation.pattern_template_invocations.first&.template&.provider
  input[:proxy_batch_size] ||= provider&.proxy_batch_size || Setting['foreman_tasks_proxy_batch_size']
  trigger_action = plan_action(Actions::TriggerProxyBatch, batch_size: proxy_batch_size, total_count: hosts.count)
  input[:trigger_run_step_id] = trigger_action.run_step_id
end

#proxy_batch_size ⇒ Object 



156
157
158
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 156

def proxy_batch_size
  input[:proxy_batch_size]
end

#queue ⇒ Object 



20
21
22
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 20

def queue
  ForemanRemoteExecution::DYNFLOW_QUEUE
end

#rescue_strategy ⇒ Object 



136
137
138
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 136

def rescue_strategy
  ::Dynflow::Action::Rescue::Skip
end

#run(event = nil) ⇒ Object 



140
141
142
143
144
145
146
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 140

def run(event = nil)
  if event == Dynflow::Action::Skip
    plan_event(Dynflow::Action::Skip, nil, step_id: input[:trigger_run_step_id])
  else
    super
  end
end

#set_up_concurrency_control(invocation) ⇒ Object 



128
129
130
131
132
133
134
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 128

def set_up_concurrency_control(invocation)
  limit_concurrency_level invocation.concurrency_level unless invocation.concurrency_level.nil?
  unless invocation.time_span.nil?
    distribute_over_time(invocation.time_span,
      invocation.targeting.hosts.count)
  end
end

#spawn_plans ⇒ Object 



56
57
58
59
60
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 56

def spawn_plans
  super
ensure
  trigger_remote_batch
end

#total_count ⇒ Object 



117
118
119
120
121
122
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 117

def total_count
  # For compatibility with already existing tasks
  return output[:total_count] || hosts.count unless output.has_key?(:host_count) || task.pending?

  output[:host_count] || hosts.count
end

#trigger_remote_batch ⇒ Object 



62
63
64
65
66
67
68
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 62

def trigger_remote_batch
  batches_ready = (output[:planned_count] - output[:remote_triggered_count]) / proxy_batch_size
  return unless batches_ready > 0

  plan_event(Actions::TriggerProxyBatch::TriggerNextBatch[batches_ready], nil, step_id: input[:trigger_run_step_id])
  output[:remote_triggered_count] += proxy_batch_size * batches_ready
end