admin/mruby-mruby
1
0
Fork 0
mirror of https://github.com/mruby/mruby synced 2026-06-08 16:11:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add ObjectSpace.memsize_of_all

Browse Source
This commit is contained in:
Rory OConnell
2020-07-17 20:09:44 -07:00
parent 1952004d5a
commit fe1ec9afdd
2 changed files with 39 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mrbgems/mruby-objectspace/src/mruby_objectspace.c
+25
View File
@@ -344,6 +344,30 @@ os_memsize_of(mrb_state *mrb, mrb_value self)
return mrb_fixnum_value(total);
}
struct os_memsize_of_all_cb_data {
mrb_int t;
struct RClass *type;
};
static int
os_memsize_of_all_cb(mrb_state *mrb, struct RBasic *obj, void *d)
{
struct os_memsize_of_all_cb_data *data = (struct os_memsize_of_all_cb_data *)d;
if(obj->c == data->type) data->t += os_memsize_of_object(mrb, mrb_obj_value(obj));
return 0;
}
static mrb_value
os_memsize_of_all(mrb_state *mrb, mrb_value self)
{
mrb_value type;
struct os_memsize_of_all_cb_data data = { 0 };
mrb_get_args(mrb, "C", &type);
data.type = mrb_class_ptr(type);
mrb_objspace_each_objects(mrb, os_memsize_of_all_cb, &data);
return mrb_fixnum_value(data.t);
}
void
mrb_mruby_objectspace_gem_init(mrb_state *mrb)
{
@@ -351,6 +375,7 @@ mrb_mruby_objectspace_gem_init(mrb_state *mrb)
mrb_define_class_method(mrb, os, "count_objects", os_count_objects, MRB_ARGS_OPT(1));
mrb_define_class_method(mrb, os, "each_object", os_each_object, MRB_ARGS_OPT(1));
mrb_define_class_method(mrb, os, "memsize_of", os_memsize_of, MRB_ARGS_REQ(1)|MRB_ARGS_OPT(1));
mrb_define_class_method(mrb, os, "memsize_of_all", os_memsize_of_all, MRB_ARGS_REQ(1));
}
void