lib/rflow/components/raw.rb in rflow-1.0.0a1 vs lib/rflow/components/raw.rb in rflow-1.0.0a2

@@ -1,12 +1,19 @@
-require 'rflow/components/raw/extensions'
-
 class RFlow
   module Components
     module Raw
+      module Extensions
+        module RawExtension
+          def self.extended(base_data)
+            base_data.data_object ||= {'raw' => ''}
+          end
 
-      # Load the schemas
+          def raw; data_object['raw']; end
+          def raw=(new_raw); data_object['raw'] = new_raw; end
+        end
+      end
+
       SCHEMA_DIRECTORY = ::File.expand_path(::File.join(::File.dirname(__FILE__), '..', '..', '..', 'schema'))
 
       SCHEMA_FILES = {
         'raw.avsc' => 'RFlow::Message::Data::Raw',
       }
@@ -14,13 +21,10 @@
       SCHEMA_FILES.each do |file_name, data_type_name|
         schema_string = ::File.read(::File.join(SCHEMA_DIRECTORY, file_name))
         RFlow::Configuration.add_available_data_type data_type_name, 'avro', schema_string
       end
 
-      # Load the data extensions
       RFlow::Configuration.add_available_data_extension('RFlow::Message::Data::Raw',
                                                         RFlow::Components::Raw::Extensions::RawExtension)
-
-
     end
   end
 end