Class: Burstflow::Workflow::Builder

Inherits:
Object
  • Object
show all
Defined in:
lib/burstflow/workflow/builder.rb

Instance Method Summary collapse

Constructor Details

#initialize(workflow, *args, &block) ⇒ Builder

Returns a new instance of Builder.



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/burstflow/workflow/builder.rb', line 4

def initialize workflow, *args, &block
  @workflow = workflow
  @deps = []
  @jobs_by_class = {}
  @jobs_by_id = {}

  @workflow.jobs_config.each_pair do |id, job_hash|
    job = Burstflow::Job.from_hash(@workflow, job_hash)

    @jobs_by_class[job.klass.to_s] ||= []
    @jobs_by_class[job.klass.to_s] << job
    @jobs_by_id[id] = job
    job.incoming.each do |from|
      @deps << { from: from, to: id }
    end

    job.outgoing.each do |to|
      @deps << { from: id, to: to }
    end

    @deps.uniq!
  end

  instance_exec *args, &block
  resolve_dependencies
end

Instance Method Details

#as_json ⇒ Object 



84
85
86
87
88
 
# File 'lib/burstflow/workflow/builder.rb', line 84

def as_json
  @jobs_by_id.each_with_object({}) do |(id, job), json|
    json[job.id] = job.as_json
  end
end

#find_job(id_or_klass) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/burstflow/workflow/builder.rb', line 56

def find_job id_or_klass
  id = if @jobs_by_id.key?(id_or_klass)
    id_or_klass
  else
    jobs = @jobs_by_class[id_or_klass.to_s]

    raise "No job with #{id_or_klass.to_s} klass or id found" if jobs.count == 0
    raise "Duplicated jobs with #{id_or_klass.to_s} klass or id detected" if jobs.count > 1

    jobs.first.id
  end

  @jobs_by_id[id]
end

#resolve_dependencies ⇒ Object 



71
72
73
74
75
76
77
78
79
80
81
82
 
# File 'lib/burstflow/workflow/builder.rb', line 71

def resolve_dependencies
  @deps.each do |dependency|
    from = find_job(dependency[:from].to_s)
    to   = find_job(dependency[:to].to_s)

    to.incoming << from.id
    from.outgoing << to.id

    to.incoming.uniq!
    from.outgoing.uniq!
  end
end

#run(klass, opts = {}) ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/burstflow/workflow/builder.rb', line 31

def run(klass, opts = {})
  opts = opts.with_indifferent_access

  before_deps = opts.delete(:before) || []
  after_deps = opts.delete(:after) || []

  job = klass.new(@workflow, opts)

  [*before_deps].each do |dep|
    @deps << { from: job.id, to: dep.to_s }
  end

  [*after_deps].each do |dep|
    @deps << { from: dep.to_s, to: job.id }
  end

  @jobs_by_class[klass.to_s] ||= []
  @jobs_by_class[klass.to_s] << job

  raise "Job id duplication" if @jobs_by_id.key?(job.id)
  @jobs_by_id[job.id] = job

  job.id
end