1 require 'whitelist_update'
3 class ContainerRequest < ArvadosModel
6 include CommonApiTemplate
7 include WhitelistUpdate
9 serialize :properties, Hash
10 serialize :environment, Hash
11 serialize :mounts, Hash
12 serialize :runtime_constraints, Hash
13 serialize :command, Array
14 serialize :scheduling_parameters, Hash
16 before_validation :fill_field_defaults, :if => :new_record?
17 before_validation :validate_runtime_constraints
18 before_validation :validate_scheduling_parameters
19 before_validation :set_container
20 validates :command, :container_image, :output_path, :cwd, :presence => true
21 validate :validate_state_change
22 validate :validate_change
23 after_save :update_priority
24 after_save :finalize_if_needed
25 before_create :set_requesting_container_uuid
27 api_accessible :user, extend: :common do |t|
29 t.add :container_count
30 t.add :container_count_max
31 t.add :container_image
46 t.add :requesting_container_uuid
47 t.add :runtime_constraints
48 t.add :scheduling_parameters
53 # Supported states for a container request
56 (Uncommitted = 'Uncommitted'),
57 (Committed = 'Committed'),
62 nil => [Uncommitted, Committed],
63 Uncommitted => [Committed],
71 def skip_uuid_read_permission_check
72 # XXX temporary until permissions are sorted out.
73 %w(modified_by_client_uuid container_uuid requesting_container_uuid)
76 def finalize_if_needed
77 if state == Committed && Container.find_by_uuid(container_uuid).final?
85 # Finalize the container request after the container has
90 c = Container.find_by_uuid(container_uuid)
91 ['output', 'log'].each do |out_type|
92 pdh = c.send(out_type)
94 if self.output_name and out_type == 'output'
95 coll_name = self.output_name
97 coll_name = "Container #{out_type} for request #{uuid}"
99 manifest = Collection.unscoped do
100 Collection.where(portable_data_hash: pdh).first.manifest_text
103 coll = Collection.new(owner_uuid: owner_uuid,
104 manifest_text: manifest,
105 portable_data_hash: pdh,
109 'container_request' => uuid,
111 coll.save_with_unique_name!
112 if out_type == 'output'
118 update_attributes!(state: Final, output_uuid: out_coll, log_uuid: log_coll)
121 def self.full_text_searchable_columns
127 def fill_field_defaults
128 self.state ||= Uncommitted
129 self.environment ||= {}
130 self.runtime_constraints ||= {}
133 self.container_count_max ||= Rails.configuration.container_count_max
134 self.scheduling_parameters ||= {}
138 if (container_uuid_changed? and
139 not current_user.andand.is_admin and
140 not container_uuid.nil?)
141 errors.add :container_uuid, "can only be updated to nil."
144 if state_changed? and state == Committed and container_uuid.nil?
145 self.container_uuid = Container.resolve(self).uuid
147 if self.container_uuid != self.container_uuid_was
148 if self.container_count_changed?
149 errors.add :container_count, "cannot be updated directly."
152 self.container_count += 1
157 def validate_runtime_constraints
162 ['keep_cache_ram', false]].each do |k, required|
163 if !required && !runtime_constraints.include?(k)
166 v = runtime_constraints[k]
167 unless (v.is_a?(Integer) && v > 0)
168 errors.add(:runtime_constraints,
169 "[#{k}]=#{v.inspect} must be a positive integer")
175 def validate_scheduling_parameters
176 if self.state == Committed
177 if scheduling_parameters.include? 'partitions' and
178 (!scheduling_parameters['partitions'].is_a?(Array) ||
179 scheduling_parameters['partitions'].reject{|x| !x.is_a?(String)}.size !=
180 scheduling_parameters['partitions'].size)
181 errors.add :scheduling_parameters, "partitions must be an array of strings"
187 permitted = [:owner_uuid]
191 # Permit updating most fields
192 permitted.push :command, :container_count_max,
193 :container_image, :cwd, :description, :environment,
194 :filters, :mounts, :name, :output_path, :priority,
195 :properties, :requesting_container_uuid, :runtime_constraints,
196 :state, :container_uuid, :use_existing, :scheduling_parameters,
200 if container_uuid.nil?
201 errors.add :container_uuid, "has not been resolved to a container."
205 errors.add :priority, "cannot be nil"
208 # Can update priority, container count, name and description
209 permitted.push :priority, :container_count, :container_count_max, :container_uuid,
212 if self.state_changed?
213 # Allow create-and-commit in a single operation.
214 permitted.push :command, :container_image, :cwd, :description, :environment,
215 :filters, :mounts, :name, :output_path, :properties,
216 :requesting_container_uuid, :runtime_constraints,
217 :state, :container_uuid, :use_existing, :scheduling_parameters,
222 if not current_user.andand.is_admin and not (self.name_changed? || self.description_changed?)
223 errors.add :state, "of container request can only be set to Final by system."
226 if self.state_changed? || self.name_changed? || self.description_changed? || self.output_uuid_changed? || self.log_uuid_changed?
227 permitted.push :state, :name, :description, :output_uuid, :log_uuid
229 errors.add :state, "does not allow updates"
233 errors.add :state, "invalid value"
236 check_update_whitelist permitted
240 if self.state_changed? or
241 self.priority_changed? or
242 self.container_uuid_changed?
243 act_as_system_user do
246 [self.container_uuid_was, self.container_uuid].compact).
247 map(&:update_priority!)
252 def set_requesting_container_uuid
253 return !new_record? if self.requesting_container_uuid # already set
255 token_uuid = current_api_client_authorization.andand.uuid
256 container = Container.where('auth_uuid=?', token_uuid).order('created_at desc').first
257 self.requesting_container_uuid = container.uuid if container