62 lines
1.3 KiB
C
62 lines
1.3 KiB
C
#include <ruby/ruby.h>
|
|
#include <vm_core.h>
|
|
|
|
VALUE rb_thread_current_status(rb_thread_t *);
|
|
void rb_add_event_hook(rb_event_hook_func_t, rb_event_flag_t, VALUE);
|
|
int rb_remove_event_hook(rb_event_hook_func_t);
|
|
|
|
static void
|
|
hash_inc(VALUE data, VALUE key)
|
|
{
|
|
VALUE num = INT2FIX(0);
|
|
|
|
if (RTEST(num = rb_hash_aref(data, key))) {
|
|
num = INT2FIX(FIX2INT(num) + 1);
|
|
}
|
|
|
|
rb_hash_aset(data, key, num);
|
|
}
|
|
|
|
static void
|
|
pprof_hook(rb_event_flag_t flag, VALUE data,
|
|
VALUE self, ID dmyid, VALUE dmyklass)
|
|
{
|
|
rb_thread_t *th = GET_THREAD();
|
|
VALUE sig = rb_thread_current_status(th);
|
|
hash_inc(data, sig);
|
|
}
|
|
|
|
static VALUE
|
|
pprof_data(VALUE mod)
|
|
{
|
|
return rb_const_get_at(mod, rb_intern("#pprof_data"));
|
|
}
|
|
|
|
static VALUE
|
|
pprof_start(VALUE self)
|
|
{
|
|
VALUE data = pprof_data(self);
|
|
rb_add_event_hook(pprof_hook, RUBY_EVENT_SWITCH, data);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
pprof_stop(VALUE self)
|
|
{
|
|
rb_remove_event_hook(pprof_hook);
|
|
return Qnil;
|
|
}
|
|
|
|
void
|
|
Init_probeprofiler(void)
|
|
{
|
|
VALUE mPProf;
|
|
|
|
mPProf = rb_define_module("ProbeProfiler");
|
|
rb_const_set(mPProf, rb_intern("#pprof_data"), rb_hash_new());
|
|
rb_define_module_function(mPProf, "start_profile", pprof_start, 0);
|
|
rb_define_module_function(mPProf, "stop_profile", pprof_stop, 0);
|
|
rb_define_module_function(mPProf, "profile_data", pprof_data, 0);
|
|
}
|
|
|