fix profiler extend system

[pettanr/pettanr.git] / lib / locmare / profiler / column / extend.rb

diff --git a/lib/locmare/profiler/column/extend.rb b/lib/locmare/profiler/column/extend.rb
index 72637c0..6a85914 100644 (file)
--- a/lib/locmare/profiler/column/extend.rb
+++ b/lib/locmare/profiler/column/extend.rb
@@ -2,13 +2,23 @@ module Locmare
   module ProfilerModule
     module ColumnModule
       class Extend < Base
+        attr :columns
+          
+        def initialize profiler, column_manifest
+          super
+          
+          @columns = {}
+          @column_manifest.columns.each {|name, column|
+            @columns[name] = Column.factory self, column
+          }
+        end
         
         def value
-          self.extend_model_name
+          self.resource
         end
         
-        def extend_model_name
-          self.item.boosters[@column.boost_name].model_name
+        def resource
+          self.item.boosters[@column.boost_name].resource
         end
         
         def extend_item