lib/ridgepole/migration_ext.rb in ridgepole-0.7.3.beta vs lib/ridgepole/migration_ext.rb in ridgepole-0.7.3.beta2

@@ -11,11 +11,11 @@
           result = nil
 
           begin
             self.time_recorder = TimeRecorder.new
             yield
-            result = self.time_recorder.result
+            result = time_recorder.result
           ensure
             self.time_recorder = nil
           end
 
           result
@@ -23,22 +23,22 @@
       end
     end
 
     def write(text = '')
       logger = Ridgepole::Logger.instance
-      logger.info(text) unless self.disable_logging
+      logger.info(text) unless disable_logging
       parse_text(text)
     end
 
     def parse_text(text)
-      return unless self.time_recorder
+      return unless time_recorder
 
       case text
       when /\A--\s+(.+)\Z/
-        self.time_recorder.add_key($1)
+        time_recorder.add_key(Regexp.last_match(1))
       when /\A\s+->\s+(\d+\.\d+)s\Z/
-        self.time_recorder.add_value($1.to_f)
+        time_recorder.add_value(Regexp.last_match(1).to_f)
       end
     end
 
     class TimeRecorder
       attr_reader :result
@@ -50,12 +50,10 @@
       def add_key(key)
         @key = key
       end
 
       def add_value(value)
-        if @key
-          @result[@key] = value
-        end
+        @result[@key] = value if @key
 
         @key = nil
       end
     end
   end