Slight denormalisation but it allows one to visualise by environment

Signed-off-by: Thom May <[EMAIL PROTECTED]>
---
 .../rails/database/002_add_environment_to_host.rb  |   13 +++++++++++++
 lib/puppet/rails/database/schema.rb                |    1 +
 lib/puppet/rails/host.rb                           |    4 ++++
 3 files changed, 18 insertions(+), 0 deletions(-)
 create mode 100644 lib/puppet/rails/database/002_add_environment_to_host.rb

diff --git a/lib/puppet/rails/database/002_add_environment_to_host.rb 
b/lib/puppet/rails/database/002_add_environment_to_host.rb
new file mode 100644
index 0000000..712265a
--- /dev/null
+++ b/lib/puppet/rails/database/002_add_environment_to_host.rb
@@ -0,0 +1,13 @@
+class AddEnvironmentToHost < ActiveRecord::Migration
+    def self.up
+        unless ActiveRecord::Base.connection.columns("hosts").collect {|c| 
c.name}.include?("created_at")
+            add_column :hosts, :environment, :string 
+        end    
+    end
+
+    def self.down
+        unless ActiveRecord::Base.connection.columns("hosts").collect {|c| 
c.name}.include?("created_at")
+            remove_column :hosts, :environment
+        end    
+    end
+end
diff --git a/lib/puppet/rails/database/schema.rb 
b/lib/puppet/rails/database/schema.rb
index d11d91a..75cbbe9 100644
--- a/lib/puppet/rails/database/schema.rb
+++ b/lib/puppet/rails/database/schema.rb
@@ -57,6 +57,7 @@ class Puppet::Rails::Schema
                 create_table :hosts do |t|
                     t.column :name, :string, :null => false
                     t.column :ip, :string
+                    t.column :environment, :string
                     t.column :last_compile, :datetime
                     t.column :last_freshcheck, :datetime
                     t.column :last_report, :datetime
diff --git a/lib/puppet/rails/host.rb b/lib/puppet/rails/host.rb
index 626edaa..a7cbfe9 100644
--- a/lib/puppet/rails/host.rb
+++ b/lib/puppet/rails/host.rb
@@ -43,6 +43,10 @@ class Puppet::Rails::Host < ActiveRecord::Base
                 host.ip = ip
             end
 
+            if env = node.parameters["environment"]
+                host.environment = env
+            end
+
             # Store the facts into the database.
             host.setfacts node.parameters
 
-- 
1.5.4.3


--~--~---------~--~----~------------~-------~--~----~
You received this message because you are subscribed to the Google Groups 
"Puppet Developers" group.
To post to this group, send email to [email protected]
To unsubscribe from this group, send email to [EMAIL PROTECTED]
For more options, visit this group at 
http://groups.google.com/group/puppet-dev?hl=en
-~----------~----~----~----~------~----~------~--~---

Reply via email to