leak_profiler 0.5.1 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b9eee59048d470d4ea437d6eabf4295b4b9fe98b547ed90f43675f56d31b30f4
-  data.tar.gz: 3f85e846b9cb4e9e6232b69c2025e32223a410af5f8c1edf5a8903ebe24d97fd
+  metadata.gz: a0cec55b42b6d790d84c3b54c009da7623040f8aa5c0d3a871eb8d44546d82aa
+  data.tar.gz: 0f576971b8e6b1734f51ab4a1bdbb8e00d2822a1eedb1e24f53dc7342c25caff
 SHA512:
-  metadata.gz: e070418ef607471a5364579825addfcde2b1d4f13b882546f88cf24ceaea72754439f770d5c44a5c5e2500520ef579285ac9fa252271176295ab56a6074ec434
-  data.tar.gz: 0be27cc605be614704a20976ec107f9dc56c63235cad4086123365c4f5b2b6ee74ad6cb55320f7cc868cd7661328348ec7797117afd8e018cc3c8a97cf45922f
+  metadata.gz: 1636ecd9234f7ba4216c2dbc22892c7aafef11cf26efda7c198dccde3c45cc5273eb2f23974d9524c6d2435f44533c2b20b60f0c70473bc69ebf0101729eddda
+  data.tar.gz: b05038cb49f6e21a1a44a3dcc8525c3ee6fdb24a8aa4ae0c04b0544f8a4121218dbe80932281970d61260adc05e38042b2272c4502da4d516e71f0569f589748

data/README.md

@@ -86,11 +86,6 @@ elapsed [sec],memory usage (rss) [MB]
   * `interval` (default `1`): The interval in seconds for report.
   * `filename` (defalut `nil`): Specify the filename if you want to use custom filename.
 
-> [!WARNING]
-> This uses this uses `ps` command for measurement.
-> So, this is not supported Windows platform.
-
-
 ### `LeakProfiler#report_memsize`
 This method outputs `ObjectSpace.memsize_of_all` values with CSV format, like:
 

data/ext/leak_profiler/leak_profiler.c

@@ -1,38 +1,52 @@
 #include "ruby.h"
-
+#include <stdio.h>
+#include <unistd.h>
 
 // get the maximum resident set size (RSS) of the current process
 // return the value in kilobytes
 #if defined(_WIN32)
 #include <psapi.h>
 
-static VALUE leak_profiler_max_rss(VALUE self)
+static VALUE leak_profiler_rss(VALUE self)
 {
-        PROCESS_MEMORY_COUNTERS pmc;
+    PROCESS_MEMORY_COUNTERS pmc;
     if (!GetProcessMemoryInfo(GetCurrentProcess(), &pmc, sizeof(pmc))) {
         rb_sys_fail("GetProcessMemoryInfo");
     }
     return LONG2NUM(pmc.PeakWorkingSetSize / 1024);
 }
 
-#else
-#include <sys/resource.h>
+#elif defined(__APPLE__)
 
-static VALUE leak_profiler_max_rss(VALUE self)
-{
-    struct rusage usage;
-    long max_rss;
+#include <mach/mach.h>
 
-    if (getrusage(RUSAGE_SELF, &usage) == -1) {
-        rb_sys_fail("getrusage");
+static VALUE leak_profiler_rss(VALUE self)
+{
+    struct mach_task_basic_info info;
+    mach_msg_type_number_t count = MACH_TASK_BASIC_INFO_COUNT;
+    kern_return_t kr = task_info(mach_task_self(), MACH_TASK_BASIC_INFO, (task_info_t)&info, &count);
+    if (kr != KERN_SUCCESS) {
+        rb_sys_fail("task_info");
     }
-    max_rss = usage.ru_maxrss;
+    return LONG2NUM(info.resident_size / 1024);
+}
 
-#if defined(__APPLE__)
-    max_rss = max_rss / 1024;
-#endif
+#else // linux
 
-    return LONG2NUM(max_rss);
+static VALUE leak_profiler_rss(VALUE self)
+{
+    long rss = 0;
+
+    FILE *file = fopen("/proc/self/statm", "r");
+    if (!file) {
+        rb_sys_fail("/proc/self/statm");
+    }
+    if (fscanf(file, "%*s%ld", &rss) != 1) {
+        fclose(file);
+        rb_sys_fail("fscanf");
+    }
+    fclose(file);
+    return LONG2NUM(rss * sysconf(_SC_PAGESIZE) / 1024);
 }
 
 #endif
@@ -43,5 +57,5 @@ void Init_leak_profiler_ext(void)
     VALUE cMemoryUsage = rb_define_class_under(cLeakProfiler, "MemoryUsage", rb_cObject);
 
 
-    rb_define_singleton_method(cMemoryUsage, "max_rss", leak_profiler_max_rss, 0);
+    rb_define_singleton_method(cMemoryUsage, "rss", leak_profiler_rss, 0);
 }

data/lib/leak_profiler/memory_usage.rb

@@ -22,7 +22,7 @@ class LeakProfiler
           f.puts('elapsed [sec],memory usage (rss) [MB]')
 
           loop do
-            rss = LeakProfiler::MemoryUsage.max_rss / 1024.0
+            rss = LeakProfiler::MemoryUsage.rss / 1024.0
             f.puts("#{i},#{rss}")
             f.fsync
             i += @interval

data/sig/leak_profiler.rbs

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

data/test/test_memory_usage.rb

@@ -5,6 +5,6 @@ require 'minitest/autorun'
 
 class MemoryUsageTest < Minitest::Test
   def test_report_rss
-    assert_equal(true, LeakProfiler::MemoryUsage.max_rss.positive?)
+    assert_equal(true, LeakProfiler::MemoryUsage.rss.positive?)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: leak_profiler
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.6.0
 platform: ruby
 authors:
 - Watson