decisiv-sharded_database 0.1.2 → 0.1.3

data/lib/sharded_database/aggregate.rb

@@ -9,7 +9,7 @@ module ShardedDatabase
       klass.class_eval do
         cattr_accessor :connection_field, :source_class, :foreign_id
         @connection_field = :oem
-        @foreign_id = :foreign_id
+        @foreign_id = :other_id
 
         class << self
           alias_method_chain :find, :raw
@@ -24,6 +24,10 @@ module ShardedDatabase
         @raw ? temporarily_remove(:after_find) { find_without_raw(*args) } : find_without_raw(*args)
       end
 
+      def preserve_attributes(*attrs)
+        @preserved_attributes = attrs.map(&:to_s)
+      end
+
     end
 
     module InstanceMethods
@@ -39,12 +43,16 @@ module ShardedDatabase
 
         metaclass.delegate :connection, :to => @klass
 
+        (self.class.instance_variable_get("@preserved_attributes") || []).each do |attr|
+          metaclass.send :alias_method, "proxy_#{attr}", attr
+        end
+
         class << self
           alias_method :proxy_class, :class
 
           include AggregateProxy
           instance_methods.each do |m|
-            undef_method(m) unless m =~ /^__|proxy_|inspect|foreign_id/
+            undef_method(m) unless m =~ /^__|proxy_|inspect/
           end
         end
         

data/lib/sharded_database.rb

@@ -4,6 +4,6 @@ require 'sharded_database/core_extensions'
 
 module ShardedDatabase
   
-  VERSION = '0.1'
+  VERSION = '0.1.3'
   
 end

data/test/lib/models.rb

@@ -19,6 +19,8 @@ end
 
 class AggregateEstimate < GlobalConnection
   include ShardedDatabase::Aggregate
+  self.foreign_id   = :other_id
+  preserve_attributes :source
   
   def determine_connection
     Connections[source.to_sym]

data/test/lib/test_case.rb

@@ -27,7 +27,7 @@ module ShardedDatabase
         silence do
           connection.create_table :aggregate_estimates, :force => true do |t|
             t.string    :source
-            t.integer   :foreign_id
+            t.integer   :other_id
             t.timestamp :created_at
           end
         end
@@ -62,10 +62,10 @@ module ShardedDatabase
       
       one_item = Class.new(Connection::One) { set_table_name 'items' ; belongs_to(:estimate) }
       one_item.create :name => 'One Test Item', :estimate_id => @one_1.id
-      
-      AggregateEstimate.create :source => 'one', :foreign_id => @one_1.id
-      AggregateEstimate.create :source => 'two', :foreign_id => @two_1.id
-      AggregateEstimate.create :source => 'two', :foreign_id => @two_2.id
+            
+      AggregateEstimate.create :source => 'one', :other_id => @one_1.id
+      AggregateEstimate.create :source => 'two', :other_id => @two_1.id
+      AggregateEstimate.create :source => 'two', :other_id => @two_2.id
     end
     
   end

data/test/sharded_database/instance_test.rb

@@ -30,6 +30,10 @@ class InstanceTest < ShardedDatabase::TestCase
       assert_same_elements @estimate.attributes.keys, Estimate.column_names
     end
     
+    should 'preserve attributes supplied to #preserve_attributes' do
+      assert_equal 'one', @estimate.proxy_source
+    end
+    
   end
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: decisiv-sharded_database
 version: !ruby/object:Gem::Version 
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors: 
 - Brennan Dunn
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-01-11 00:00:00 -08:00
+date: 2009-01-12 00:00:00 -08:00
 default_executable: 
 dependencies: []