Class: Actions::RemoteExecution::RunHostsJob

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

Defined Under Namespace

Classes: CheckOnProxyActions

Instance Method Summary collapse

Instance Method Details

#batch(from, size) ⇒ Object 



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

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

#create_sub_plans ⇒ Object 



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

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 



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

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

#finalize ⇒ Object 



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

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 



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

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

#humanized_input ⇒ Object 



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

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

#humanized_name ⇒ Object 



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

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

#initiate ⇒ Object 



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

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

#job_invocation ⇒ Object 



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

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

#notify_on_failure(_plan) ⇒ Object 



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

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 



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

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 



68
69
70
71
 
# File 'app/lib/actions/remote_execution/run_hosts_job.rb', line 68

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

#plan(job_invocation) ⇒ Object 



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

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 



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

def proxy_batch_size
  input[:proxy_batch_size]
end

#queue ⇒ Object 



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

def queue
  ForemanRemoteExecution::DYNFLOW_QUEUE
end

#rescue_strategy ⇒ Object 



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

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

#run(event = nil) ⇒ Object 



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

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 



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

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 



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

def spawn_plans
  super
ensure
  trigger_remote_batch
end

#total_count ⇒ Object 



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

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 



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

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