leak_profiler 0.4.1 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 386f532ac6ac35f8f971e9fcc94d29e98d509feb16ffa49f471235e07f74fe64
-  data.tar.gz: 138da4ee3ac1cd9e2057e3be13e955a3d0ea044426de4bca298ae8255bd62777
+  metadata.gz: 3e7942b0713d364d60dcfbeb171894c69aa63028adc6795d4b48e85d6019152d
+  data.tar.gz: '05967d0eefaf0301b380620c23b5eb925ac62a26ad45c1504a7ff2e6690f537a'
 SHA512:
-  metadata.gz: 6d02acbee4be5c1ea02e22cff0ad8653a5028be7b660bfac35533c75eca52d6eb040d596e817117c189e72318c79b60bd0ee17ffe499e273c5f98ea8d1e872d6
-  data.tar.gz: 06aee1fc3dac01bf958d60870bfdd82debcc4605f2589b4477435e9f6955c2035d4cf220d718673074c633e1209e823e420e2ab8499d60805b5fe46f8e652eee
+  metadata.gz: b430a5fe0d7ad6fc90d1dd1f949aeb0b2948487a6f651b4c52975774b4e5cb21ab98a9f7219e8c665f90daa3c70f6aa3d3c803ffdc9aa48acddd0e0e95d8bf15
+  data.tar.gz: 175e865aa10c22816d9b86c04301a0c574af0260becbe3fdf76b78d8f338b466fdc2170431a4244e0296ed47b27223879e52c60060363654f1adc8da7cd49638

data/.rubocop.yml

@@ -14,6 +14,9 @@ Metrics:
 Style/Copyright:
   Enabled: false
 
+Style/GlobalVars:
+  Enabled: false
+
 Style/ImplicitRuntimeError:
   Enabled: false
 

data/Rakefile

@@ -1,9 +1,14 @@
 # frozen_string_literal: true
 
 require 'bundler/gem_tasks'
+require 'rake/extensiontask'
 require 'rake/testtask'
 
-task default: :test
+task default: %i[compile test]
+
+Rake::ExtensionTask.new('leak_profiler_ext') do |ext|
+  ext.ext_dir = 'ext/leak_profiler'
+end
 
 Rake::TestTask.new do |task|
   task.pattern = 'test/test_*.rb'

data/ext/leak_profiler/extconf.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+require 'mkmf'
+
+$LDFLAGS += ' -lpsapi' if RUBY_PLATFORM.include?('mingw')
+
+create_makefile('leak_profiler_ext')

data/ext/leak_profiler/leak_profiler.c

@@ -0,0 +1,37 @@
+#include "ruby.h"
+
+#if defined(_WIN32)
+#include <psapi.h>
+
+static VALUE leak_profiler_max_rss(VALUE self)
+{
+        PROCESS_MEMORY_COUNTERS pmc;
+    if (!GetProcessMemoryInfo(GetCurrentProcess(), &pmc, sizeof(pmc))) {
+        rb_sys_fail("GetProcessMemoryInfo");
+    }
+    return LONG2NUM(pmc.PeakWorkingSetSize);
+}
+
+#else
+#include <sys/resource.h>
+
+static VALUE leak_profiler_max_rss(VALUE self)
+{
+    struct rusage usage;
+    if (getrusage(RUSAGE_SELF, &usage) == -1) {
+        rb_sys_fail("getrusage");
+    }
+
+    return LONG2NUM(usage.ru_maxrss);
+}
+
+#endif
+
+void Init_leak_profiler_ext(void)
+{
+    VALUE cLeakProfiler = rb_define_class("LeakProfiler", rb_cObject);
+    VALUE cMemoryUsage = rb_define_class_under(cLeakProfiler, "MemoryUsage", rb_cObject);
+
+
+    rb_define_singleton_method(cMemoryUsage, "max_rss", leak_profiler_max_rss, 0);
+}

data/lib/leak_profiler/allocations.rb

@@ -1,11 +1,17 @@
 # frozen_string_literal: true
 
+# rbs_inline: enabled
+
 require 'objspace'
 
 class LeakProfiler
   class Allocations
     attr_reader :thread
 
+    # @rbs logger: untyped
+    # @rbs interval: Integer
+    # @rbs max_allocations: Integer
+    # @rbs max_referrers: Integer
     def initialize(logger:, interval:, max_allocations:, max_referrers:)
       @logger = logger
       @interval = interval

data/lib/leak_profiler/memory_memsize.rb

@@ -1,11 +1,16 @@
 # frozen_string_literal: true
 
+# rbs_inline: enabled
+
 require 'objspace'
 
 class LeakProfiler
   class MemoryMemsize
     attr_reader :thread
 
+    # @rbs output_dir: String
+    # @rbs interval: Integer
+    # @rbs filename: String
     def initialize(output_dir:, interval:, filename: nil)
       @output_dir = output_dir
       @interval = interval

data/lib/leak_profiler/memory_usage.rb

@@ -1,9 +1,14 @@
 # frozen_string_literal: true
 
+# rbs_inline: enabled
+
 class LeakProfiler
   class MemoryUsage
     attr_reader :thread
 
+    # @rbs output_dir: String
+    # @rbs interval: Integer
+    # @rbs filename: String
     def initialize(output_dir:, interval:, filename: nil)
       @output_dir = output_dir
       @interval = interval
@@ -11,17 +16,13 @@ class LeakProfiler
     end
 
     def report
-      raise('Not supported Windows platform because this uses `ps` command for measurement.') if /mingw/.match?(RUBY_PLATFORM)
-
-      pid = Process.pid
-
       @thread = Thread.start do
         i = 0
         File.open(File.expand_path(File.join(@output_dir, @filename)), 'w') do |f|
           f.puts('elapsed [sec],memory usage (rss) [MB]')
 
           loop do
-            rss = Integer(`ps -o rss= -p #{pid}`) / 1024.0
+            rss = LeakProfiler::MemoryUsage.max_rss / 1024.0
             f.puts("#{i},#{rss}")
             f.fsync
             i += @interval

data/lib/leak_profiler.rb

@@ -1,3 +1,4 @@
 # frozen_string_literal: true
 
 require 'leak_profiler/leak_profiler'
+require_relative 'leak_profiler_ext.so'

data/sig/generated/leak_profiler/allocations.rbs

@@ -0,0 +1,27 @@
+# Generated from lib/leak_profiler/allocations.rb with RBS::Inline
+
+class LeakProfiler
+  class Allocations
+    attr_reader thread: untyped
+
+    # @rbs logger: untyped
+    # @rbs interval: Integer
+    # @rbs max_allocations: Integer
+    # @rbs max_referrers: Integer
+    def initialize: (logger: untyped, interval: Integer, max_allocations: Integer, max_referrers: Integer) -> untyped
+
+    def report: () -> untyped
+
+    private
+
+    def report_allocations: (untyped allocations) -> untyped
+
+    def report_referrer_objects: (untyped allocations) -> untyped
+
+    def detect_referrer_objects: (untyped object) -> untyped
+
+    def allocated_location: (untyped obj) -> untyped
+
+    def sort: (untyped allocations) -> untyped
+  end
+end

data/sig/generated/leak_profiler/memory_memsize.rbs

@@ -0,0 +1,14 @@
+# Generated from lib/leak_profiler/memory_memsize.rb with RBS::Inline
+
+class LeakProfiler
+  class MemoryMemsize
+    attr_reader thread: untyped
+
+    # @rbs output_dir: String
+    # @rbs interval: Integer
+    # @rbs filename: String
+    def initialize: (output_dir: String, interval: Integer, ?filename: String) -> untyped
+
+    def report: () -> untyped
+  end
+end

data/sig/generated/leak_profiler/memory_usage.rbs

@@ -0,0 +1,14 @@
+# Generated from lib/leak_profiler/memory_usage.rb with RBS::Inline
+
+class LeakProfiler
+  class MemoryUsage
+    attr_reader thread: untyped
+
+    # @rbs output_dir: String
+    # @rbs interval: Integer
+    # @rbs filename: String
+    def initialize: (output_dir: String, interval: Integer, ?filename: String) -> untyped
+
+    def report: () -> untyped
+  end
+end

data/sig/leak_profiler.rbs

@@ -0,0 +1,5 @@
+class LeakProfiler
+  class MemoryUsage
+    def self.max_rss: () -> Integer
+  end
+end

data/test/test_memory_usage.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require 'leak_profiler'
+require 'minitest/autorun'
+
+class MemoryUsageTest < Minitest::Test
+  def test_report_rss
+    assert_equal(true, LeakProfiler::MemoryUsage.max_rss.positive?)
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: leak_profiler
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
 platform: ruby
 authors:
 - Watson
@@ -27,7 +27,8 @@ description: A simple profiler for Ruby to detect memory leak.
 email:
 - watson1978@gmail.com
 executables: []
-extensions: []
+extensions:
+- ext/leak_profiler/extconf.rb
 extra_rdoc_files: []
 files:
 - ".rubocop.yml"
@@ -35,13 +36,20 @@ files:
 - LICENSE
 - README.md
 - Rakefile
+- ext/leak_profiler/extconf.rb
+- ext/leak_profiler/leak_profiler.c
 - lib/leak_profiler.rb
 - lib/leak_profiler/allocations.rb
 - lib/leak_profiler/leak_profiler.rb
 - lib/leak_profiler/memory_memsize.rb
 - lib/leak_profiler/memory_usage.rb
+- sig/generated/leak_profiler/allocations.rbs
 - sig/generated/leak_profiler/leak_profiler.rbs
+- sig/generated/leak_profiler/memory_memsize.rbs
+- sig/generated/leak_profiler/memory_usage.rbs
+- sig/leak_profiler.rbs
 - test/test_leak_profiler.rb
+- test/test_memory_usage.rb
 homepage: https://github.com/Watson1978/leak_profiler
 licenses:
 - MIT