Change case statement to be a Hash lookup

digideskio · Dec 14, 2013 · 6744cab · 6744cab
1 parent 101e358
commit 6744cab
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/lib/adamantium/freezer.rb b/lib/adamantium/freezer.rb
@@ -94,6 +94,12 @@ class UnknownFreezerError < RuntimeError; end
     # Error raised when memoizer options contain unknown keys
     class OptionError < RuntimeError; end
 
+    @freezers = {
+      noop: Noop,
+      deep: Deep,
+      flat: Flat,
+    }.freeze
+
     # Return freezer for name
     #
     # @param [Symbol] name
@@ -103,11 +109,7 @@ class OptionError < RuntimeError; end
     #
     # @api private
     def self.get(name)
-      case name
-      when :noop then Noop
-      when :deep then Deep
-      when :flat then Flat
-      else
+      @freezers.fetch(name) do
         fail UnknownFreezerError, "Freezer with name #{name.inspect} is unknown"
       end
     end