Module: Legion::Runner

Defined in:
lib/legion/runner.rb,
lib/legion/runner/log.rb,
lib/legion/runner/status.rb

Defined Under Namespace

Modules: Log, Status

Class Method Summary collapse

Class Method Details

.run(runner_class:, function:, task_id: nil, args: nil, check_subtask: true, generate_task: true, parent_id: nil, master_id: nil, catch_exceptions: false, **opts) ⇒ Object

rubocop:disable Layout/LineLength, Metrics/CyclomaticComplexity, Metrics/ParameterLists



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/legion/runner.rb', line 8

def self.run(runner_class:, function:, task_id: nil, args: nil, check_subtask: true, generate_task: true, parent_id: nil, master_id: nil, catch_exceptions: false, **opts) # rubocop:disable Layout/LineLength, Metrics/CyclomaticComplexity, Metrics/ParameterLists
  runner_class = Kernel.const_get(runner_class) if runner_class.is_a? String

  if task_id.nil? && generate_task
    task_gen = Legion::Runner::Status.generate_task_id(
      function: function,
      runner_class: runner_class,
      parent_id: parent_id, master_id: master_id, task_id: task_id, **opts
    )
    task_id = task_gen[:task_id] unless task_gen.nil?
  end

  args = opts if args.nil?
  args[:task_id] = task_id unless task_id.nil?
  args[:master_id] = master_id unless master_id.nil?
  args[:parent_id] = parent_id unless parent_id.nil?

  # result = Fiber.new { Fiber.yield runner_class.send(function, **args) }
  raise 'No Function defined' if function.nil?

  result = runner_class.send(function, **args)
rescue Legion::Exception::HandledTask
  status = 'task.exception'
  result = { error: {} }
rescue StandardError => e
  runner_class.handle_exception(e,
                                **opts,
                                runner_class:  runner_class,
                                args:          args,
                                function:      function,
                                task_id:       task_id,
                                generate_task: generate_task,
                                check_subtask: check_subtask)
  status = 'task.exception'
  result = { success: false, status: status, error: { message: e.message, backtrace: e.backtrace } }
  raise e unless catch_exceptions
ensure
  status = 'task.completed' if status.nil?
  Legion::Runner::Status.update(task_id: task_id, status: status) unless task_id.nil?
  if check_subtask && status == 'task.completed'
    Legion::Transport::Messages::CheckSubtask.new(runner_class:  runner_class,
                                                  function:      function,
                                                  result:        result,
                                                  original_args: args,
                                                  **opts).publish
  end
  return { success: true, status: status, result: result, task_id: task_id } # rubocop:disable Lint/EnsureReturn
end