+ extract different ids method implementations

floere · Sep 6, 2012 · af5bc74 · af5bc74
1 parent 5b219a5
commit af5bc74
Showing 1 changed file with 83 additions and 75 deletions.
diff --git a/server/lib/picky/backends/redis.rb b/server/lib/picky/backends/redis.rb
@@ -171,89 +171,18 @@ def ids combinations, amount, offset
 
             # Scripting version of #ids.
             #
-            class << self
+            extend Scripting
-              def ids combinations, amount, offset
-                identifiers = combinations.inject([]) do |identifiers, combination|
-                  identifiers << "#{combination.identifier}"
-                end
-
-                # Assume it's using EVALSHA.
-                #
-                begin
-                  if identifiers.size > 1
-                    client.evalsha @@ids_sent_once,
-                                   identifiers.size,
-                                   *identifiers,
-                                   generate_intermediate_result_id,
-                                   offset,
-                                   (offset + amount)
-                  else
-                    client.zrange identifiers.first,
-                                  offset,
-                                  (offset + amount)
-                  end
-                rescue RuntimeError => e
-                  # Make the server have a SHA-1 for the script.
-                  #
-                  @@ids_sent_once = Digest::SHA1.hexdigest @@ids_script
-                  client.eval @@ids_script,
-                              identifiers.size,
-                              *identifiers,
-                              generate_intermediate_result_id,
-                              offset,
-                              (offset + amount)
-                end
-              end
-            end
           else
-            # Non-Scripting version of #ids.
+            extend NonScripting
-            #
-            class << self
-              def ids combinations, amount, offset
-                identifiers = combinations.inject([]) do |identifiers, combination|
-                  identifiers << "#{combination.identifier}"
-                end
-
-                result_id = generate_intermediate_result_id
-
-                # Little optimization.
-                #
-                if identifiers.size > 1
-                  # Intersect and store.
-                  #
-                  intersected = client.zinterstore result_id, identifiers
-
-                  # Return clean and early if there has been no intersection.
-                  #
-                  if intersected.zero?
-                    client.del result_id
-                    return []
-                  end
-
-                  # Get the stored result.
-                  #
-                  results = client.zrange result_id, offset, (offset + amount)
-
-                  # Delete the stored result as it was only for temporary purposes.
-                  #
-                  # Note: I could also not delete it, but that
-                  #       would not be clean at all.
-                  #
-                  client.del result_id
-                else
-                  results = client.zrange identifiers.first, offset, (offset + amount)
-                end
-
-                results
-              end
-            end
           end
           # Call the newly installed version.
           #
           ids combinations, amount, offset
         else
           # Simply super call.
           #
+          # TODO Remove the ids method.
+          #
           super
         end
       end
@@ -278,6 +207,85 @@ def pid
       def generate_intermediate_result_id
         @intermediate_result_id ||= "#{host}:#{pid}:picky:result"
       end
+
+      # Uses Lua scripting on Redis 2.6.
+      #
+      module Scripting
+        def ids combinations, amount, offset
+          identifiers = combinations.inject([]) do |identifiers, combination|
+            identifiers << "#{combination.identifier}"
+          end
+
+          # Assume it's using EVALSHA.
+          #
+          begin
+            if identifiers.size > 1
+              client.evalsha @@ids_sent_once,
+                             identifiers.size,
+                             *identifiers,
+                             generate_intermediate_result_id,
+                             offset,
+                             (offset + amount)
+            else
+              client.zrange identifiers.first,
+                            offset,
+                            (offset + amount)
+            end
+          rescue RuntimeError => e
+            # Make the server have a SHA-1 for the script.
+            #
+            @@ids_sent_once = Digest::SHA1.hexdigest @@ids_script
+            client.eval @@ids_script,
+                        identifiers.size,
+                        *identifiers,
+                        generate_intermediate_result_id,
+                        offset,
+                        (offset + amount)
+          end
+        end
+      end
+
+      # Does not use Lua scripting, < Redis 2.6.
+      #
+      module NonScripting
+        def ids combinations, amount, offset
+          identifiers = combinations.inject([]) do |identifiers, combination|
+            identifiers << "#{combination.identifier}"
+          end
+
+          result_id = generate_intermediate_result_id
+
+          # Little optimization.
+          #
+          if identifiers.size > 1
+            # Intersect and store.
+            #
+            intersected = client.zinterstore result_id, identifiers
+
+            # Return clean and early if there has been no intersection.
+            #
+            if intersected.zero?
+              client.del result_id
+              return []
+            end
+
+            # Get the stored result.
+            #
+            results = client.zrange result_id, offset, (offset + amount)
+
+            # Delete the stored result as it was only for temporary purposes.
+            #
+            # Note: I could also not delete it, but that
+            #       would not be clean at all.
+            #
+            client.del result_id
+          else
+            results = client.zrange identifiers.first, offset, (offset + amount)
+          end
+
+          results
+        end
+      end
 
     end