Module: GCCounter

Defined in:

lib/steep.rb

Class Method Summary

• .count_objects(title, regexp = /^Steep/, skip: false) ⇒ Object

Class Method Details

.count_objects(title, regexp = /^Steep/, skip: false) ⇒ Object

# File 'lib/steep.rb', line 307

def count_objects(title, regexp = /^Steep/, skip: false)
  if ENV["COUNT_GC_OBJECTS"] && !skip
    unless GC.disable
      GC.start(immediate_sweep: true, immediate_mark: true, full_mark: true)

      begin
        yield
      ensure
        Steep.logger.fatal "===== #{title} ==============================="

        klasses = [] #: Array[Class]

        ObjectSpace.each_object(Class) do |klass|
          if (klass.name || "") =~ regexp
            klasses << klass
          end
        end

        before = {} #: Hash[Class, Integer]

        klasses.each do |klass|
          count = ObjectSpace.each_object(klass).count
          before[klass] = count
        end

        GC.start(immediate_sweep: true, immediate_mark: true, full_mark: true)

        gceds = [] #: Array[[Class, Integer]]

        klasses.each do |klass|
          count = ObjectSpace.each_object(klass).count
          gced = (before[klass] || 0) - count

          gceds << [klass, gced] if gced > 0
        end

        gceds.sort_by! {|_, count| -count }
        gceds.each do |klass, count|
          Steep.logger.fatal { "#{klass.name} => #{count}"}
        end

        GC.enable
      end
    else
      yield
    end
  else
    yield
  end
end