memoize for #decorate

sonots · Sep 2, 2015 · a634d3e · a634d3e
1 parent 21db558
commit a634d3e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/sinatra/decorator/decoratable.rb b/lib/sinatra/decorator/decoratable.rb
@@ -2,14 +2,15 @@ module Sinatra
   module Decorator
     module Decoratable
       def decorate(options = {})
+        return @decorator if @decorator
 
         klass = options[:as]
         klass_name = "#{self.class}Decorator"
         klass = klass_name.constantize if klass.nil?
         decorator = klass.new(self, context: self)
 
         yield decorator if block_given?
-        decorator
+        @decorator = decorator
       end
     end # Decoratable
   end # Decorator