Ruby 1.8.7 support.

OutOfOrder · May 18, 2011 · 5b24a72 · 5b24a72
1 parent e6369c4
commit 5b24a72
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/multidb/balancer.rb b/lib/multidb/balancer.rb
@@ -38,7 +38,8 @@ def initialize(configuration)
     def get(name, &block)
       candidates = @candidates[name] || []
       raise ArgumentError, "No such database connection '#{name}'" if candidates.blank?
-      candidate = candidates.sample
+      candidate = candidates.respond_to?(:sample) ? 
+        candidates.sample : candidates[rand(candidates.length)]
       block_given? ? yield(candidate) : candidate
     end