+require 'whitelist_update'
+
class ContainerRequest < ArvadosModel
include HasUuid
include KindAndEtag
include CommonApiTemplate
+ include WhitelistUpdate
serialize :properties, Hash
serialize :environment, Hash
serialize :runtime_constraints, Hash
serialize :command, Array
- before_create :set_state_before_save
- validate :validate_change_permitted
- validate :validate_status
+ before_validation :fill_field_defaults, :if => :new_record?
+ before_validation :set_container
+ validates :command, :container_image, :output_path, :cwd, :presence => true
validate :validate_state_change
+ validate :validate_change
+ after_save :update_priority
api_accessible :user, extend: :common do |t|
t.add :command
(Final = 'Final'),
]
- def set_state_before_save
+ State_transitions = {
+ nil => [Uncommitted, Committed],
+ Uncommitted => [Committed],
+ Committed => [Final]
+ }
+
+ def state_transitions
+ State_transitions
+ end
+
+ def skip_uuid_read_permission_check
+ # XXX temporary until permissions are sorted out.
+ %w(modified_by_client_uuid container_uuid requesting_container_uuid)
+ end
+
+ def container_completed!
+ # may implement retry logic here in the future.
+ self.state = ContainerRequest::Final
+ self.save!
+ end
+
+ protected
+
+ def fill_field_defaults
self.state ||= Uncommitted
+ self.environment ||= {}
+ self.runtime_constraints ||= {}
+ self.mounts ||= {}
+ self.cwd ||= "."
end
- def validate_change_permitted
- if self.changed?
- ok = case self.state
- when nil
- true
- when Uncommitted
- true
- when Committed
- # only allow state and priority to change.
- not (self.command_changed? or
- self.container_count_max_changed? or
- self.container_image_changed? or
- self.container_uuid_changed? or
- self.cwd_changed? or
- self.description_changed? or
- self.environment_changed? or
- self.expires_at_changed? or
- self.filters_changed? or
- self.mounts_changed? or
- self.name_changed? or
- self.output_path_changed? or
- self.properties_changed? or
- self.requesting_container_uuid_changed? or
- self.runtime_constraints_changed?)
- when Final
- false
- else
- false
- end
- if not ok
- errors.add :state, "Invalid update of container request in #{self.state} state"
- end
+ # Create a new container (or find an existing one) to satisfy this
+ # request.
+ def resolve
+ # TODO: resolve symbolic git and keep references to content
+ # addresses.
+ c = act_as_system_user do
+ Container.create!(command: self.command,
+ container_image: self.container_image,
+ cwd: self.cwd,
+ environment: self.environment,
+ mounts: self.mounts,
+ output_path: self.output_path,
+ runtime_constraints: self.runtime_constraints)
end
+ self.container_uuid = c.uuid
end
- def validate_status
- if self.state.in?(States)
- true
- else
- errors.add :state, "#{state.inspect} must be one of: #{States.inspect}"
- false
+ def set_container
+ if (container_uuid_changed? and
+ not current_user.andand.is_admin and
+ not container_uuid.nil?)
+ errors.add :container_uuid, "can only be updated to nil."
+ return false
+ end
+ if state_changed? and state == Committed and container_uuid.nil?
+ resolve
end
end
- def validate_state_change
- ok = true
- if self.state_changed?
- ok = case self.state_was
- when nil
- # Must go to Uncommitted
- self.state == Uncommitted
- when Uncommitted
- # Must go to Committed
- self.state == Committed
- when Committed
- # Must to go Final
- self.state == Final
- when Final
- # Once in a final state, don't permit any more state changes
- false
- else
- # Any other state transition is also invalid
- false
- end
- if not ok
- errors.add :state, "invalid change from #{self.state_was} to #{self.state}"
+ def validate_change
+ permitted = [:owner_uuid]
+
+ case self.state
+ when Uncommitted
+ # Permit updating most fields
+ permitted.push :command, :container_count_max,
+ :container_image, :cwd, :description, :environment,
+ :filters, :mounts, :name, :output_path, :priority,
+ :properties, :requesting_container_uuid, :runtime_constraints,
+ :state, :container_uuid
+
+ when Committed
+ if container_uuid.nil?
+ errors.add :container_uuid, "has not been resolved to a container."
end
+
+ if priority.nil?
+ errors.add :priority, "cannot be nil"
+ end
+
+ # Can update priority, container count.
+ permitted.push :priority, :container_count_max, :container_uuid
+
+ if self.state_changed?
+ # Allow create-and-commit in a single operation.
+ permitted.push :command, :container_image, :cwd, :description, :environment,
+ :filters, :mounts, :name, :output_path, :properties,
+ :requesting_container_uuid, :runtime_constraints,
+ :state, :container_uuid
+ end
+
+ when Final
+ if not current_user.andand.is_admin
+ errors.add :state, "of container request can only be set to Final by system."
+ end
+
+ if self.state_changed?
+ permitted.push :state
+ else
+ errors.add :state, "does not allow updates"
+ end
+
+ else
+ errors.add :state, "invalid value"
end
- ok
+
+ check_update_whitelist permitted
end
+ def update_priority
+ if self.state_changed? or
+ self.priority_changed? or
+ self.container_uuid_changed?
+ act_as_system_user do
+ Container.
+ where('uuid in (?)',
+ [self.container_uuid_was, self.container_uuid].compact).
+ map(&:update_priority!)
+ end
+ end
+ end
end
projects / arvados.git / blobdiff