[deltacloud-devel] [PATCH 6/6] Added failure codes to task model

[Martyn Taylor]

From: martyntaylor

---
 src/app/models/task.rb                        |   10 ++++++++++
 src/db/migrate/20090831140000_create_tasks.rb |    1 +
 2 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/src/app/models/task.rb b/src/app/models/task.rb
index 5c5bf34..f3aaa63 100644
--- a/src/app/models/task.rb
+++ b/src/app/models/task.rb
@@ -38,6 +38,16 @@ class Task < ActiveRecord::Base
   COMPLETED_STATES = [STATE_FINISHED, STATE_FAILED, STATE_CANCELED]
   WORKING_STATES   = [STATE_QUEUED, STATE_RUNNING, STATE_PAUSED, STATE_PENDING]

+  # Failures Codes
+  FAILURE_PROVIDER_NOT_FOUND = "provider_not_found"
+  FAILURE_PROVIDER_CONTACT_FAILED = "provider_contact_failed"
+  FAILURE_PROVIDER_RETURNED_FAILED = "provider_returned_failed"
+
+  FAILURE_CODES = [FAILURE_PROVIDER_NOT_FOUND, FAILURE_PROVIDER_CONTACT_FAILED, FAILURE_PROVIDER_RETURNED_FAILED]
+
+  validates_inclusion_of :failure_code,
+    :in => FAILURE_CODES + [nil]
+
   validates_inclusion_of :type,
    :in => %w( InstanceTask )

diff --git a/src/db/migrate/20090831140000_create_tasks.rb b/src/db/migrate/20090831140000_create_tasks.rb
index 2c3e0e2..612b0b5 100644
--- a/src/db/migrate/20090831140000_create_tasks.rb
+++ b/src/db/migrate/20090831140000_create_tasks.rb
@@ -32,6 +32,7 @@ class CreateTasks < ActiveRecord::Migration
       t.timestamp :time_started
       t.timestamp :time_ended
       t.text      :message
+      t.string    :failure_code
       t.integer   :lock_version, :default => 0
     end
   end