Class: Actions::RemoteExecution::RunHostsJob

Inherits:
ActionWithSubPlans
  • Object
show all
Extended by:
ApipieDSL::Class
Includes:
ObservableAction, Actions::RecurringAction
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 



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

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

Instance Method Details

#batch(from, size) ⇒ Object 



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

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

#create_sub_plans ⇒ Object 



42
43
44
45
46
47
48
49
50
51
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 42

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)
  end
end

#delay(delay_options, job_invocation) ⇒ Object 



21
22
23
24
25
26
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 21

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 



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

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

#finalize ⇒ Object 



80
81
82
83
84
85
86
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 80

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 



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

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

#humanized_input ⇒ Object 



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

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

#humanized_name ⇒ Object 



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

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

#initiate ⇒ Object 



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

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

#job_invocation ⇒ Object 



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

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

#job_invocation_id ⇒ Object 



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

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

#notify_on_failure(_plan) ⇒ Object 



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

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 



88
89
90
91
92
93
94
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 88

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 



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

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

#plan(job_invocation) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 28

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 



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

def proxy_batch_size
  input[:proxy_batch_size]
end

#queue ⇒ Object 



17
18
19
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 17

def queue
  ForemanRemoteExecution::DYNFLOW_QUEUE
end

#rescue_strategy ⇒ Object 



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

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

#run(event = nil) ⇒ Object 



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

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 



133
134
135
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 133

def set_up_concurrency_control(invocation)
  limit_concurrency_level! invocation.concurrency_level unless invocation.concurrency_level.nil?
end

#spawn_plans ⇒ Object 



53
54
55
56
57
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 53

def spawn_plans
  super
ensure
  trigger_remote_batch
end

#total_count ⇒ Object 



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

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 



59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 59

def trigger_remote_batch
  remaining = output[:planned_count] - output[:remote_triggered_count]
  return if remaining.zero?
  batches_ready = remaining / proxy_batch_size
  if concurrency_limit
    count = remaining
  else
    return unless batches_ready > 0
    count = proxy_batch_size * batches_ready
  end
  batches_ready = [1, batches_ready].max

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