jemalloc_ctl 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0ee6bbbd66f548328f75198782d1981fb0b2b06772b6d19cb81496157e3efcf4
-  data.tar.gz: fab1affaa3c8c1d94c4b681d1eebb35262acbbfef6faa37ba6dc302a9223bef8
+  metadata.gz: d7be6090c8c75247efea4b0c3448e68d688dd449d13ad7e53a82850b54b5c83c
+  data.tar.gz: '08906468b2623758dbcdc3804df5ad24074783f83cb589274e4c5f4d819f952c'
 SHA512:
-  metadata.gz: c64a56894264596d6679879cac23f9ace7b17dbcfb6e0f7bd60198e0f5dd8326baa651757d2f465aea80803b735161e2886563805add92d0e1be0c81ae5af52c
-  data.tar.gz: ba4c2cb7e37d667485cb58c431ba3479ea5a8394928278279d9271b8479e2a60f3fce6aca477635158fbf59a1276f28c637a93e52ea767a248b28d2577da60cc
+  metadata.gz: 445148d11c6a400b26fb814ebb0698858a50e7ecceea3a7fce29d3e37efc7f66c75445337179b77c0b3db0d9a80b7a0657d5a09f01c78df0166cf1aa5e6fb844
+  data.tar.gz: 8168e928e33be6f7a3234c3a457d99fcae86b65dcd13d4ff4763de992dc9d5fd2b0dd5666e60c654060c252d0d51b1f65934d1dd8bf2458b5872018cf2916636

data/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2026 Luke Gruber
+Copyright (c) Shopify Inc.
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

data/ext/jemalloc_ctl/jemalloc_ctl.c

@@ -1,4 +1,5 @@
 #include <ruby.h>
+#include <ruby/debug.h>
 #include <dlfcn.h>
 #include <stdbool.h>
 
@@ -36,6 +37,13 @@ static VALUE rb_mJemallocCtl;
 static VALUE rb_eJemallocCtlError;
 static VALUE rb_eJemallocCtlJemallocNotFound;
 
+static VALUE main_thread_val = Qnil;
+static int gc_exit_hook_installed = 0;
+
+static int major_gc_purge_hook_installed = 0;
+static size_t last_major_gc_count = 0;
+static VALUE major_gc_count_sym = 0;
+
 static VALUE rb_jemalloc_ctl_enabled_p(VALUE self)
 {
     return my_mallctl != NULL ? Qtrue : Qfalse;
@@ -97,6 +105,76 @@ static VALUE rb_jemalloc_ctl_flush_current_thread_tcache(VALUE self)
     return Qtrue;
 }
 
+static void gc_exit_tcache_flush_hook(rb_event_flag_t flag, VALUE data,
+                                       VALUE self, ID mid, VALUE klass)
+{
+    // Runs inside GC: must not allocate Ruby objects or raise.
+    if (my_mallctl == NULL) return;
+    if (rb_thread_current() == main_thread_val) return;
+    my_mallctl("thread.tcache.flush", NULL, NULL, NULL, 0);
+}
+
+static VALUE rb_jemalloc_ctl_enable_gc_exit_tcache_flush(VALUE self)
+{
+    if (my_mallctl == NULL)
+        rb_raise(rb_eJemallocCtlJemallocNotFound, "jemalloc is not available");
+
+    if (gc_exit_hook_installed)
+        return Qfalse;
+
+    if (main_thread_val == Qnil) {
+        main_thread_val = rb_thread_main();
+        rb_gc_register_address(&main_thread_val);
+    }
+    rb_add_event_hook(gc_exit_tcache_flush_hook, RUBY_INTERNAL_EVENT_GC_EXIT, Qnil);
+    gc_exit_hook_installed = 1;
+    return Qtrue;
+}
+
+static VALUE rb_jemalloc_ctl_disable_gc_exit_tcache_flush(VALUE self)
+{
+    if (!gc_exit_hook_installed)
+        return Qfalse;
+
+    rb_remove_event_hook(gc_exit_tcache_flush_hook);
+    gc_exit_hook_installed = 0;
+    return Qtrue;
+}
+
+static void major_gc_arena_purge_hook(rb_event_flag_t flag, VALUE data,
+                                       VALUE self, ID mid, VALUE klass)
+{
+    // Runs inside GC: must not allocate Ruby objects or raise.
+    if (my_mallctl == NULL) return;
+
+    size_t current = rb_gc_stat(major_gc_count_sym); // safe
+    if (current == last_major_gc_count) return;
+    last_major_gc_count = current;
+
+    char ctl[32];
+    snprintf(ctl, sizeof(ctl), "arena.%d.purge", MALLCTL_ARENAS_ALL);
+    my_mallctl(ctl, NULL, NULL, NULL, 0);
+}
+
+static VALUE rb_jemalloc_ctl_toggle_major_gc_arena_purge(VALUE self, VALUE onoff)
+{
+    if (my_mallctl == NULL)
+        rb_raise(rb_eJemallocCtlJemallocNotFound, "jemalloc is not available");
+
+    bool enable = RTEST(onoff);
+    if (enable && !major_gc_purge_hook_installed) {
+        if (major_gc_count_sym == 0)
+            major_gc_count_sym = ID2SYM(rb_intern("major_gc_count"));
+        last_major_gc_count = rb_gc_stat(major_gc_count_sym);
+        rb_add_event_hook(major_gc_arena_purge_hook, RUBY_INTERNAL_EVENT_GC_EXIT, Qnil);
+        major_gc_purge_hook_installed = 1;
+    } else if (!enable && major_gc_purge_hook_installed) {
+        rb_remove_event_hook(major_gc_arena_purge_hook);
+        major_gc_purge_hook_installed = 0;
+    }
+    return Qtrue;
+}
+
 static VALUE rb_jemalloc_ctl_purge_arenas(VALUE self)
 {
     if (my_mallctl == NULL)
@@ -184,4 +262,7 @@ void Init_jemalloc_ctl(void)
     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);
+    rb_define_singleton_method(rb_mJemallocCtl, "enable_gc_exit_tcache_flush!", rb_jemalloc_ctl_enable_gc_exit_tcache_flush, 0);
+    rb_define_singleton_method(rb_mJemallocCtl, "disable_gc_exit_tcache_flush!", rb_jemalloc_ctl_disable_gc_exit_tcache_flush, 0);
+    rb_define_singleton_method(rb_mJemallocCtl, "toggle_major_gc_arena_purge!", rb_jemalloc_ctl_toggle_major_gc_arena_purge, 1);
 }

metadata

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