Merge branch '8876-work-unit' into 8650-container-work-unit

[arvados.git] / apps / workbench / app / models / proxy_work_unit.rb

diff --git a/apps/workbench/app/models/proxy_work_unit.rb b/apps/workbench/app/models/proxy_work_unit.rb
index 502e6d250c151a4279c9e045a693fccb6950ed8c..5d3ecabbcacd642e8d36c748beaac68eb0089f4b 100644 (file)
--- a/apps/workbench/app/models/proxy_work_unit.rb
+++ b/apps/workbench/app/models/proxy_work_unit.rb
@@ -1,6 +1,10 @@
 class ProxyWorkUnit < WorkUnit
+  require 'time'
+
   attr_accessor :lbl
   attr_accessor :proxied
+  attr_accessor :my_children
+  attr_accessor :unreadable_children
 
   def initialize proxied, label
     self.lbl = label
@@ -12,35 +16,43 @@ class ProxyWorkUnit < WorkUnit
   end
 
   def uuid
-    self.proxied[:uuid]
+    get(:uuid)
   end
 
   def modified_by_user_uuid
-    self.proxied[:modified_by_user_uuid]
+    get(:modified_by_user_uuid)
   end
 
   def created_at
-    self.proxied[:created_at]
+    t = get(:created_at)
+    t = Time.parse(t) if (t.andand.class == String)
+    t
   end
 
   def started_at
-    self.proxied[:started_at]
+    t = get(:started_at)
+    t = Time.parse(t) if (t.andand.class == String)
+    t
   end
 
   def finished_at
-    self.proxied[:finished_at]
+    t = get(:finished_at)
+    t = Time.parse(t) if (t.andand.class == String)
+    t
   end
 
   def state_label
-    if ["Running", "RunningOnServer", "RunningOnClient"].include? self.proxied[:state].to_s
-      "running"
+    state = get(:state)
+    if ["Running", "RunningOnServer", "RunningOnClient"].include? state
+      "Running"
     else
-      self.proxied[:state].to_s.downcase
+      state
     end
   end
 
   def state_bootstrap_class
-    case self.proxied[:state]
+    state = get(:state)
+    case state
     when 'Complete'
       'success'
     when 'Failed', 'Cancelled'
@@ -53,36 +65,154 @@ class ProxyWorkUnit < WorkUnit
   end
 
   def success?
-    if self.proxied[:state] == 'Complete'
+    state = get(:state)
+    if state == 'Complete'
       true
-    elsif self.proxied[:state] == 'Failed'
+    elsif state == 'Failed' or state == 'Cancelled'
       false
     else
       nil
     end
   end
 
+  def child_summary
+    done = 0
+    failed = 0
+    todo = 0
+    running = 0
+    children.each do |c|
+      case c.state_label
+      when 'Complete'
+        done = done+1
+      when 'Failed', 'Cancelled'
+        failed = failed+1
+      when 'Running'
+        running = running+1
+      else
+        todo = todo+1
+      end
+    end
+
+    summary = {}
+    summary[:done] = done
+    summary[:failed] = failed
+    summary[:todo] = todo
+    summary[:running] = running
+    summary
+  end
+
+  def child_summary_str
+    summary = child_summary
+    summary_txt = ''
+
+    if state_label == 'Running'
+      done = summary[:done] || 0
+      running = summary[:running] || 0
+      failed = summary[:failed] || 0
+      todo = summary[:todo] || 0
+      total = done + running + failed + todo
+
+      if total > 0
+        summary_txt += "#{summary[:done]} #{'child'.pluralize(summary[:done])} done,"
+        summary_txt += "#{summary[:failed]} failed,"
+        summary_txt += "#{summary[:running]} running,"
+        summary_txt += "#{summary[:todo]} pending"
+      end
+    end
+    summary_txt
+  end
+
+  def progress
+    state = get(:state)
+    if state == 'Complete'
+      return 1.0
+    elsif state == 'Failed' or state == 'Cancelled'
+      return 0.0
+    end
+
+    summary = child_summary
+    return 0.0 if summary.nil?
+
+    done = summary[:done] || 0
+    running = summary[:running] || 0
+    failed = summary[:failed] || 0
+    todo = summary[:todo] || 0
+    total = done + running + failed + todo
+    if total > 0
+      (done+failed).to_f / total
+    else
+      0.0
+    end
+  end
+
   def parameters
-    self.proxied[:script_parameters]
+    get(:script_parameters)
   end
 
-  def script
-    self.proxied[:script]
+  def repository
+    get(:repository)
   end
 
-  def script_repository
-    self.proxied[:repository]
+  def script
+    get(:script)
   end
 
   def script_version
-    self.proxied[:script_version]
+    get(:script_version)
   end
 
   def supplied_script_version
-    self.proxied[:supplied_script_version]
+    get(:supplied_script_version)
   end
 
   def runtime_constraints
-    self.proxied[:runtime_constraints]
+    get(:runtime_constraints)
+  end
+
+  def docker_image
+    get(:docker_image_locator)
+  end
+
+  def nondeterministic
+    get(:nondeterministic)
+  end
+
+  def priority
+    get(:priority)
+  end
+
+  def log_collection
+    get(:log)
+  end
+
+  def output
+    get(:output)
+  end
+
+  def children
+    []
+  end
+
+  def title
+    "work unit"
+  end
+
+  def has_unreadable_children
+    self.unreadable_children
+  end
+
+  def readable?
+    resource_class = ArvadosBase::resource_class_for_uuid(uuid)
+    resource_class.where(uuid: [uuid]).first rescue nil
+  end
+
+  protected
+
+  def get key
+    if self.proxied.respond_to? key
+      self.proxied.send(key)
+    elsif self.proxied.is_a?(Hash)
+      self.proxied[key]
+    end
   end
 end