├── .rspec ├── .github └── dependabot.yml ├── lib ├── beaker-benchmark │ ├── version.rb │ └── helpers.rb └── beaker-benchmark.rb ├── .simplecov ├── spec ├── spec_helper.rb ├── beaker_test_helpers.rb ├── beaker-benchmark │ ├── performance_result_spec.rb │ └── helpers_spec.rb └── helpers.rb ├── acceptance ├── config │ └── nodes │ │ └── vagrant-ubuntu-1404.yml └── tests │ └── measure_perf_on.rb ├── .gitignore ├── MAINTAINERS ├── Gemfile ├── bin └── beaker-benchmark ├── README.md ├── beaker-benchmark.gemspec ├── Rakefile └── LICENSE /.rspec: -------------------------------------------------------------------------------- 1 | --format documentation 2 | --color 3 | --tty 4 | -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- 1 | version: 2 2 | updates: 3 | - package-ecosystem: bundler 4 | directory: "/" 5 | schedule: 6 | interval: daily 7 | time: "13:00" 8 | open-pull-requests-limit: 10 9 | -------------------------------------------------------------------------------- /lib/beaker-benchmark/version.rb: -------------------------------------------------------------------------------- 1 | module Beaker 2 | module DSL 3 | module BeakerBenchmark 4 | 5 | module Version 6 | STRING = '0.0.9' 7 | end 8 | 9 | end 10 | end 11 | end 12 | -------------------------------------------------------------------------------- /.simplecov: -------------------------------------------------------------------------------- 1 | SimpleCov.configure do 2 | add_filter 'spec/' 3 | add_filter 'vendor/' 4 | add_filter do |file| 5 | file.lines_of_code < 10 6 | end 7 | end 8 | 9 | SimpleCov.start if ENV['BEAKER_BENCHMARK_COVERAGE'] 10 | -------------------------------------------------------------------------------- /spec/spec_helper.rb: -------------------------------------------------------------------------------- 1 | require 'simplecov' 2 | require 'beaker' 3 | require 'beaker_test_helpers' 4 | require 'beaker-benchmark' 5 | require 'helpers' 6 | 7 | require 'rspec/its' 8 | 9 | RSpec.configure do |config| 10 | config.include TestFileHelpers 11 | config.include HostHelpers 12 | end 13 | -------------------------------------------------------------------------------- /acceptance/config/nodes/vagrant-ubuntu-1404.yml: -------------------------------------------------------------------------------- 1 | HOSTS: 2 | ubuntu-server-1404-x64: 3 | roles: 4 | - master 5 | platform: ubuntu-14.04-amd64 6 | box: puppetlabs/ubuntu-14.04-64-nocm 7 | box_url: https://vagrantcloud.com/puppetlabs/boxes/ubuntu-14.04-64-nocm 8 | hypervisor: vagrant 9 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | *.swp 2 | log/* 3 | !.gitignore 4 | junit 5 | acceptance-tests 6 | pkg 7 | Gemfile.lock 8 | options.rb 9 | test.cfg 10 | .yardoc 11 | coverage 12 | .bundle 13 | .vendor 14 | _vendor 15 | tmp/ 16 | doc 17 | # JetBrains IDEA 18 | *.iml 19 | .idea/ 20 | # rbenv file 21 | .ruby-version 22 | .ruby-gemset 23 | # Vagrant folder 24 | .vagrant/ 25 | .vagrant_files/ 26 | -------------------------------------------------------------------------------- /MAINTAINERS: -------------------------------------------------------------------------------- 1 | { 2 | "version": 1, 3 | "file_format": "This MAINTAINERS file format is described at https://github.com/puppetlabs/maintainers", 4 | "issues": "https://tickets.puppetlabs.com/browse/QA", 5 | "internal_list": "https://groups.google.com/a/puppet.com/forum/?hl=en#!forum/silver-team", 6 | "people": [ 7 | { 8 | "github": "samwoods1", 9 | "email": "sam.woods@puppet.com", 10 | "name": "Sam Woods" 11 | } 12 | ] 13 | } 14 | -------------------------------------------------------------------------------- /spec/beaker_test_helpers.rb: -------------------------------------------------------------------------------- 1 | # These are specifically to mock Beaker methods necessary for testing 2 | # that will be available during runtime because this is never run separate 3 | # from Beaker itself. 4 | # 5 | # Including Beaker as a dependency would not work as a solution to this issue, 6 | # since that would make a cycle in the dependency graph, at least until 7 | # Beaker 3.0 happens and this is no longer a dependency of Beaker's. 8 | module BeakerTestHelpers 9 | include Beaker::DSL 10 | end 11 | 12 | module Beaker 13 | module DSL 14 | def self.register( helper ) 15 | end 16 | end 17 | end -------------------------------------------------------------------------------- /lib/beaker-benchmark.rb: -------------------------------------------------------------------------------- 1 | require 'stringify-hash' 2 | require 'beaker-benchmark/helpers' 3 | require 'beaker-benchmark/version' 4 | 5 | 6 | module Beaker 7 | module DSL 8 | module BeakerBenchmark 9 | include Beaker::DSL::BeakerBenchmark::Helpers 10 | end 11 | end 12 | end 13 | 14 | 15 | # Boilerplate DSL inclusion mechanism: 16 | # First we register our module with the Beaker DSL 17 | Beaker::DSL.register( Beaker::DSL::BeakerBenchmark ) 18 | # Modules added into a module which has previously been included are not 19 | # retroactively included in the including class. 20 | # 21 | # https://github.com/adrianomitre/retroactive_module_inclusion 22 | Beaker::TestCase.class_eval { include Beaker::DSL } 23 | -------------------------------------------------------------------------------- /acceptance/tests/measure_perf_on.rb: -------------------------------------------------------------------------------- 1 | require 'beaker-benchmark' 2 | require 'fileutils' 3 | 4 | # Acceptance level testing goes into files in the tests directory like this one, 5 | # Each file corresponding to a new test made up of individual testing steps 6 | test_name "measure_perf_on test" do 7 | if Dir.exist?('tmp/atop') 8 | FileUtils.rm_r('tmp/atop') 9 | end 10 | result = measure_perf_on master, 'sleep test' do 11 | on master, 'sleep 10' 12 | end 13 | 14 | assert(File.exist?("tmp/atop/#{@@session_timestamp}/ubuntu-server-1404-x64/atop_log_measure_perf_on_test_sleep_test.log")) 15 | result.log_csv 16 | assert(File.exist?("tmp/atop/#{@@session_timestamp}/ubuntu-server-1404-x64/atop_log_measure_perf_on_test_sleep_test.csv")) 17 | end 18 | -------------------------------------------------------------------------------- /Gemfile: -------------------------------------------------------------------------------- 1 | source ENV['GEM_SOURCE'] || "https://rubygems.org" 2 | 3 | gemspec 4 | 5 | 6 | 7 | def location_for(place, fake_version = nil) 8 | if place =~ /^(git:[^#]*)#(.*)/ 9 | [fake_version, { :git => $1, :branch => $2, :require => false }].compact 10 | elsif place =~ /^file:\/\/(.*)/ 11 | ['>= 0', { :path => File.expand_path($1), :require => false }] 12 | else 13 | [place, { :require => false }] 14 | end 15 | end 16 | 17 | 18 | # We don't put beaker in as a test dependency because we 19 | # don't want to create a transitive dependency 20 | group :acceptance_testing do 21 | gem "beaker", *location_for(ENV['BEAKER_VERSION'] || '~> 3.0') 22 | end 23 | 24 | 25 | if File.exists? "#{__FILE__}.local" 26 | eval(File.read("#{__FILE__}.local"), binding) 27 | end 28 | -------------------------------------------------------------------------------- /bin/beaker-benchmark: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env ruby 2 | 3 | require 'rubygems' unless defined?(Gem) 4 | require 'beaker-benchmark' 5 | 6 | VERSION_STRING = 7 | " 8 | _ .--. 9 | ( ` ) 10 | beaker-benchmark .-' `--, 11 | _..----.. ( )`-. 12 | .'_|` _|` _|( .__, ) 13 | /_| _| _| _( (_, .-' 14 | ;| _| _| _| '-'__,--'`--' 15 | | _| _| _| _| | 16 | _ || _| _| _| _| %s 17 | _( `--.\\_| _| _| _|/ 18 | .-' )--,| _| _|.` 19 | (__, (_ ) )_| _| / 20 | `-.__.\\ _,--'\\|__|__/ 21 | ;____; 22 | \\YT/ 23 | || 24 | |\"\"| 25 | '==' 26 | " 27 | 28 | 29 | 30 | puts VERSION_STRING % [Beaker::DSL::Helpers::BeakerBenchmark::Version::STRING] 31 | 32 | exit 0 33 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # beaker-benchmark 2 | 3 | A tool for monitoring performance on a puppet infrastructure node while performing some actions. 4 | 5 | # Usage 6 | # Monitor performance on the master while pe-puppetserver is starting up 7 | test_name('measure_perf_on_puppetserver_start') { 8 | on(master, 'puppet resource service pe-puppetserver ensure=stopped', true) 9 | measure_perf_on('start_pe-puppetserver', master) { 10 | on(master, 'puppet resource service pe-puppetserver ensure=running') 11 | } 12 | } 13 | 14 | # Example output: 15 | Action: measure_perf_on_puppetserver_start_start_pe-puppetserver, Duration: 37.463595 16 | Avg CPU: 172%, Avg MEM: 1634829, Avg DSK read: 0, Avg DSK Write: 45 17 | Additional output if include_processes: 18 | Process pid: 14067, command: '/opt/puppetlabs/server/apps/postgresql/bin/postgres -D /opt/puppetlabs/server/data/postgresql/9.6/data -c log_directory=/var/log/puppetlabs/postgresql' 19 | Avg CPU: '0', Avg MEM: 48888, Avg DSK read: 0, Avg DSK Write: 48888 20 | -------------------------------------------------------------------------------- /beaker-benchmark.gemspec: -------------------------------------------------------------------------------- 1 | # -*- encoding: utf-8 -*- 2 | $LOAD_PATH.unshift File.expand_path("../lib", __FILE__) 3 | require 'beaker-benchmark/version' 4 | 5 | Gem::Specification.new do |s| 6 | s.name = "beaker-benchmark" 7 | s.version = Beaker::DSL::BeakerBenchmark::Version::STRING 8 | s.authors = ["Puppet"] 9 | s.email = ["team-system-level-validation@puppet.com"] 10 | s.homepage = "https://github.com/puppetlabs/beaker-benchmark" 11 | s.summary = %q{Beaker benchmark Helpers!} 12 | s.description = %q{Used to monitor performance on a puppet infrastructure node in a Beaker test} 13 | s.license = "Apache-2.0" 14 | 15 | s.files = `git ls-files`.split("\n") 16 | s.test_files = `git ls-files -- {test,spec,features}/*`.split("\n") 17 | s.executables = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) } 18 | s.require_paths = ["lib"] 19 | 20 | # Testing dependencies 21 | s.add_development_dependency 'rspec', '~> 3.0' 22 | s.add_development_dependency 'rspec-its' 23 | s.add_development_dependency 'fakefs', '~> 0.6' 24 | s.add_development_dependency 'rake', '~> 10.1' 25 | s.add_development_dependency 'simplecov' 26 | s.add_development_dependency 'pry', '~> 0.10' 27 | 28 | # Documentation dependencies 29 | s.add_development_dependency 'yard' 30 | s.add_development_dependency 'markdown' 31 | s.add_development_dependency 'thin' 32 | 33 | # Run time dependencies 34 | s.add_runtime_dependency 'stringify-hash', '~> 0.0.0' 35 | 36 | end 37 | 38 | -------------------------------------------------------------------------------- /spec/beaker-benchmark/performance_result_spec.rb: -------------------------------------------------------------------------------- 1 | require 'spec_helper' 2 | 3 | class ClassMixedWithDSLHelpers 4 | include BeakerTestHelpers 5 | include Beaker::DSL::BeakerBenchmark::Helpers 6 | 7 | def logger 8 | RSpec::Mocks::Double.new('logger').as_null_object 9 | end 10 | 11 | end 12 | 13 | describe ClassMixedWithDSLHelpers do 14 | subject { Beaker::DSL::BeakerBenchmark::Helpers::PerformanceResult.new ( 15 | {:cpu => [60, 40], :mem => [6000, 4000], :disk_read => [600, 400], :disk_write => [600, 400], :action => 'test_action', :duration => 10, 16 | :processes => {1000 => {:cmd => 'proc1', :cpu_usage => [10, 20], :mem_usage => [1000, 2000], :disk_read => [100, 200], :disk_write => [100, 200]}, 17 | 2000 => {:cmd => 'proc2', :cpu_usage => [20, 40], :mem_usage => [2000, 4000], :disk_read => [200, 400], :disk_write => [200, 400]}}, 18 | :logger => logger, :hostname => 'my_host'}) } 19 | 20 | describe 'initialize' do 21 | 22 | it 'properly averages arrays of values' do 23 | 24 | end 25 | 26 | it 'sets overall averages to 0 if no values are set' do 27 | 28 | end 29 | 30 | it 'raises an exception if :mem, :cpu, :disk_read or :disk_write args do not exist' do 31 | 32 | end 33 | 34 | it 'creates a result without no process data if @processes_to_monitor is empty or nil' do 35 | 36 | end 37 | 38 | it 'raises an exception if :mem, :cpu, :disk_read or :disk_write args do not exist' do 39 | 40 | end 41 | 42 | end 43 | 44 | describe 'log_summary' do 45 | 46 | it 'prints out the results' do 47 | 48 | end 49 | 50 | end 51 | 52 | describe 'log_csv' do 53 | 54 | it 'prints out the results in CSV format' do 55 | file_path = subject.log_csv 56 | file = File.open file_path 57 | csv_file_content = file.read 58 | expected_content = <<-EOS 59 | Action,Duration,Avg CPU,Avg MEM,Avg DSK read,Avg DSK Write 60 | test_action,10,50,5000,500,500 61 | 62 | Process pid,command,Avg CPU,Avg MEM,Avg DSK read,Avg DSK Write 63 | 1000,'proc1',15,1500,150,150 64 | 2000,'proc2',30,3000,300,300 65 | EOS 66 | expect(csv_file_content).to eq(expected_content) 67 | end 68 | 69 | end 70 | 71 | end 72 | -------------------------------------------------------------------------------- /spec/beaker-benchmark/helpers_spec.rb: -------------------------------------------------------------------------------- 1 | require 'spec_helper' 2 | 3 | class ClassMixedWithDSLHelpers 4 | include BeakerTestHelpers 5 | include Beaker::DSL::BeakerBenchmark::Helpers 6 | 7 | def logger 8 | RSpec::Mocks::Double.new('logger').as_null_object 9 | end 10 | 11 | end 12 | 13 | describe ClassMixedWithDSLHelpers do 14 | subject { ClassMixedWithDSLHelpers.new } 15 | 16 | describe 'release conditions' do 17 | 18 | it 'has updated the version number from the original template' do 19 | expect( Beaker::DSL::BeakerBenchmark::Version::STRING ).to_not be === '0.0.1rc0' 20 | end 21 | 22 | it 'has a MAINTAINERS doc' do 23 | expect( File.exist?( 'MAINTAINERS' ) ).to be_truthy 24 | end 25 | 26 | end 27 | 28 | describe '.setup_atop' do 29 | 30 | it 'installs atop on first run' do 31 | 32 | end 33 | 34 | it 'sets up epel repo on el hosts on first run' do 35 | 36 | end 37 | 38 | it 'does not install atop after first run' do 39 | 40 | end 41 | 42 | 43 | 44 | 45 | end 46 | 47 | describe '.measure_perf_on' do 48 | 49 | it 'concatenates test name with action name if available' do 50 | 51 | end 52 | 53 | it 'only uses action name if test name is not available' do 54 | 55 | end 56 | 57 | end 58 | 59 | describe '.start_monitoring' do 60 | 61 | it 'executes atop command' do 62 | 63 | end 64 | 65 | it 'executes atop command with additional arguments if include_processes=true' do 66 | 67 | end 68 | 69 | it 'sets appropriate value for @beaker_benchmark_start' do 70 | 71 | end 72 | 73 | end 74 | 75 | describe '.stop_monitoring' do 76 | 77 | it 'sets duration, if @beaker_benchmark_start has a value' do 78 | 79 | end 80 | 81 | it 'defaults duration to 0, if @beaker_benchmark_start has no value' do 82 | 83 | end 84 | 85 | it 'kills the background atop process' do 86 | 87 | end 88 | 89 | 90 | end 91 | 92 | describe '.set_processes_to_monitor' do 93 | 94 | it 'populates @processes_to_monitor if process_regex matches any running process commands' do 95 | 96 | end 97 | 98 | it 'sets @processes_to_monitor an empty hash if process_regex does not match any running process' do 99 | 100 | end 101 | 102 | it 'sets @processes_to_monitor an empty hash if process_regex is nil' do 103 | 104 | end 105 | 106 | it 'sets @processes_to_monitor to all processes if process_regex is .*' do 107 | 108 | end 109 | 110 | end 111 | 112 | describe '.parse_atop_log' do 113 | 114 | it 'creates new PerformanceResult object with valid values' do 115 | 116 | end 117 | 118 | end 119 | 120 | end 121 | 122 | 123 | -------------------------------------------------------------------------------- /spec/helpers.rb: -------------------------------------------------------------------------------- 1 | module TestFileHelpers 2 | def create_files file_array 3 | file_array.each do |f| 4 | FileUtils.mkdir_p File.dirname(f) 5 | FileUtils.touch f 6 | end 7 | end 8 | 9 | def fog_file_contents 10 | { :default => { :aws_access_key_id => "IMANACCESSKEY", 11 | :aws_secret_access_key => "supersekritkey", 12 | :aix_hypervisor_server => "aix_hypervisor.labs.net", 13 | :aix_hypervisor_username => "aixer", 14 | :aix_hypervisor_keyfile => "/Users/user/.ssh/id_rsa-acceptance", 15 | :solaris_hypervisor_server => "solaris_hypervisor.labs.net", 16 | :solaris_hypervisor_username => "harness", 17 | :solaris_hypervisor_keyfile => "/Users/user/.ssh/id_rsa-old.private", 18 | :solaris_hypervisor_vmpath => "rpoooool/zs", 19 | :solaris_hypervisor_snappaths => ["rpoooool/USER/z0"], 20 | :vsphere_server => "vsphere.labs.net", 21 | :vsphere_username => "vsphere@labs.com", 22 | :vsphere_password => "supersekritpassword"} } 23 | end 24 | 25 | end 26 | 27 | module HostHelpers 28 | HOST_DEFAULTS = { :platform => 'unix', 29 | :snapshot => 'pe', 30 | :box => 'box_name', 31 | :roles => ['agent'], 32 | :snapshot => 'snap', 33 | :ip => 'default.ip.address', 34 | :box => 'default_box_name', 35 | :box_url => 'http://default.box.url', 36 | } 37 | 38 | HOST_NAME = "vm%d" 39 | HOST_SNAPSHOT = "snapshot%d" 40 | HOST_IP = "ip.address.for.%s" 41 | HOST_BOX = "%s_of_my_box" 42 | HOST_BOX_URL = "http://address.for.my.box.%s" 43 | HOST_TEMPLATE = "%s_has_a_template" 44 | 45 | def logger 46 | double( 'logger' ).as_null_object 47 | end 48 | 49 | def make_opts 50 | opts = StringifyHash.new 51 | opts.merge( { :logger => logger, 52 | :host_config => 'sample.config', 53 | :type => nil, 54 | :pooling_api => 'http://vcloud.delivery.puppetlabs.net/', 55 | :datastore => 'instance0', 56 | :folder => 'Delivery/Quality Assurance/Staging/Dynamic', 57 | :resourcepool => 'delivery/Quality Assurance/Staging/Dynamic', 58 | :gce_project => 'beaker-compute', 59 | :gce_keyfile => '/path/to/keyfile.p12', 60 | :gce_password => 'notasecret', 61 | :gce_email => '12345678910@developer.gserviceaccount.com' } ) 62 | end 63 | 64 | def generate_result (name, opts ) 65 | result = double( 'result' ) 66 | stdout = opts.has_key?(:stdout) ? opts[:stdout] : name 67 | stderr = opts.has_key?(:stderr) ? opts[:stderr] : name 68 | exit_code = opts.has_key?(:exit_code) ? opts[:exit_code] : 0 69 | exit_code = [exit_code].flatten 70 | allow( result ).to receive( :stdout ).and_return( stdout ) 71 | allow( result ).to receive( :stderr ).and_return( stderr ) 72 | allow( result ).to receive( :exit_code ).and_return( *exit_code ) 73 | result 74 | end 75 | 76 | def make_host_opts name, opts 77 | make_opts.merge( { 'HOSTS' => { name => opts } } ).merge( opts ) 78 | end 79 | 80 | def make_host name, host_hash 81 | host_hash = StringifyHash.new.merge(HOST_DEFAULTS.merge(host_hash)) 82 | 83 | host = make_opts.merge(host_hash) 84 | 85 | allow(host).to receive( :name ).and_return( name ) 86 | allow(host).to receive( :to_s ).and_return( name ) 87 | allow(host).to receive( :exec ).and_return( generate_result( name, host_hash ) ) 88 | host 89 | end 90 | 91 | def make_hosts preset_opts = {}, amt = 3 92 | hosts = [] 93 | (1..amt).each do |num| 94 | name = HOST_NAME % num 95 | opts = { :snapshot => HOST_SNAPSHOT % num, 96 | :ip => HOST_IP % name, 97 | :template => HOST_TEMPLATE % name, 98 | :box => HOST_BOX % name, 99 | :box_url => HOST_BOX_URL % name }.merge( preset_opts ) 100 | hosts << make_host(name, opts) 101 | end 102 | hosts 103 | end 104 | 105 | def make_instance instance_data = {} 106 | OpenStruct.new instance_data 107 | end 108 | 109 | end 110 | -------------------------------------------------------------------------------- /Rakefile: -------------------------------------------------------------------------------- 1 | require "rspec/core/rake_task" 2 | 3 | namespace :test do 4 | 5 | namespace :spec do 6 | 7 | desc "Run spec tests" 8 | RSpec::Core::RakeTask.new(:run) do |t| 9 | t.rspec_opts = ['--color'] 10 | t.pattern = 'spec/' 11 | end 12 | 13 | desc "Run spec tests with coverage" 14 | RSpec::Core::RakeTask.new(:coverage) do |t| 15 | ENV['BEAKER_BENCHMARK_COVERAGE'] = 'y' 16 | t.rspec_opts = ['--color'] 17 | t.pattern = 'spec/' 18 | end 19 | 20 | end 21 | 22 | namespace :acceptance do 23 | 24 | desc <<-EOS 25 | A quick acceptance test, named because it has no pre-suites to run 26 | EOS 27 | task :quick do 28 | 29 | sh("beaker", 30 | "--hosts", ENV['CONFIG'] || "acceptance/config/nodes/vagrant-ubuntu-1404.yml", 31 | "--tests", "acceptance/tests", 32 | "--log-level", "debug", 33 | "--keyfile", ENV['KEY'] || "#{ENV['HOME']}/.ssh/id_rsa") 34 | end 35 | 36 | end 37 | 38 | end 39 | 40 | # namespace-named default tasks. 41 | # these are the default tasks invoked when only the namespace is referenced. 42 | # they're needed because `task :default` in those blocks doesn't work as expected. 43 | task 'test:spec' => 'test:spec:run' 44 | task 'test:acceptance' => 'test:acceptance:quick' 45 | 46 | # global defaults 47 | task :test => 'test:spec' 48 | task :default => :test 49 | 50 | ########################################################### 51 | # 52 | # Documentation Tasks 53 | # 54 | ########################################################### 55 | DOCS_DAEMON = "yard server --reload --daemon --server thin" 56 | FOREGROUND_SERVER = 'bundle exec yard server --reload --verbose --server thin lib/beaker' 57 | 58 | def running?( cmdline ) 59 | ps = `ps -ef` 60 | found = ps.lines.grep( /#{Regexp.quote( cmdline )}/ ) 61 | if found.length > 1 62 | raise StandardError, "Found multiple YARD Servers. Don't know what to do." 63 | end 64 | 65 | yes = found.empty? ? false : true 66 | return yes, found.first 67 | end 68 | 69 | def pid_from( output ) 70 | output.squeeze(' ').strip.split(' ')[1] 71 | end 72 | 73 | desc 'Start the documentation server in the foreground' 74 | task :docs => 'docs:clear' do 75 | original_dir = Dir.pwd 76 | Dir.chdir( File.expand_path(File.dirname(__FILE__)) ) 77 | sh FOREGROUND_SERVER 78 | Dir.chdir( original_dir ) 79 | end 80 | 81 | namespace :docs do 82 | 83 | desc 'Clear the generated documentation cache' 84 | task :clear do 85 | original_dir = Dir.pwd 86 | Dir.chdir( File.expand_path(File.dirname(__FILE__)) ) 87 | sh 'rm -rf docs' 88 | Dir.chdir( original_dir ) 89 | end 90 | 91 | desc 'Generate static documentation' 92 | task :gen => 'docs:clear' do 93 | original_dir = Dir.pwd 94 | Dir.chdir( File.expand_path(File.dirname(__FILE__)) ) 95 | output = `bundle exec yard doc` 96 | puts output 97 | if output =~ /\[warn\]|\[error\]/ 98 | fail "Errors/Warnings during yard documentation generation" 99 | end 100 | Dir.chdir( original_dir ) 101 | end 102 | 103 | desc 'Run the documentation server in the background, alias `bg`' 104 | task :background => 'docs:clear' do 105 | yes, output = running?( DOCS_DAEMON ) 106 | if yes 107 | puts "Not starting a new YARD Server..." 108 | puts "Found one running with pid #{pid_from( output )}." 109 | else 110 | original_dir = Dir.pwd 111 | Dir.chdir( File.expand_path(File.dirname(__FILE__)) ) 112 | sh "bundle exec #{DOCS_DAEMON}" 113 | Dir.chdir( original_dir ) 114 | end 115 | end 116 | 117 | task(:bg) { Rake::Task['docs:background'].invoke } 118 | 119 | desc 'Check the status of the documentation server' 120 | task :status do 121 | yes, output = running?( DOCS_DAEMON ) 122 | if yes 123 | pid = pid_from( output ) 124 | puts "Found a YARD Server running with pid #{pid}" 125 | else 126 | puts "Could not find a running YARD Server." 127 | end 128 | end 129 | 130 | desc "Stop a running YARD Server" 131 | task :stop do 132 | yes, output = running?( DOCS_DAEMON ) 133 | if yes 134 | pid = pid_from( output ) 135 | puts "Found a YARD Server running with pid #{pid}" 136 | `kill #{pid}` 137 | puts "Stopping..." 138 | yes, output = running?( DOCS_DAEMON ) 139 | if yes 140 | `kill -9 #{pid}` 141 | yes, output = running?( DOCS_DAEMON ) 142 | if yes 143 | puts "Could not Stop Server!" 144 | else 145 | puts "Server stopped." 146 | end 147 | else 148 | puts "Server stopped." 149 | end 150 | else 151 | puts "Could not find a running YARD Server" 152 | end 153 | end 154 | end 155 | 156 | namespace :gem do 157 | # add the Bundler gem tasks (build, clean, clobber, install, release) 158 | require "bundler/gem_tasks" 159 | end 160 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | Apache License 2 | Version 2.0, January 2004 3 | http://www.apache.org/licenses/ 4 | 5 | TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 6 | 7 | 1. Definitions. 8 | 9 | "License" shall mean the terms and conditions for use, reproduction, 10 | and distribution as defined by Sections 1 through 9 of this document. 11 | 12 | "Licensor" shall mean the copyright owner or entity authorized by 13 | the copyright owner that is granting the License. 14 | 15 | "Legal Entity" shall mean the union of the acting entity and all 16 | other entities that control, are controlled by, or are under common 17 | control with that entity. For the purposes of this definition, 18 | "control" means (i) the power, direct or indirect, to cause the 19 | direction or management of such entity, whether by contract or 20 | otherwise, or (ii) ownership of fifty percent (50%) or more of the 21 | outstanding shares, or (iii) beneficial ownership of such entity. 22 | 23 | "You" (or "Your") shall mean an individual or Legal Entity 24 | exercising permissions granted by this License. 25 | 26 | "Source" form shall mean the preferred form for making modifications, 27 | including but not limited to software source code, documentation 28 | source, and configuration files. 29 | 30 | "Object" form shall mean any form resulting from mechanical 31 | transformation or translation of a Source form, including but 32 | not limited to compiled object code, generated documentation, 33 | and conversions to other media types. 34 | 35 | "Work" shall mean the work of authorship, whether in Source or 36 | Object form, made available under the License, as indicated by a 37 | copyright notice that is included in or attached to the work 38 | (an example is provided in the Appendix below). 39 | 40 | "Derivative Works" shall mean any work, whether in Source or Object 41 | form, that is based on (or derived from) the Work and for which the 42 | editorial revisions, annotations, elaborations, or other modifications 43 | represent, as a whole, an original work of authorship. For the purposes 44 | of this License, Derivative Works shall not include works that remain 45 | separable from, or merely link (or bind by name) to the interfaces of, 46 | the Work and Derivative Works thereof. 47 | 48 | "Contribution" shall mean any work of authorship, including 49 | the original version of the Work and any modifications or additions 50 | to that Work or Derivative Works thereof, that is intentionally 51 | submitted to Licensor for inclusion in the Work by the copyright owner 52 | or by an individual or Legal Entity authorized to submit on behalf of 53 | the copyright owner. For the purposes of this definition, "submitted" 54 | means any form of electronic, verbal, or written communication sent 55 | to the Licensor or its representatives, including but not limited to 56 | communication on electronic mailing lists, source code control systems, 57 | and issue tracking systems that are managed by, or on behalf of, the 58 | Licensor for the purpose of discussing and improving the Work, but 59 | excluding communication that is conspicuously marked or otherwise 60 | designated in writing by the copyright owner as "Not a Contribution." 61 | 62 | "Contributor" shall mean Licensor and any individual or Legal Entity 63 | on behalf of whom a Contribution has been received by Licensor and 64 | subsequently incorporated within the Work. 65 | 66 | 2. Grant of Copyright License. Subject to the terms and conditions of 67 | this License, each Contributor hereby grants to You a perpetual, 68 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 69 | copyright license to reproduce, prepare Derivative Works of, 70 | publicly display, publicly perform, sublicense, and distribute the 71 | Work and such Derivative Works in Source or Object form. 72 | 73 | 3. Grant of Patent License. Subject to the terms and conditions of 74 | this License, each Contributor hereby grants to You a perpetual, 75 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 76 | (except as stated in this section) patent license to make, have made, 77 | use, offer to sell, sell, import, and otherwise transfer the Work, 78 | where such license applies only to those patent claims licensable 79 | by such Contributor that are necessarily infringed by their 80 | Contribution(s) alone or by combination of their Contribution(s) 81 | with the Work to which such Contribution(s) was submitted. If You 82 | institute patent litigation against any entity (including a 83 | cross-claim or counterclaim in a lawsuit) alleging that the Work 84 | or a Contribution incorporated within the Work constitutes direct 85 | or contributory patent infringement, then any patent licenses 86 | granted to You under this License for that Work shall terminate 87 | as of the date such litigation is filed. 88 | 89 | 4. Redistribution. You may reproduce and distribute copies of the 90 | Work or Derivative Works thereof in any medium, with or without 91 | modifications, and in Source or Object form, provided that You 92 | meet the following conditions: 93 | 94 | (a) You must give any other recipients of the Work or 95 | Derivative Works a copy of this License; and 96 | 97 | (b) You must cause any modified files to carry prominent notices 98 | stating that You changed the files; and 99 | 100 | (c) You must retain, in the Source form of any Derivative Works 101 | that You distribute, all copyright, patent, trademark, and 102 | attribution notices from the Source form of the Work, 103 | excluding those notices that do not pertain to any part of 104 | the Derivative Works; and 105 | 106 | (d) If the Work includes a "NOTICE" text file as part of its 107 | distribution, then any Derivative Works that You distribute must 108 | include a readable copy of the attribution notices contained 109 | within such NOTICE file, excluding those notices that do not 110 | pertain to any part of the Derivative Works, in at least one 111 | of the following places: within a NOTICE text file distributed 112 | as part of the Derivative Works; within the Source form or 113 | documentation, if provided along with the Derivative Works; or, 114 | within a display generated by the Derivative Works, if and 115 | wherever such third-party notices normally appear. The contents 116 | of the NOTICE file are for informational purposes only and 117 | do not modify the License. You may add Your own attribution 118 | notices within Derivative Works that You distribute, alongside 119 | or as an addendum to the NOTICE text from the Work, provided 120 | that such additional attribution notices cannot be construed 121 | as modifying the License. 122 | 123 | You may add Your own copyright statement to Your modifications and 124 | may provide additional or different license terms and conditions 125 | for use, reproduction, or distribution of Your modifications, or 126 | for any such Derivative Works as a whole, provided Your use, 127 | reproduction, and distribution of the Work otherwise complies with 128 | the conditions stated in this License. 129 | 130 | 5. Submission of Contributions. Unless You explicitly state otherwise, 131 | any Contribution intentionally submitted for inclusion in the Work 132 | by You to the Licensor shall be under the terms and conditions of 133 | this License, without any additional terms or conditions. 134 | Notwithstanding the above, nothing herein shall supersede or modify 135 | the terms of any separate license agreement you may have executed 136 | with Licensor regarding such Contributions. 137 | 138 | 6. Trademarks. This License does not grant permission to use the trade 139 | names, trademarks, service marks, or product names of the Licensor, 140 | except as required for reasonable and customary use in describing the 141 | origin of the Work and reproducing the content of the NOTICE file. 142 | 143 | 7. Disclaimer of Warranty. Unless required by applicable law or 144 | agreed to in writing, Licensor provides the Work (and each 145 | Contributor provides its Contributions) on an "AS IS" BASIS, 146 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or 147 | implied, including, without limitation, any warranties or conditions 148 | of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A 149 | PARTICULAR PURPOSE. You are solely responsible for determining the 150 | appropriateness of using or redistributing the Work and assume any 151 | risks associated with Your exercise of permissions under this License. 152 | 153 | 8. Limitation of Liability. In no event and under no legal theory, 154 | whether in tort (including negligence), contract, or otherwise, 155 | unless required by applicable law (such as deliberate and grossly 156 | negligent acts) or agreed to in writing, shall any Contributor be 157 | liable to You for damages, including any direct, indirect, special, 158 | incidental, or consequential damages of any character arising as a 159 | result of this License or out of the use or inability to use the 160 | Work (including but not limited to damages for loss of goodwill, 161 | work stoppage, computer failure or malfunction, or any and all 162 | other commercial damages or losses), even if such Contributor 163 | has been advised of the possibility of such damages. 164 | 165 | 9. Accepting Warranty or Additional Liability. While redistributing 166 | the Work or Derivative Works thereof, You may choose to offer, 167 | and charge a fee for, acceptance of support, warranty, indemnity, 168 | or other liability obligations and/or rights consistent with this 169 | License. However, in accepting such obligations, You may act only 170 | on Your own behalf and on Your sole responsibility, not on behalf 171 | of any other Contributor, and only if You agree to indemnify, 172 | defend, and hold each Contributor harmless for any liability 173 | incurred by, or claims asserted against, such Contributor by reason 174 | of your accepting any such warranty or additional liability. 175 | 176 | END OF TERMS AND CONDITIONS 177 | 178 | APPENDIX: How to apply the Apache License to your work. 179 | 180 | To apply the Apache License to your work, attach the following 181 | boilerplate notice, with the fields enclosed by brackets "{}" 182 | replaced with your own identifying information. (Don't include 183 | the brackets!) The text should be enclosed in the appropriate 184 | comment syntax for the file format. We also recommend that a 185 | file or class name and description of purpose be included on the 186 | same "printed page" as the copyright notice for easier 187 | identification within third-party archives. 188 | 189 | Copyright {yyyy} {name of copyright owner} 190 | 191 | Licensed under the Apache License, Version 2.0 (the "License"); 192 | you may not use this file except in compliance with the License. 193 | You may obtain a copy of the License at 194 | 195 | http://www.apache.org/licenses/LICENSE-2.0 196 | 197 | Unless required by applicable law or agreed to in writing, software 198 | distributed under the License is distributed on an "AS IS" BASIS, 199 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 200 | See the License for the specific language governing permissions and 201 | limitations under the License. 202 | 203 | -------------------------------------------------------------------------------- /lib/beaker-benchmark/helpers.rb: -------------------------------------------------------------------------------- 1 | require "csv" 2 | require "fileutils" 3 | require "minitar" 4 | require "time" 5 | require "zlib" 6 | 7 | module Beaker 8 | module DSL 9 | module BeakerBenchmark 10 | module Helpers 11 | include Beaker::HostPrebuiltSteps 12 | # Column indexes for atop CSV style output 13 | MEASURE_TYPE_INDEX = 0 14 | 15 | SYSTEM_CPU_INDEX = 8 16 | USR_CPU_INDEX = 9 17 | IOWAIT_CPU_INDEX = 12 18 | IDLE_CPU_INDEX = 11 19 | 20 | MEM_INDEX = 10 21 | 22 | DISK_READ_INDEX = 9 23 | DISK_WRITE_INDEX = 11 24 | 25 | PROC_PID_INDEX = 6 26 | 27 | PROC_CPU_INDEX = 16 28 | 29 | PROC_MEM_INDEX = 11 30 | 31 | PROC_DISK_READ_INDEX = 11 32 | PROC_DISK_WRITE_INDEX = 13 33 | 34 | @@session_timestamp = Time.now.getutc.to_i 35 | 36 | TMP_DIR = "tmp/atop/#{@@session_timestamp}" 37 | 38 | # Example usage: 39 | # test_name('measure_perf_on_puppetserver_start') { 40 | # on(master, 'puppet resource service pe-puppetserver ensure=stopped') 41 | # result = measure_perf_on(master, 'start_pe-puppetserver', true) { 42 | # on(master, 'puppet resource service pe-puppetserver ensure=running') 43 | # # on(master, 'facter fqdn') 44 | # } 45 | # raise("pe-puppetserver restart took longer than 120 seconds: #{result.duration} seconds") if result.duration > 120 46 | # raise("pe-puppetserver restart used an average of more than 1gb of memory: #{result.avg_mem} bytes") if result.avg_mem > 1073741824 47 | # process_cpu = result.processes.select{ |pid, process| process[:cmd] =~ /puppet-server-release.jar/ }.values[0][:avg_cpu] 48 | # raise("pe-puppetserver restart caused pe-puppetserver service to use more than 95% of CPU: #{process_cpu}") if process_cpu > 95 49 | # } 50 | def measure_perf_on(infrastructure_host, action_name, include_processes=false, process_regex='/opt/puppetlabs', &block) 51 | # Append action name to test case name if test name is available 52 | action_name = metadata[:case][:name] + "_#{action_name}" if defined? metadata && metadata[:case] && metadata[:case][:name] 53 | 54 | start_monitoring(infrastructure_host, action_name, include_processes) 55 | 56 | yield 57 | 58 | stop_monitoring(infrastructure_host, include_processes.nil? ? nil : process_regex) 59 | end 60 | 61 | def setup_atop(infrastructure_host) 62 | # Only install atop once per host 63 | unless infrastructure_host.check_for_package('atop') 64 | add_el_extras(infrastructure_host, @options) 65 | infrastructure_host.install_package('atop') 66 | end 67 | end 68 | 69 | def start_monitoring(infrastructure_host, action_name, include_processes=false, sample_interval=1) 70 | raise('Monitoring already in progress, call stop_monitoring before calling start_monitoring a second time') unless @beaker_benchmark_start.nil? 71 | @atop_log_filename = "atop_log_#{action_name.downcase.gsub(/[^a-z0-9]/i, '_')}.log" 72 | @action_name = action_name 73 | setup_atop(infrastructure_host) 74 | additional_args = '' 75 | additional_args = ',PRC,PRM,PRD' if include_processes 76 | atop_cmd = "sh -c 'nohup atop -P CPU,SWP,DSK#{additional_args} -i #{sample_interval} > #{@atop_log_filename} 2>&1 &'" 77 | 78 | on(infrastructure_host, atop_cmd) 79 | @beaker_benchmark_start = Time.now 80 | return @@session_timestamp 81 | end 82 | 83 | def stop_monitoring(infrastructure_host, process_regex='.*') 84 | begin 85 | if defined?@beaker_benchmark_start && !@beaker_benchmark_start.nil? 86 | duration = Time.now - @beaker_benchmark_start 87 | else 88 | raise('No monitoring in progress, call start_monitoring before calling stop_monitoring') 89 | end 90 | 91 | # The atop process sticks around unless killed 92 | # It can also take some time to kill depending on how long it has been running and sampling rate. 93 | retry_on infrastructure_host, 'pkill -15 -f atop', {:max_retries => 3, :retry_interval => 5} 94 | set_processes_to_monitor(infrastructure_host, process_regex) 95 | parse_atop_log(infrastructure_host, duration) 96 | ensure 97 | @beaker_benchmark_start = nil 98 | end 99 | end 100 | 101 | def parse_atop_log(infrastructure_host, duration) 102 | unless infrastructure_host.file_exist?(@atop_log_filename) 103 | raise("atop log does not exist at #{@atop_log_filename}") 104 | end 105 | 106 | log_dir = "#{TMP_DIR}/#{infrastructure_host.hostname}" 107 | FileUtils::mkdir_p log_dir unless Dir.exist? log_dir 108 | 109 | # tar the atop log file before copying to avoid timeouts for large files (e.g. 2 week soak test) 110 | tar_file_name = "#{@atop_log_filename}.tar.gz" 111 | compress_cmd = "tar czf #{tar_file_name} #{@atop_log_filename}" 112 | on infrastructure_host, compress_cmd 113 | 114 | # copy the atop log file 115 | scp_from(infrastructure_host, tar_file_name, log_dir) 116 | 117 | # extract the tar file 118 | tgz = Zlib::GzipReader.new(File.open("#{log_dir}/#{tar_file_name}", "rb")) 119 | Archive::Tar::Minitar.unpack(tgz, log_dir) 120 | 121 | cpu_usage = [] 122 | mem_usage = [] 123 | disk_read = [] 124 | disk_write = [] 125 | 126 | skip = true 127 | CSV.parse(File.read(File.expand_path(@atop_log_filename, log_dir)), { :col_sep => ' ' }) do |row| 128 | #skip the first entry, until the first separator 'SEP'. 129 | measure_type = row[MEASURE_TYPE_INDEX] 130 | if skip 131 | skip = (measure_type != 'SEP') 132 | next 133 | end 134 | case measure_type 135 | when 'CPU' 136 | # system + usr + iowait 137 | cpu_active = row[SYSTEM_CPU_INDEX].to_i + row[USR_CPU_INDEX].to_i + row[IOWAIT_CPU_INDEX].to_i 138 | # active + idle 139 | cpu_total = cpu_active + row[IDLE_CPU_INDEX].to_i 140 | cpu_percent = cpu_active * 100 / cpu_total 141 | cpu_usage.push(cpu_percent) 142 | when 'SWP' 143 | mem_usage.push(row[MEM_INDEX].to_i) 144 | when 'DSK' 145 | disk_read.push(row[DISK_READ_INDEX].to_i) 146 | disk_write.push(row[DISK_WRITE_INDEX].to_i) 147 | when 'PRC' 148 | add_process_measure(:cpu_usage, row[PROC_PID_INDEX], row[PROC_CPU_INDEX].to_i) 149 | when 'PRM' 150 | add_process_measure(:mem_usage, row[PROC_PID_INDEX], row[PROC_MEM_INDEX].to_i) 151 | when 'PRD' 152 | add_process_measure(:disk_read, row[PROC_PID_INDEX], row[PROC_DISK_READ_INDEX].to_i) 153 | add_process_measure(:disk_write, row[PROC_PID_INDEX], row[PROC_DISK_WRITE_INDEX].to_i) 154 | end 155 | end 156 | 157 | # rounding duration to improve results formatting 158 | PerformanceResult.new({ :cpu => cpu_usage, :mem => mem_usage, :disk_read => disk_read, :disk_write => disk_write, :action => @action_name, :duration => duration.round(2), :processes => @processes_to_monitor, :logger => @logger, :hostname => infrastructure_host}) 159 | end 160 | 161 | def set_processes_to_monitor(infrastructure_host, process_regex) 162 | @processes_to_monitor = {} 163 | return unless process_regex 164 | result = on(infrastructure_host, "ps -eo pid,cmd | grep #{process_regex}").output 165 | result.each_line do |line| 166 | # use the PID as key and command with args as value 167 | # also ignore the ps and grep commands executed above. 168 | unless line.include? "grep #{process_regex}" 169 | @processes_to_monitor[line.split(' ').first] = { :cmd => line.split(' ')[1..-1].join(' '), :cpu_usage => [], :mem_usage => [], :disk_read => [], :disk_write => [] } 170 | end 171 | end 172 | end 173 | 174 | def add_process_measure measure_type, pid, value 175 | if @processes_to_monitor.keys.include? pid 176 | @processes_to_monitor[pid][measure_type].push value 177 | end 178 | end 179 | 180 | # Example output: 181 | # Action: measure_perf_on_puppetserver_start_start_pe-puppetserver, Duration: 37.463595 182 | # Avg CPU: 72%, Avg MEM: 1634829, Avg DSK read: 0, Avg DSK Write: 45 183 | # Additional output if include_processes: 184 | # Process pid: 14067, command: '/opt/puppetlabs/server/apps/postgresql/bin/postgres -D /opt/puppetlabs/server/data/postgresql/9.6/data -c log_directory=/var/log/puppetlabs/postgresql' 185 | # Avg CPU: '1', Avg MEM: 48888, Avg DSK Write: 20 186 | class PerformanceResult 187 | attr_accessor :avg_cpu, :avg_mem, :avg_disk_read, :avg_disk_write, :action_name, :duration, :processes, :hostname 188 | def initialize(args) 189 | @avg_cpu = args[:cpu].empty? ? 0 : args[:cpu].inject{ |sum, el| sum + el } / args[:cpu].size 190 | @avg_mem = args[:mem].empty? ? 0 : args[:mem].inject{ |sum, el| sum + el } / args[:mem].size 191 | @avg_disk_read = args[:disk_read].empty? ? 0 : args[:disk_read].inject{ |sum, el| sum + el } / args[:disk_read].size 192 | @avg_disk_write = args[:disk_write].empty? ? 0 : args[:disk_write].inject{ |sum, el| sum + el } / args[:disk_write].size 193 | @action_name = args[:action] 194 | @duration = args[:duration] 195 | @processes = args[:processes] 196 | @logger = args[:logger] 197 | @hostname = args[:hostname] || '' 198 | 199 | 200 | @processes.keys.each do |key| 201 | @processes[key][:avg_cpu] = @processes[key][:cpu_usage].inject{ |sum, el| sum + el } / @processes[key][:cpu_usage].size unless @processes[key][:cpu_usage].empty? 202 | @processes[key][:avg_mem] = @processes[key][:mem_usage].inject{ |sum, el| sum + el } / @processes[key][:mem_usage].size unless @processes[key][:mem_usage].empty? 203 | @processes[key][:avg_disk_read] = @processes[key][:disk_read].inject{ |sum, el| sum + el } / @processes[key][:disk_read].size unless @processes[key][:disk_read].empty? 204 | @processes[key][:avg_disk_write] = @processes[key][:disk_write].inject{ |sum, el| sum + el } / @processes[key][:disk_write].size unless @processes[key][:disk_write].empty? 205 | end if @processes 206 | # TODO: At this point, we need to push these results into bigquery or elasticsearch 207 | # so we can normalize results over time and report on increases in trends 208 | end 209 | 210 | def log_summary 211 | @logger.info "Action: #{@action_name}, Duration: #{@duration}" 212 | @logger.info "Avg CPU: #{@avg_cpu}%, Avg MEM: #{@avg_mem}, Avg DSK read: #{@avg_disk_read}, Avg DSK Write: #{@avg_disk_write}" 213 | @processes.keys.each do |key| 214 | @logger.info "Process pid: #{key}, command: '#{@processes[key][:cmd]}'" 215 | @logger.info " Avg CPU: '#{@processes[key][:avg_cpu]}%', Avg MEM: #{@processes[key][:avg_mem]}, Avg DSK read: #{@processes[key][:avg_disk_read]}, Avg DSK Write: #{@processes[key][:avg_disk_write]}" 216 | end 217 | end 218 | 219 | def log_csv file_path=nil 220 | file_path = file_path || "#{TMP_DIR}/#{@hostname}/atop_log_#{action_name.downcase.gsub(/[^a-z0-9]/i, '_')}.csv" 221 | FileUtils.mkdir_p "#{TMP_DIR}/#{@hostname}/" unless Dir.exist?("#{TMP_DIR}/#{@hostname}/") 222 | file = File.open file_path, 'w' 223 | file.write "Action,Duration,Avg CPU,Avg MEM,Avg DSK read,Avg DSK Write\n" 224 | file.write "#{@action_name},#{@duration},#{@avg_cpu},#{@avg_mem},#{@avg_disk_read},#{@avg_disk_write}\n\n" 225 | file.write "Process pid,command,Avg CPU,Avg MEM,Avg DSK read,Avg DSK Write\n" 226 | @processes.keys.each do |key| 227 | file.write "#{key},'#{@processes[key][:cmd]}',#{@processes[key][:avg_cpu]},#{@processes[key][:avg_mem]},#{@processes[key][:avg_disk_read]},#{@processes[key][:avg_disk_write]}\n" 228 | end 229 | file.close 230 | file.path 231 | end 232 | end 233 | 234 | end 235 | end 236 | end 237 | end 238 | --------------------------------------------------------------------------------