lib/looksist/redis_service.rb in looksist-0.2.4 vs lib/looksist/redis_service.rb in looksist-0.2.5

@@ -34,13 +34,13 @@
     def find_all(entity, ids)
       raise 'Buffer overflow! Increase buffer size' if  ids.length > @buffer_size && @buffer_size != 0
       keys = ids.collect { |id| redis_key(entity, id) }
       missed_keys = cache_op(proc_from { keys.uniq }) { (keys - @cache.keys).uniq }
       unless missed_keys.empty?
-        values = @client.mget *missed_keys
-        cache_op { @cache.merge!(Hash[*missed_keys.zip(values).flatten]) }
+        response = Hash[*missed_keys.zip(@client.mget *missed_keys).flatten]
+        cache_op { @cache.merge!(response) }
       end
-      (cache_op(proc_from { values }) { @cache.mslice(keys) })
+      (cache_op(proc_from { keys.collect {|k| response[k] } }) { @cache.mslice(keys) })
     end
 
     def cache_op(computed = nil)
       if @buffer_size == 0
         return computed.call if computed