W3SLAV/micropython
1
Fork 0
mirror of https://github.com/W3SLAV/micropython.git synced 2025-06-19 12:05:32 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

webassembly/modjsffi: Add mem_info function to get detailed stats.

Browse Source 
This allows querying the GC heap size/used/free values, as well as the
number of alive JsProxy and PyProxy objects, referenced by proxy_c_ref and
proxy_js_ref.

Signed-off-by: Damien George <damien@micropython.org>
This commit is contained in:
Damien George 2024-05-21 11:43:00 +10:00
parent 57a9ffa632
commit a919ce26d3
1 changed files with 40 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
ports/webassembly/modjsffi.c
View File

@ -25,6 +25,7 @@
*/ */
#include "emscripten.h" #include "emscripten.h"
#include "py/gc.h"
#include "py/objmodule.h" #include "py/objmodule.h"
#include "py/runtime.h" #include "py/runtime.h"
#include "proxy_c.h" #include "proxy_c.h"
@ -75,6 +76,44 @@ static mp_obj_t mp_jsffi_async_timeout_ms(mp_obj_t arg) {
} }
static MP_DEFINE_CONST_FUN_OBJ_1(mp_jsffi_async_timeout_ms_obj, mp_jsffi_async_timeout_ms); static MP_DEFINE_CONST_FUN_OBJ_1(mp_jsffi_async_timeout_ms_obj, mp_jsffi_async_timeout_ms);
// *FORMAT-OFF*
EM_JS(void, js_get_proxy_js_ref_info, (uint32_t * out), {
let used = 0;
for (const elem of proxy_js_ref) {
if (elem !== undefined) {
++used;
}
}
Module.setValue(out, proxy_js_ref.length, "i32");
Module.setValue(out + 4, used, "i32");
});
// *FORMAT-ON*
static mp_obj_t mp_jsffi_mem_info(void) {
mp_obj_list_t *l = (mp_obj_list_t *)MP_OBJ_TO_PTR(MP_STATE_PORT(proxy_c_ref));
mp_int_t used = 0;
for (size_t i = 0; i < l->len; ++i) {
if (l->items[i] != MP_OBJ_NULL) {
++used;
}
}
uint32_t proxy_js_ref_info[2];
js_get_proxy_js_ref_info(proxy_js_ref_info);
gc_info_t info;
gc_info(&info);
mp_obj_t elems[] = {
MP_OBJ_NEW_SMALL_INT(info.total), // GC heap total bytes
MP_OBJ_NEW_SMALL_INT(info.used), // GC heap used bytes
MP_OBJ_NEW_SMALL_INT(info.free), // GC heap free bytes
MP_OBJ_NEW_SMALL_INT(l->len), // proxy_c_ref allocated size
MP_OBJ_NEW_SMALL_INT(used), // proxy_c_ref used
MP_OBJ_NEW_SMALL_INT(proxy_js_ref_info[0]), // proxy_js_ref allocated size
MP_OBJ_NEW_SMALL_INT(proxy_js_ref_info[1]), // proxy_js_ref used
};
return mp_obj_new_tuple(MP_ARRAY_SIZE(elems), elems);
}
static MP_DEFINE_CONST_FUN_OBJ_0(mp_jsffi_mem_info_obj, mp_jsffi_mem_info);
static const mp_rom_map_elem_t mp_module_jsffi_globals_table[] = { static const mp_rom_map_elem_t mp_module_jsffi_globals_table[] = {
{ MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_jsffi) }, { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_jsffi) },
@ -83,6 +122,7 @@ static const mp_rom_map_elem_t mp_module_jsffi_globals_table[] = {
{ MP_ROM_QSTR(MP_QSTR_create_proxy), MP_ROM_PTR(&mp_jsffi_create_proxy_obj) }, { MP_ROM_QSTR(MP_QSTR_create_proxy), MP_ROM_PTR(&mp_jsffi_create_proxy_obj) },
{ MP_ROM_QSTR(MP_QSTR_to_js), MP_ROM_PTR(&mp_jsffi_to_js_obj) }, { MP_ROM_QSTR(MP_QSTR_to_js), MP_ROM_PTR(&mp_jsffi_to_js_obj) },
{ MP_ROM_QSTR(MP_QSTR_async_timeout_ms), MP_ROM_PTR(&mp_jsffi_async_timeout_ms_obj) }, { MP_ROM_QSTR(MP_QSTR_async_timeout_ms), MP_ROM_PTR(&mp_jsffi_async_timeout_ms_obj) },
{ MP_ROM_QSTR(MP_QSTR_mem_info), MP_ROM_PTR(&mp_jsffi_mem_info_obj) },
}; };
static MP_DEFINE_CONST_DICT(mp_module_jsffi_globals, mp_module_jsffi_globals_table); static MP_DEFINE_CONST_DICT(mp_module_jsffi_globals, mp_module_jsffi_globals_table);