apps/workbench/app/models/arvados_resource_list.rb

   1 class ArvadosResourceList
   2   include ArvadosApiClientHelper
   3   include Enumerable
   4
   5   def initialize(resource_class)
   6     @resource_class = resource_class
   7   end
   8
   9   def eager(bool=true)
  10     @eager = bool
  11     self
  12   end
  13
  14   def limit(max_results)
  15     @limit = max_results
  16     self
  17   end
  18
  19   def offset(skip)
  20     @offset = skip
  21     self
  22   end
  23
  24   def order(orderby_spec)
  25     @orderby_spec = orderby_spec
  26     self
  27   end
  28
  29   def filter _filters
  30     @filters ||= []
  31     @filters += _filters
  32     self
  33   end
  34
  35   def where(cond)
  36     cond = cond.dup
  37     cond.keys.each do |uuid_key|
  38       if cond[uuid_key] and (cond[uuid_key].is_a? Array or
  39                              cond[uuid_key].is_a? ArvadosBase)
  40         # Coerce cond[uuid_key] to an array of uuid strings.  This
  41         # allows caller the convenience of passing an array of real
  42         # objects and uuids in cond[uuid_key].
  43         if !cond[uuid_key].is_a? Array
  44           cond[uuid_key] = [cond[uuid_key]]
  45         end
  46         cond[uuid_key] = cond[uuid_key].collect do |item|
  47           if item.is_a? ArvadosBase
  48             item.uuid
  49           else
  50             item
  51           end
  52         end
  53       end
  54     end
  55     cond.keys.select { |x| x.match /_kind$/ }.each do |kind_key|
  56       if cond[kind_key].is_a? Class
  57         cond = cond.merge({ kind_key => 'arvados#' + arvados_api_client.class_kind(cond[kind_key]) })
  58       end
  59     end
  60     api_params = {
  61       _method: 'GET',
  62       where: cond
  63     }
  64     api_params[:eager] = '1' if @eager
  65     api_params[:limit] = @limit if @limit
  66     api_params[:offset] = @offset if @offset
  67     api_params[:order] = @orderby_spec if @orderby_spec
  68     api_params[:filters] = @filters if @filters
  69     res = arvados_api_client.api @resource_class, '', api_params
  70     @results = arvados_api_client.unpack_api_response res
  71     self
  72   end
  73
  74   def results
  75     self.where({}) if !@results
  76     @results
  77   end
  78
  79   def results=(r)
  80     @results = r
  81   end
  82
  83   def all
  84     where({})
  85   end
  86
  87   def each(&block)
  88     results.each do |m|
  89       block.call m
  90     end
  91     self
  92   end
  93
  94   def first
  95     results.first
  96   end
  97
  98   def last
  99     results.last
 100   end
 101
 102   def [](*x)
 103     results.send('[]', *x)
 104   end
 105
 106   def |(x)
 107     if x.is_a? Hash
 108       self.to_hash | x
 109     else
 110       results | x.to_ary
 111     end
 112   end
 113
 114   def to_ary
 115     results
 116   end
 117
 118   def to_hash
 119     Hash[results.collect { |x| [x.uuid, x] }]
 120   end
 121
 122   def empty?
 123     results.empty?
 124   end
 125
 126   def items_available
 127     results.items_available if results.respond_to? :items_available
 128   end
 129
 130   def result_limit
 131     results.limit if results.respond_to? :limit
 132   end
 133
 134   def result_offset
 135     results.offset if results.respond_to? :offset
 136   end
 137
 138 end