Module: Pwrake::MasterApplication

Included in:
Rake::Application
Defined in:
lib/pwrake/master/master_application.rb

Overview

a mixin for managing Rake application.

Instance Method Summary collapse

Instance Method Details

#invoke_task(task_string) ⇒ Object 



49
50
51
52
53
 
# File 'lib/pwrake/master/master_application.rb', line 49

def invoke_task(task_string)
  name, args = parse_task_string(task_string)
  t = self[name]
  @master.invoke(t,args)
end

#pwrake_options ⇒ Object 



8
9
10
 
# File 'lib/pwrake/master/master_application.rb', line 8

def pwrake_options
  @role.option
end

#run ⇒ Object

Run the Pwrake application.



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
 
# File 'lib/pwrake/master/master_application.rb', line 17

def run
  standard_exception_handling do
    init("pwrake")  # <- parse options here
    @role = @master = Master.new
    t = Time.now
    @master.setup_branches
    load_rakefile
    begin
      Log.debug "init: #{Time.now-t} sec"
      t = Time.now
      top_level
      Log.debug "main: #{Time.now-t} sec"
      t = Time.now
    rescue Exception => e
      # Exit with error message
      m = Log.bt(e)
      if @master.thread
        m += "\nIn branch thread #{@master.thread}:\n "
        m += @master.thread.backtrace.join("\n ")
      end
      Log.fatal m
      $stderr.puts m
      @master.signal_trap("INT")
    ensure
      @failed = @master.finish
      Log.debug "finish: #{Time.now-t} sec"
      Log.info "pwrake elapsed time: #{Time.now-START_TIME} sec"
    end
    Kernel.exit(false) if @failed
  end
end

#standard_rake_options ⇒ Object 



55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
 
# File 'lib/pwrake/master/master_application.rb', line 55

def standard_rake_options
  opts = super
  opts.each_with_index do |a,i|
    if a[0] == '--version'
      a[3] = lambda { |value|
        if defined? RAKEVERSION
          puts "rake, version #{RAKEVERSION}"
        elsif defined? Rake::VERSION
          puts "rake, version #{Rake::VERSION}"
        end
        puts "pwrake, version #{Pwrake::VERSION}"
        exit
      }
    end
  end

  opts.concat(
  [
   ['-F', '--hostfile FILE',
    "[Pw] Read hostnames from FILE",
    lambda { |value|
      options.hostfile = value
    }
   ],
   ['-j', '--jobs [N]',
    "[Pw] Number of threads at localhost (default: # of processors)",
    lambda { |value|
      if value
        if /^[+-]?\d+$/ =~ value
          options.num_threads = value.to_i
        else
          raise ArgumentError,"Invalid argument for -j: #{value}"
        end
      else
        options.num_threads = 0
      end
    }
   ],
   ['-L', '--log', '--log-dir [DIRECTORY]', "[Pw] Write log to DIRECTORY",
    lambda { |value|
      if value.kind_of? String
        options.log_dir = value
      else
        options.log_dir = ""
      end
    }
   ],
   ['--ssh-opt', '--ssh-option OPTION', "[Pw] Option passed to SSH",
    lambda { |value|
      options.ssh_option = value
    }
   ],
   ['--filesystem FILESYSTEM', "[Pw] Specify FILESYSTEM (nfs|gfarm)",
    lambda { |value|
      options.filesystem = value
    }
   ],
   ['--gfarm', "[Pw] FILESYSTEM=gfarm",
    lambda { |value|
      options.filesystem = "gfarm"
    }
   ],
   ['-A', '--disable-affinity', "[Pw] Turn OFF affinity (AFFINITY=off)",
    lambda { |value|
      options.disable_affinity = true
    }
   ],
   ['-S', '--disable-steal', "[Pw] Turn OFF task steal",
    lambda { |value|
      options.disable_steal = true
    }
   ],
   ['-d', '--debug',
    "[Pw] Output Debug messages",
    lambda { |value|
      options.debug = true
    }
   ],
   ['--pwrake-conf [FILE]',
    "[Pw] Pwrake configuation file in YAML",
    lambda {|value| options.pwrake_conf = value}
   ],
   ['--show-conf','--show-config',
    "[Pw] Show Pwrake configuration options",
    lambda {|value| options.show_conf = true }
   ],
   ['--report LOGDIR',"[Pw] Generate `report.html' (Report of workflow statistics) in LOGDIR and exit.",
    lambda {|value| options.report_dir = value }
   ],
   ['--report-image IMAGE_TYPE',"[Pw] Gnuplot output format (png,jpg,svg etc.) in report.html.",
    lambda {|value| options.report_image = value }
   ],
   ['--clear-gfarm2fs',"[Pw] Clear gfarm2fs mountpoints left after failure.",
     lambda { |value|
       Option.new.clear_gfarm2fs
       exit
     }
   ],


  ])
  opts
end

#task_queue ⇒ Object 



12
13
14
 
# File 'lib/pwrake/master/master_application.rb', line 12

def task_queue
  @role.task_queue
end