aboutsummaryrefslogblamecommitdiffstats
path: root/activesupport/lib/active_support/testing/performance/jruby.rb
blob: b347539f13b4c960cf5a3c9e91aba0d16c6a98bb (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11
                        
              
                                                  
 

                    
                      



                                                                             
                                     
                                         
                   
 

                  
                                             
           

                                



                                                       
 
               
                                  
 



                                                                                    
             
           
 
                  
                                  
 
                                                                                                                                    







                                                   
           
 

                                            







                                                                             


             
                    

                     
                 



                             

                                                 
               
           







                             


                                                                                                         
           
 



                                                                                                          

           
                                             


                                                                                                                                 
           
 
                             


                                                                                                                                         


                           


                                                                                                                                                           




           
require 'jruby/profiler'
require 'java'
java_import java.lang.management.ManagementFactory

module ActiveSupport
  module Testing
    module Performance
      DEFAULTS.merge!(
        if ARGV.include?('--benchmark')
          {:metrics => [:wall_time, :user_time, :memory, :gc_runs, :gc_time]}
        else
          { :metrics => [:wall_time],
            :formats => [:flat, :graph] }
        end).freeze

      protected
        def run_gc
          ManagementFactory.memory_mx_bean.gc
        end

      class Profiler < Performer
        def initialize(*args)
          super
          @supported = @metric.is_a?(Metrics::WallTime)
        end

        def run
          return unless @supported

          @total = time_with_block do
            @data = JRuby::Profiler.profile do
              full_profile_options[:runs].to_i.times { run_test(@metric, :profile) }
            end
          end
        end

        def record
          return unless @supported

          klasses = full_profile_options[:formats].map { |f| JRuby::Profiler.const_get("#{f.to_s.camelize}ProfilePrinter") }.compact

          klasses.each do |klass|
            fname = output_filename(klass)
            FileUtils.mkdir_p(File.dirname(fname))
            file = File.open(fname, 'wb') do |file|
              klass.new(@data).printProfile(file)
            end
          end
        end

        protected
          def output_filename(printer_class)
            suffix =
              case printer_class.name.demodulize
                when 'FlatProfilePrinter';  'flat.txt'
                when 'GraphProfilePrinter'; 'graph.txt'
                else printer_class.name.sub(/ProfilePrinter$/, '').underscore
              end

            "#{super()}_#{suffix}"
          end
      end

      module Metrics
        class Base
          def profile
            yield
          end

          protected
            def with_gc_stats
              ManagementFactory.memory_mx_bean.gc
              yield
            end
        end

        class WallTime < Time
          def measure
            super
          end
        end

        class CpuTime < Time
          def measure
            ManagementFactory.thread_mx_bean.get_current_thread_cpu_time / 1000 / 1000 / 1000.0 # seconds
          end
        end

        class UserTime < Time
          def measure
            ManagementFactory.thread_mx_bean.get_current_thread_user_time / 1000 / 1000 / 1000.0 # seconds
          end
        end

        class Memory < DigitalInformationUnit
          def measure
            ManagementFactory.memory_mx_bean.non_heap_memory_usage.used + ManagementFactory.memory_mx_bean.heap_memory_usage.used
          end
        end

        class GcRuns < Amount
          def measure
            ManagementFactory.garbage_collector_mx_beans.inject(0) { |total_runs, current_gc| total_runs += current_gc.collection_count }
          end
        end

        class GcTime < Time
          def measure
            ManagementFactory.garbage_collector_mx_beans.inject(0) { |total_time, current_gc| total_time += current_gc.collection_time } / 1000.0 # seconds
          end
        end
      end
    end
  end
end