dnrops
/
metasploit-framework
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add fix for counter_cache migration to keep from throwing readonly column error.

This commit is contained in:
Joe Vennix 2012-06-01 13:30:43 -05:00
parent ced5b9916e
commit 315d68b6f5
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
data/sql/migrate/20120601152442_add_counter_caches_to_hosts.rb
View File

@ -1,14 +1,15 @@
class AddCounterCachesToHosts < ActiveRecord::Migration
def self.up
add_column :hosts, :note_count, :integer, :default => 0
add_column :hosts, :vuln_count, :integer, :default => 0
add_column :hosts, :service_count, :integer, :default => 0
Mdm::Host.reset_column_information
Mdm::Host.all.each do |h|
h.update_attribute :note_count, h.notes.length
h.update_attribute :vuln_count, h.vulns.length
h.update_attribute :service_count, h.services.length
Mdm::Host.reset_counters h.id, :notes
Mdm::Host.reset_counters h.id, :vulns
Mdm::Host.reset_counters h.id, :services
end
end
@ -17,4 +18,4 @@ class AddCounterCachesToHosts < ActiveRecord::Migration
remove_column :hosts, :vuln_count
remove_column :hosts, :service_count
end
end
end