diff options
author | SHIBATA Hiroshi <hsbt@ruby-lang.org> | 2014-10-29 15:31:46 +0900 |
---|---|---|
committer | SHIBATA Hiroshi <hsbt@ruby-lang.org> | 2014-10-29 15:31:46 +0900 |
commit | dfce792446411e0648718f884df2a76492d0e1c3 (patch) | |
tree | 00d0f3376f0ce586530251f8eab9c8325b5ec224 /test | |
parent | b3f89ec269012c05aa53884cd105a6513ce2d8eb (diff) | |
download | ruby-openssl-dfce792446411e0648718f884df2a76492d0e1c3.tar.gz |
added helper file for memory profile
Diffstat (limited to 'test')
-rw-r--r-- | test/memory_status.rb | 134 |
1 files changed, 134 insertions, 0 deletions
diff --git a/test/memory_status.rb b/test/memory_status.rb new file mode 100644 index 00000000..5347e34c --- /dev/null +++ b/test/memory_status.rb @@ -0,0 +1,134 @@ +module Memory + keys = [] + vals = [] + + case + when File.exist?(procfile = "/proc/self/status") && (pat = /^Vm(\w+):\s+(\d+)/) =~ File.binread(procfile) + PROC_FILE = procfile + VM_PAT = pat + def self.read_status + IO.foreach(PROC_FILE, encoding: Encoding::ASCII_8BIT) do |l| + yield($1.downcase.intern, $2.to_i * 1024) if VM_PAT =~ l + end + end + + read_status {|k, v| keys << k; vals << v} + + when /mswin|mingw/ =~ RUBY_PLATFORM + begin + require 'fiddle/import' + rescue LoadError + $LOAD_PATH.unshift File.join(File.join(__dir__, '..'), 'lib') + require_relative 'envutil' + EnvUtil.suppress_warning do + require 'dl/import' + end + end + begin + require 'fiddle/types' + rescue LoadError + require_relative 'envutil' + EnvUtil.suppress_warning do + require 'dl/types' + end + end + + module Win32 + begin + extend Fiddle::Importer + rescue NameError + extend DL::Importer + end + dlload "kernel32.dll", "psapi.dll" + begin + include Fiddle::Win32Types + rescue NameError + include DL::Win32Types + end + typealias "SIZE_T", "size_t" + + PROCESS_MEMORY_COUNTERS = struct [ + "DWORD cb", + "DWORD PageFaultCount", + "SIZE_T PeakWorkingSetSize", + "SIZE_T WorkingSetSize", + "SIZE_T QuotaPeakPagedPoolUsage", + "SIZE_T QuotaPagedPoolUsage", + "SIZE_T QuotaPeakNonPagedPoolUsage", + "SIZE_T QuotaNonPagedPoolUsage", + "SIZE_T PagefileUsage", + "SIZE_T PeakPagefileUsage", + ] + + typealias "PPROCESS_MEMORY_COUNTERS", "PROCESS_MEMORY_COUNTERS*" + + extern "HANDLE GetCurrentProcess()", :stdcall + extern "BOOL GetProcessMemoryInfo(HANDLE, PPROCESS_MEMORY_COUNTERS, DWORD)", :stdcall + + module_function + def memory_info + size = PROCESS_MEMORY_COUNTERS.size + data = PROCESS_MEMORY_COUNTERS.malloc + data.cb = size + data if GetProcessMemoryInfo(GetCurrentProcess(), data, size) + end + end + + keys << :peak << :size + def self.read_status + if info = Win32.memory_info + yield :peak, info.PeakPagefileUsage + yield :size, info.PagefileUsage + end + end + else + PAT = /^\s*(\d+)\s+(\d+)$/ + require_relative 'find_executable' + if PSCMD = EnvUtil.find_executable("ps", "-ovsz=", "-orss=", "-p", $$.to_s) {|out| PAT =~ out} + PSCMD.pop + end + raise MiniTest::Skip, "ps command not found" unless PSCMD + + keys << :size << :rss + def self.read_status + if PAT =~ IO.popen(PSCMD + [$$.to_s], "r", err: [:child, :out], &:read) + yield :size, $1.to_i*1024 + yield :rss, $2.to_i*1024 + end + end + end + + Status = Struct.new(*keys) + + class Status + def _update + Memory.read_status do |key, val| + self[key] = val + end + end + end + + class Status + Header = members.map {|k| k.to_s.upcase.rjust(6)}.join('') + Format = "%6d" + + def initialize + _update + end + + def to_s + status = each_pair.map {|n,v| + "#{n}:#{v}" + } + "{#{status.join(",")}}" + end + + def self.parse(str) + status = allocate + str.scan(/(?:\A\{|\G,)(#{members.join('|')}):(\d+)(?=,|\}\z)/) do + status[$1] = $2.to_i + end + status + end + end +end |