decisiv-sharded_database 0.2.0.1 → 0.2.0.2

data/lib/sharded_database/model_with_connection.rb

@@ -0,0 +1,44 @@
+module ShardedDatabase
+  module ModelWithConnection
+    
+    def self.included(klass)
+      klass.extend ClassMethods
+      
+      klass.class_eval do
+        class << self
+          alias_method_chain :find, :connection
+        end
+      end
+    end
+    
+    def self.borrow_connection(requesting_class, target_class, &block)
+      eigen = requesting_class.metaclass
+      eigen.delegate :connection, :to => target_class
+      yield(requesting_class)
+    end
+    
+    
+    module ClassMethods
+      
+      def find_with_connection(*args)
+        if args.last.is_a?(Hash) && connection_arg = args.last.delete(:connection)
+          connection = 
+          case connection_arg
+          when Symbol then send(connection_arg, *args)
+          when Proc then connection_arg.call(*args)
+          else connection_arg
+          end
+          ModelWithConnection.borrow_connection(self, connection) { find_without_connection(*args) }
+        else
+          find_without_connection(*args)
+        end
+      end
+      
+    end
+    
+    module InstanceMethods
+      
+    end
+    
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: decisiv-sharded_database
 version: !ruby/object:Gem::Version 
-  version: 0.2.0.1
+  version: 0.2.0.2
 platform: ruby
 authors: 
 - Brennan Dunn
@@ -29,6 +29,7 @@ files:
 - lib/sharded_database/aggregate.rb
 - lib/sharded_database/aggregate_proxy.rb
 - lib/sharded_database/core_extensions.rb
+- lib/sharded_database/model_with_connection.rb
 has_rdoc: true
 homepage: http://github.com/brennandunn/sharded_database/
 post_install_message: