apps/workbench/app/models/container_work_unit.rb

   1 class ContainerWorkUnit < ProxyWorkUnit
   2   attr_accessor :container
   3
   4   def initialize proxied, label
   5     super
   6     if @proxied.is_a?(ContainerRequest)
   7       container_uuid = get(:container_uuid)
   8       if container_uuid
   9         @container = Container.where(uuid: container_uuid).first
  10       end
  11     end
  12   end
  13
  14   def children
  15     return self.my_children if self.my_children
  16
  17     container_uuid = nil
  18     container_uuid = if @proxied.is_a?(Container) then uuid else get(:container_uuid) end
  19
  20     items = []
  21     if container_uuid
  22       reqs = ContainerRequest.where(requesting_container_uuid: container_uuid).results
  23       reqs.each do |cr|
  24         items << cr.work_unit(cr.name || 'this container')
  25       end
  26     end
  27
  28     self.my_children = items
  29   end
  30
  31   def title
  32     "container"
  33   end
  34
  35   def uri
  36     uuid = get(:uuid)
  37
  38     return nil unless uuid
  39
  40     if @proxied.class.respond_to? :table_name
  41       "/#{@proxied.class.table_name}/#{uuid}"
  42     else
  43       resource_class = ArvadosBase.resource_class_for_uuid(uuid)
  44       "#{resource_class.table_name}/#{uuid}" if resource_class
  45     end
  46   end
  47
  48   def can_cancel?
  49     @proxied.is_a?(ContainerRequest) && state_label.in?(["Queued", "Locked", "Running"]) && priority > 0
  50   end
  51
  52   def container_uuid
  53     get(:container_uuid)
  54   end
  55
  56   # For the following properties, use value from the @container if exists
  57   # This applies to a ContainerRequest with container_uuid
  58
  59   def started_at
  60     t = get_combined(:started_at)
  61     t = Time.parse(t) if (t.is_a? String)
  62     t
  63   end
  64
  65   def modified_at
  66     t = get_combined(:modified_at)
  67     t = Time.parse(t) if (t.is_a? String)
  68     t
  69   end
  70
  71   def finished_at
  72     t = get_combined(:finished_at)
  73     t = Time.parse(t) if (t.is_a? String)
  74     t
  75   end
  76
  77   def state_label
  78     ec = exit_code
  79     return "Failed" if (ec && ec != 0)
  80     get_combined(:state)
  81   end
  82
  83   def exit_code
  84     get_combined(:exit_code)
  85   end
  86
  87   def docker_image
  88     get_combined(:container_image)
  89   end
  90
  91   def runtime_constraints
  92     get_combined(:runtime_constraints)
  93   end
  94
  95   def priority
  96     get_combined(:priority)
  97   end
  98
  99   def log_collection
 100     get_combined(:log)
 101   end
 102
 103   def outputs
 104     items = []
 105     items << get_combined(:output) if get_combined(:output)
 106     items
 107   end
 108
 109   def command
 110     get_combined(:command)
 111   end
 112
 113   def cwd
 114     get_combined(:cwd)
 115   end
 116
 117   def environment
 118     env = get_combined(:environment)
 119     env = nil if env.andand.empty?
 120     env
 121   end
 122
 123   def mounts
 124     mnt = get_combined(:mounts)
 125     mnt = nil if mnt.andand.empty?
 126     mnt
 127   end
 128
 129   def output_path
 130     get_combined(:output_path)
 131   end
 132
 133   def log_object_uuids
 134     [get_combined(:uuid), get(:uuid)].uniq
 135   end
 136
 137   def live_log_lines(limit=2000)
 138     event_types = ["stdout", "stderr", "arv-mount", "crunch-run"]
 139     log_lines = Log.where(event_type: event_types, object_uuid: log_object_uuids).order("id DESC").limit(limit)
 140     log_lines.results.reverse.
 141       flat_map { |log| log.properties[:text].split("\n") rescue [] }
 142   end
 143
 144   def render_log
 145     collection = Collection.find(log_collection) rescue nil
 146     if collection
 147       return {log: collection, partial: 'collections/show_files', locals: {object: collection, no_checkboxes: true}}
 148     end
 149   end
 150
 151   def template_uuid
 152     properties = get(:properties)
 153     if properties
 154       properties[:template_uuid]
 155     end
 156   end
 157
 158   # End combined propeties
 159
 160   protected
 161   def get_combined key
 162     get(key, @container) || get(key, @proxied)
 163   end
 164 end