jemalloc_ctl 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ca9382bca94ee59c45b5cc6f1a5cd6bfafedb32b8ad62c8b24c4c4e1ddad33b8
-  data.tar.gz: 5dc23dfa0cc0a2ccc4c034aba3db9e554a2eaa75c0b8664f27c878b5132d0669
+  metadata.gz: 0ee6bbbd66f548328f75198782d1981fb0b2b06772b6d19cb81496157e3efcf4
+  data.tar.gz: fab1affaa3c8c1d94c4b681d1eebb35262acbbfef6faa37ba6dc302a9223bef8
 SHA512:
-  metadata.gz: e5db0172c45f9c25f7f2783731098d948913586d23234b5bb9cafdf83702861eefe6fddd376473fe18611912acb1f9d73ef3a1eb4273ac018cadffbedc35cc52
-  data.tar.gz: a1211d9003b487b6a1b399f03db1556971478a3bb933798afe8d2455674943bafa3d8a008dd24cb058aeffb2230b6bcfa496452fd2cad79a0ee9a448148dc479
+  metadata.gz: c64a56894264596d6679879cac23f9ace7b17dbcfb6e0f7bd60198e0f5dd8326baa651757d2f465aea80803b735161e2886563805add92d0e1be0c81ae5af52c
+  data.tar.gz: ba4c2cb7e37d667485cb58c431ba3479ea5a8394928278279d9271b8479e2a60f3fce6aca477635158fbf59a1276f28c637a93e52ea767a248b28d2577da60cc

data/ext/jemalloc_ctl/jemalloc_ctl.c

@@ -33,6 +33,8 @@ static malloc_stats_print_fn my_malloc_stats_print = NULL;
 #endif
 
 static VALUE rb_mJemallocCtl;
+static VALUE rb_eJemallocCtlError;
+static VALUE rb_eJemallocCtlJemallocNotFound;
 
 static VALUE rb_jemalloc_ctl_enabled_p(VALUE self)
 {
@@ -42,55 +44,55 @@ static VALUE rb_jemalloc_ctl_enabled_p(VALUE self)
 static VALUE rb_jemalloc_ctl_version(VALUE self)
 {
     if (my_mallctl == NULL)
-        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+        rb_raise(rb_eJemallocCtlJemallocNotFound, "jemalloc is not available");
 
     const char *version;
     size_t sz = sizeof(version);
     int err = my_mallctl("version", &version, &sz, NULL, 0);
 
     if (err != 0)
-        rb_raise(rb_eRuntimeError, "mallctl(\"version\") failed with error %d", err);
+        rb_raise(rb_eJemallocCtlError, "mallctl(\"version\") failed with error %d", err);
 
     return rb_str_new_cstr(version);
 }
 
-static VALUE rb_jemalloc_ctl_enable_background_thread(VALUE self)
+static VALUE rb_jemalloc_ctl_toggle_background_thread(VALUE self, VALUE onoff)
 {
     if (my_mallctl == NULL)
-        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+        rb_raise(rb_eJemallocCtlJemallocNotFound, "jemalloc is not available");
 
-    bool val = true;
+    bool val = RTEST(onoff);
     int err = my_mallctl("background_thread", NULL, NULL, &val, sizeof(bool));
 
     if (err != 0)
-        rb_raise(rb_eRuntimeError, "mallctl(\"background_thread\") failed with error %d", err);
+        rb_raise(rb_eJemallocCtlError, "mallctl(\"background_thread\") failed with error %d", err);
 
     return Qtrue;
 }
 
-static VALUE rb_jemalloc_ctl_disable_background_thread(VALUE self)
+static VALUE rb_jemalloc_ctl_toggle_current_thread_tcache(VALUE self, VALUE onoff)
 {
     if (my_mallctl == NULL)
-        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+        rb_raise(rb_eJemallocCtlJemallocNotFound, "jemalloc is not available");
 
-    bool val = false;
-    int err = my_mallctl("background_thread", NULL, NULL, &val, sizeof(bool));
+    bool enabled = RTEST(onoff);
+    int err = my_mallctl("thread.tcache.enabled", NULL, NULL, &enabled, sizeof(enabled));
 
     if (err != 0)
-        rb_raise(rb_eRuntimeError, "mallctl(\"background_thread\") failed with error %d", err);
+        rb_raise(rb_eJemallocCtlError, "mallctl(\"thread.tcache.enabled\") failed with error %d", err);
 
-    return Qfalse;
+    return Qtrue;
 }
 
 static VALUE rb_jemalloc_ctl_flush_current_thread_tcache(VALUE self)
 {
     if (my_mallctl == NULL)
-        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+        rb_raise(rb_eJemallocCtlJemallocNotFound, "jemalloc is not available");
 
     int err = my_mallctl("thread.tcache.flush", NULL, NULL, NULL, 0);
 
     if (err != 0)
-        rb_raise(rb_eRuntimeError, "mallctl(\"thread.tcache.flush\") failed with error %d", err);
+        rb_raise(rb_eJemallocCtlError, "mallctl(\"thread.tcache.flush\") failed with error %d", err);
 
     return Qtrue;
 }
@@ -98,7 +100,7 @@ static VALUE rb_jemalloc_ctl_flush_current_thread_tcache(VALUE self)
 static VALUE rb_jemalloc_ctl_purge_arenas(VALUE self)
 {
     if (my_mallctl == NULL)
-        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+        rb_raise(rb_eJemallocCtlJemallocNotFound, "jemalloc is not available");
 
     char ctl[32];
     snprintf(ctl, sizeof(ctl), "arena.%d.purge", MALLCTL_ARENAS_ALL);
@@ -106,7 +108,7 @@ static VALUE rb_jemalloc_ctl_purge_arenas(VALUE self)
     int err = my_mallctl(ctl, NULL, NULL, NULL, 0);
 
     if (err != 0)
-        rb_raise(rb_eRuntimeError, "mallctl(\"%s\") failed with error %d", ctl, err);
+        rb_raise(rb_eJemallocCtlError, "mallctl(\"%s\") failed with error %d", ctl, err);
 
     return Qtrue;
 }
@@ -118,11 +120,8 @@ static void write_stats_cb(void *opaque, const char *buf)
 
 static VALUE rb_jemalloc_ctl_print_stats(VALUE self, VALUE filename)
 {
-    if (my_mallctl == NULL)
-        rb_raise(rb_eRuntimeError, "jemalloc is not available");
-
     if (my_malloc_stats_print == NULL)
-        rb_raise(rb_eRuntimeError, "malloc_stats_print is not available");
+        rb_raise(rb_eJemallocCtlError, "malloc_stats_print is not available");
 
     const char *path = StringValueCStr(filename);
     FILE *f = fopen(path, "w");
@@ -176,11 +175,13 @@ void Init_jemalloc_ctl(void)
 #endif
 
     rb_mJemallocCtl = rb_define_module("JemallocCtl");
+    rb_eJemallocCtlError = rb_define_class_under(rb_mJemallocCtl, "Error", rb_eStandardError);
+    rb_eJemallocCtlJemallocNotFound = rb_define_class_under(rb_mJemallocCtl, "JemallocNotFound", rb_eJemallocCtlError);
     rb_define_singleton_method(rb_mJemallocCtl, "enabled?", rb_jemalloc_ctl_enabled_p, 0);
     rb_define_singleton_method(rb_mJemallocCtl, "version", rb_jemalloc_ctl_version, 0);
-    rb_define_singleton_method(rb_mJemallocCtl, "enable_background_thread", rb_jemalloc_ctl_enable_background_thread, 0);
-    rb_define_singleton_method(rb_mJemallocCtl, "disable_background_thread", rb_jemalloc_ctl_disable_background_thread, 0);
     rb_define_singleton_method(rb_mJemallocCtl, "print_stats", rb_jemalloc_ctl_print_stats, 1);
+    rb_define_singleton_method(rb_mJemallocCtl, "toggle_background_thread", rb_jemalloc_ctl_toggle_background_thread, 1);
     rb_define_singleton_method(rb_mJemallocCtl, "purge_arenas", rb_jemalloc_ctl_purge_arenas, 0);
     rb_define_singleton_method(rb_mJemallocCtl, "flush_current_thread_tcache", rb_jemalloc_ctl_flush_current_thread_tcache, 0);
+    rb_define_singleton_method(rb_mJemallocCtl, "toggle_current_thread_tcache", rb_jemalloc_ctl_toggle_current_thread_tcache, 1);
 }

data/lib/jemalloc_ctl.rb

@@ -1,3 +1,16 @@
 require "jemalloc_ctl/jemalloc_ctl"
 module JemallocCtl
+  def self.disable_background_thread
+    toggle_background_thread(false)
+  end
+  def self.enable_background_thread
+    toggle_background_thread(true)
+  end
+
+  def self.disable_current_thread_tcache
+    toggle_current_thread_tcache(false)
+  end
+  def self.enable_current_thread_tcache
+    toggle_current_thread_tcache(true)
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: jemalloc_ctl
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Luke Gruber