Sha256: 17502823a7e43ace6bb4586159f5f6a19c77fabe8c15726820c3ef828aee3b53
Contents?: true
Size: 972 Bytes
Versions: 7
Compression:
Stored size: 972 Bytes
Contents
require 'active_record/migration'
class ActiveRecord::Migration
cattr_accessor :time_recorder
def write_with_logging(text = '')
logger = Ridgepole::Logger.instance
logger.info(text)
parse_text(text)
end
alias_method_chain :write, :logging
def parse_text(text)
return unless self.time_recorder
case text
when /\A--\s+(.+)\Z/
self.time_recorder.add_key($1)
when /\A\s+->\s+(\d+\.\d+)s\Z/
self.time_recorder.add_value($1.to_f)
end
end
def self.record_time
result = nil
begin
self.time_recorder = TimeRecorder.new
yield
result = self.time_recorder.result
ensure
self.time_recorder = nil
end
return result
end
class TimeRecorder
attr_reader :result
def initialize
@result = {}
end
def add_key(key)
@key = key
end
def add_value(value)
if @key
@result[@key] = value
end
@key = nil
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems