lib/tako.rb in tako-0.2.2 vs lib/tako.rb in tako-0.3.0

@@ -12,11 +12,11 @@
   extend MultiShardExecution
 
   class << self
     def shard(shard_name)
       if block_given?
-        Tako::Repository.shard(shard_name).in_proxy do
+        Tako::Repository.create_proxy(shard_name).with_shard do
           yield
         end
       else
         raise "gimme a block!"
       end
@@ -37,15 +37,16 @@
     end
   end
 end
 
 require 'active_record'
-
-ActiveRecord::ConnectionHandling.class_eval do
-  prepend Tako::ActiveRecordExt::ConnectionHandling::Prepend
-end
-
-ActiveRecord::Base.class_eval do
-  extend Tako::ActiveRecordExt::Base::Extend
-end
+require 'tako/active_record_ext/sharded_methods'
+require 'tako/active_record_ext/connection_handling'
+require 'tako/active_record_ext/base'
+require 'tako/active_record_ext/association'
+require 'tako/active_record_ext/collection_association'
+require 'tako/active_record_ext/singular_association'
+require 'tako/active_record_ext/collection_proxy'
+require 'tako/active_record_ext/association_relation'
+require 'tako/active_record_ext/log_subscriber'
 
 require 'tako/railtie' if defined?(::Rails)