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     get_combined(:state)
  79   end
  80
  81   def docker_image
  82     get_combined(:container_image)
  83   end
  84
  85   def runtime_constraints
  86     get_combined(:runtime_constraints)
  87   end
  88
  89   def priority
  90     get_combined(:priority)
  91   end
  92
  93   def log_collection
  94     get_combined(:log)
  95   end
  96
  97   def outputs
  98     items = []
  99     items << get_combined(:output) if get_combined(:output)
 100     items
 101   end
 102
 103   def command
 104     get_combined(:command)
 105   end
 106
 107   def cwd
 108     get_combined(:cwd)
 109   end
 110
 111   def environment
 112     env = get_combined(:environment)
 113     env = nil if env.andand.empty?
 114     env
 115   end
 116
 117   def mounts
 118     mnt = get_combined(:mounts)
 119     mnt = nil if mnt.andand.empty?
 120     mnt
 121   end
 122
 123   def output_path
 124     get_combined(:output_path)
 125   end
 126
 127   def log_object_uuids
 128     [get_combined(:uuid), get(:uuid)].uniq
 129   end
 130
 131   def live_log_lines(limit=2000)
 132     event_types = ["stdout", "stderr", "arv-mount", "crunch-run"]
 133     log_lines = Log.where(event_type: event_types, object_uuid: log_object_uuids).order("id DESC").limit(limit)
 134     log_lines.results.reverse.
 135       flat_map { |log| log.properties[:text].split("\n") rescue [] }
 136   end
 137
 138   def render_log
 139     collection = Collection.find(log_collection) rescue nil
 140     if collection
 141       return {log: collection, partial: 'collections/show_files', locals: {object: collection, no_checkboxes: true}}
 142     end
 143   end
 144
 145   # End combined propeties
 146
 147   protected
 148   def get_combined key
 149     get(key, @container) || get(key, @proxied)
 150   end
 151 end