lib/tako/repository.rb in tako-0.1.0 vs lib/tako/repository.rb in tako-0.2.0

@@ -1,32 +1,33 @@
 module Tako
   class Repository
     class << self
       def proxy_configs
-        @proxy_configs
+        @proxy_configs ||= {}
       end
 
       def proxy_connections
-        @proxy_connections
+        @proxy_connections ||= {}
       end
 
       def add(shard_name, conf)
-        @proxy_configs ||= {}
-        @proxy_connections ||= {}
-
         shard_name = shard_name.to_sym
-        return if @proxy_configs[shard_name]
+        return if proxy_configs[shard_name]
 
         temporary_class = Class.new(ActiveRecord::Base)
         const_set("TAKO_AR_CLASS_#{shard_name.upcase}", temporary_class)
         temporary_class.establish_connection(conf)
 
-        @proxy_connections[shard_name] = temporary_class.connection
-        @proxy_configs[shard_name] = conf
+        proxy_connections[shard_name] = temporary_class.connection
+        proxy_configs[shard_name] = conf
       end
 
       def shard(shard_name, base = nil)
-        Proxy.new(shard_name, @proxy_connections[shard_name.to_sym], base)
+        Proxy.new(shard_name, proxy_connections[shard_name.to_sym], base)
+      end
+
+      def shard_names
+        proxy_configs.keys
       end
     end
   end
 end