Class: MiGA::Daemon

Inherits:
MiGA
  • Object
show all
Defined in:
lib/miga/daemon.rb

Overview

MiGA Daemons handling job submissions.

Constant Summary

Constants included from MiGA

CITATION, VERSION, VERSION_DATE, VERSION_NAME

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from MiGA

CITATION, DEBUG, DEBUG_OFF, DEBUG_ON, DEBUG_TRACE_OFF, DEBUG_TRACE_ON, FULL_VERSION, LONG_VERSION, VERSION, VERSION_DATE, clean_fasta_file, initialized?, #result_files_exist?, root_path, script_path, tabulate

Constructor Details

#initialize(project) ⇒ Daemon

Initialize an unactive daemon for the MiGA::Project project. See #daemon to wake the daemon.



39
40
41
42
43
44
45
46
47
48
 
# File 'lib/miga/daemon.rb', line 39

def initialize(project)
  $_MIGA_DAEMON_LAIR << self
  @project = project
  @runopts = JSON.parse(
    File.read(File.expand_path("daemon/daemon.json", project.path)),
      {:symbolize_names=>true})
  @jobs_to_run = []
  @jobs_running = []
  @loop_i = -1
end

Instance Attribute Details

#jobs_running ⇒ Object (readonly)

Array of jobs currently running.



32
33
34
 
# File 'lib/miga/daemon.rb', line 32

def jobs_running
  @jobs_running
end

#jobs_to_run ⇒ Object (readonly)

Array of jobs next to be executed.



30
31
32
 
# File 'lib/miga/daemon.rb', line 30

def jobs_to_run
  @jobs_to_run
end

#loop_i ⇒ Object (readonly)

Integer indicating the current iteration.



34
35
36
 
# File 'lib/miga/daemon.rb', line 34

def loop_i
  @loop_i
end

#options ⇒ Object (readonly)

Options used to setup the daemon.



28
29
30
 
# File 'lib/miga/daemon.rb', line 28

def options
  @options
end

#project ⇒ Object (readonly)

MiGA::Project in which the daemon is running.



26
27
28
 
# File 'lib/miga/daemon.rb', line 26

def project
  @project
end

Class Method Details

.last_alive(project) ⇒ Object

When was the last time a daemon for the MiGA::Project project was seen active? Returns DateTime.



15
16
17
18
19
 
# File 'lib/miga/daemon.rb', line 15

def self.last_alive(project)
  f = File.expand_path("daemon/alive", project.path)
  return nil unless File.exist? f
  DateTime.parse(File.read(f))
end

Instance Method Details

#check_datasets ⇒ Object

Traverse datasets



150
151
152
153
154
155
156
157
158
159
160
 
# File 'lib/miga/daemon.rb', line 150

def check_datasets
  project.each_dataset do |n, ds|
    if ds.nil?
      say "Warning: Dataset #{n} listed but not loaded, reloading project."
      project.load
    else
      to_run = ds.next_preprocessing(true)
      queue_job(to_run, ds) unless to_run.nil?
    end
  end
end

#check_project ⇒ Object

Check if all reference datasets are pre-processed. If yes, check the project-level tasks



165
166
167
168
169
170
171
172
 
# File 'lib/miga/daemon.rb', line 165

def check_project
  return if project.dataset_names.empty?
  if project.done_preprocessing?(false)
    to_run = project.next_distances(true)
    to_run = project.next_inclade(true) if to_run.nil?
    queue_job(to_run) unless to_run.nil?
  end
end

#daemon(task, opts = []) ⇒ Object

Launches the task with options opts (as command-line arguments). Supported tasks include: start, stop, restart, status.



122
123
124
125
126
127
128
129
 
# File 'lib/miga/daemon.rb', line 122

def daemon(task, opts=[])
  options = default_options
  opts.unshift(task)
  options[:ARGV] = opts
  Daemons.run_proc("MiGA:#{project.name}", options) do
    loop { break unless in_loop }
  end
end

#declare_alive ⇒ Object

Tell the world that you’re alive.



133
134
135
136
137
 
# File 'lib/miga/daemon.rb', line 133

def declare_alive
  f = File.open(File.expand_path("daemon/alive", project.path), "w")
  f.print Time.now.to_s
  f.close
end

#default_options ⇒ Object

Returns Hash containing the default options for the daemon.



59
60
61
62
 
# File 'lib/miga/daemon.rb', line 59

def default_options
  { dir_mode: :normal, dir: File.expand_path("daemon", project.path),
    multiple: false, log_output: true }
end

#flush! ⇒ Object

Remove finished jobs from the internal queue and launch as many as possible respecting #maxjobs.



220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
 
# File 'lib/miga/daemon.rb', line 220

def flush!
  # Check for finished jobs
  @jobs_running.select! do |job|
    r = (job[:ds].nil? ? project : job[:ds]).add_result(job[:job], false)
    say "Completed pid:#{job[:pid]} for #{job[:task_name]}." unless r.nil?
    r.nil?
  end
  # Avoid single datasets hogging resources
  @jobs_to_run.rotate! rand(jobs_to_run.size)
  # Launch as many +jobs_to_run+ as possible
  while jobs_running.size < maxjobs
    break if jobs_to_run.empty?
    launch_job @jobs_to_run.shift
  end
end

#get_job(job, ds = nil) ⇒ Object

Get the taks with key symbol job in dataset ds. For project-wide tasks let ds be nil.



207
208
209
210
211
212
213
214
215
 
# File 'lib/miga/daemon.rb', line 207

def get_job(job, ds=nil)
  (jobs_to_run + jobs_running).find do |j|
    if ds==nil
      j[:ds].nil? and j[:job]==job
    else
      (! j[:ds].nil?) and j[:ds].name==ds.name and j[:job]==job
    end
  end
end

#in_loop ⇒ Object

Run one loop step. Returns a Boolean indicating if the loop should continue.



246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
 
# File 'lib/miga/daemon.rb', line 246

def in_loop
  if loop_i == -1
    say "-----------------------------------"
    say "MiGA:#{project.name} launched."
    say "-----------------------------------"
    @loop_i = 0
  end
  @loop_i += 1
  declare_alive
  project.load
  check_datasets
  check_project
  flush!
  if loop_i==4
    say "Housekeeping for sanity"
    @loop_i = 0
    purge!
  end
  report_status
  sleep(latency)
  if shutdown_when_done? and jobs_running.size+jobs_to_run.size == 0
    say "Nothing else to do, shutting down."
    return false
  end
  true
end

#last_alive ⇒ Object

When was the last time a daemon for the current project was seen active? Returns DateTime.



53
54
55
 
# File 'lib/miga/daemon.rb', line 53

def last_alive
  MiGA::Daemon.last_alive project
end

#latency ⇒ Object

Returns Integer indicating the number of seconds to sleep between checks.



88
 
# File 'lib/miga/daemon.rb', line 88

def latency() runopts(:latency) ; end

#maxjobs ⇒ Object

Returns Integer indicating the maximum number of concurrent jobs to run.



92
 
# File 'lib/miga/daemon.rb', line 92

def maxjobs() runopts(:maxjobs) ; end

#ppn ⇒ Object

Returns Integer indicating the number of CPUs per job.



96
 
# File 'lib/miga/daemon.rb', line 96

def ppn() runopts(:ppn) ; end

#purge! ⇒ Object

Remove dead jobs.



238
239
240
241
242
 
# File 'lib/miga/daemon.rb', line 238

def purge!
  @jobs_running.select! do |job|
    `#{sprintf(runopts(:alive), job[:pid])}`.chomp.to_i == 1
  end
end

#queue_job(job, ds = nil) ⇒ Object

Add the task to the internal queue with symbol key job. If the task is dataset-specific, ds specifies the dataset. To submit jobs to the scheduler (or to bash) see #flush!.



178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
 
# File 'lib/miga/daemon.rb', line 178

def queue_job(job, ds=nil)
  return nil unless get_job(job, ds).nil?
  ds_name = (ds.nil? ? "miga-project" : ds.name)
  say "Queueing ", ds_name, ":#{job}"
  vars = { "PROJECT"=>project.path, "RUNTYPE"=>runopts(:type),
    "CORES"=>ppn, "MIGA"=>MiGA::MiGA.root_path }
  vars["DATASET"] = ds.name unless ds.nil?
  log_dir = File.expand_path("daemon/#{job}", project.path)
  Dir.mkdir(log_dir) unless Dir.exist? log_dir
  task_name = "#{project.metadata[:name][0..9]}:#{job}:#{ds_name}"
  to_run = {ds: ds, job: job, task_name: task_name,
    cmd: sprintf(runopts(:cmd),
      # 1: script
      MiGA::MiGA.script_path(job, miga:vars["MIGA"], project:project),
      # 2: vars
      vars.keys.map { |k|
 sprintf(runopts(:var), k, vars[k]) }.join(runopts(:varsep)),
      # 3: CPUs
      ppn,
      # 4: log file
      File.expand_path("#{ds_name}.log", log_dir),
      # 5: task name
      task_name)}
  @jobs_to_run << to_run
end

#report_status ⇒ Object

Report status in a JSON file.



141
142
143
144
145
146
 
# File 'lib/miga/daemon.rb', line 141

def report_status
  f = File.open(File.expand_path("daemon/status.json", project.path), "w")
  f.print JSON.pretty_generate(
    jobs_running:@jobs_running, jobs_to_run:@jobs_to_run)
  f.close
end

#restart(opts = []) ⇒ Object

Restarts the daemon with opts.



113
 
# File 'lib/miga/daemon.rb', line 113

def restart(opts=[]) daemon("restart", opts) ; end

#runopts(k, v = nil, force = false) ⇒ Object

Set/get #options, where k is the Symbol of the option and v is the value (or nil to use as getter). Skips consistency tests if force. Returns new value.



68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/miga/daemon.rb', line 68

def runopts(k, v=nil, force=false)
  k = k.to_sym
  unless v.nil?
    v = [:latency, :maxjobs, :ppn].include?(k) ? v.to_i :
        [:shutdown_when_done].include?(k) ? !!v : v
    raise "Daemon's #{k} cannot be set to zero." if !force and v==0
    @runopts[k] = v
  end
  if k==:kill and v.nil?
    case @runopts[:type].to_s
      when "bash" ; return "kill -9 '%s'"
      when "qsub" ; return "qdel '%s'"
      else        ; return "canceljob '%s'"
    end
  end
  @runopts[k]
end

#say(*opts) ⇒ Object

Send a datestamped message to the log.



275
276
277
 
# File 'lib/miga/daemon.rb', line 275

def say(*opts)
  print "[#{Time.new.inspect}] ", *opts, "\n"
end

#shutdown_when_done? ⇒ Boolean

Returns Boolean indicating if the daemon should shutdown when processing is complete.

Returns:

  • (Boolean) 


101
 
# File 'lib/miga/daemon.rb', line 101

def shutdown_when_done?() !!runopts(:shutdown_when_done) ; end

#start(opts = []) ⇒ Object

Initializes the daemon with opts.



105
 
# File 'lib/miga/daemon.rb', line 105

def start(opts=[]) daemon("start", opts) ; end

#status(opts = []) ⇒ Object

Returns the status of the daemon with opts.



117
 
# File 'lib/miga/daemon.rb', line 117

def status(opts=[]) daemon("status", opts) ; end

#stop(opts = []) ⇒ Object

Stops the daemon with opts.



109
 
# File 'lib/miga/daemon.rb', line 109

def stop(opts=[]) daemon("stop", opts) ; end

#terminate ⇒ Object

Terminates a daemon.



281
282
283
284
285
286
287
288
289
290
291
 
# File 'lib/miga/daemon.rb', line 281

def terminate
  say "Terminating daemon..."
  report_status
  k = runopts(:kill)
  @jobs_running.each do |i|
    `#{k % i[:pid]}`
    puts "Terminating pid:#{i[:pid]} for #{i[:task_name]}"
  end
  f = File.expand_path("daemon/alive", project.path)
  File.unlink(f) if File.exist? f
end