mirror of https://github.com/nodejs/node.git
105 lines
3.7 KiB
C
105 lines
3.7 KiB
C
#include <node_api.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#define NODE_API_CALL(call) \
|
|
do { \
|
|
napi_status status = call; \
|
|
if (status != napi_ok) { \
|
|
fprintf(stderr, #call " failed: %d\n", status); \
|
|
abort(); \
|
|
} \
|
|
} while (0)
|
|
|
|
#define ABORT_IF_FALSE(condition) \
|
|
if (!(condition)) { \
|
|
fprintf(stderr, #condition " failed\n"); \
|
|
abort(); \
|
|
}
|
|
|
|
static napi_value Runner(napi_env env,
|
|
napi_callback_info info,
|
|
napi_property_attributes attr) {
|
|
napi_value argv[2], undefined, js_array_length, start, end;
|
|
size_t argc = 2;
|
|
napi_valuetype val_type = napi_undefined;
|
|
bool is_array = false;
|
|
uint32_t array_length = 0;
|
|
napi_value* native_array;
|
|
|
|
// Validate params and retrieve start and end function.
|
|
NODE_API_CALL(napi_get_cb_info(env, info, &argc, argv, NULL, NULL));
|
|
ABORT_IF_FALSE(argc == 2);
|
|
NODE_API_CALL(napi_typeof(env, argv[0], &val_type));
|
|
ABORT_IF_FALSE(val_type == napi_object);
|
|
NODE_API_CALL(napi_is_array(env, argv[1], &is_array));
|
|
ABORT_IF_FALSE(is_array);
|
|
NODE_API_CALL(napi_get_array_length(env, argv[1], &array_length));
|
|
NODE_API_CALL(napi_get_named_property(env, argv[0], "start", &start));
|
|
NODE_API_CALL(napi_typeof(env, start, &val_type));
|
|
ABORT_IF_FALSE(val_type == napi_function);
|
|
NODE_API_CALL(napi_get_named_property(env, argv[0], "end", &end));
|
|
NODE_API_CALL(napi_typeof(env, end, &val_type));
|
|
ABORT_IF_FALSE(val_type == napi_function);
|
|
|
|
NODE_API_CALL(napi_get_undefined(env, &undefined));
|
|
NODE_API_CALL(napi_create_uint32(env, array_length, &js_array_length));
|
|
|
|
// Copy objects into a native array.
|
|
native_array = malloc(array_length * sizeof(*native_array));
|
|
for (uint32_t idx = 0; idx < array_length; idx++) {
|
|
NODE_API_CALL(napi_get_element(env, argv[1], idx, &native_array[idx]));
|
|
}
|
|
|
|
const napi_property_descriptor desc = {
|
|
"prop", NULL, NULL, NULL, NULL, js_array_length, attr, NULL};
|
|
|
|
// Start the benchmark.
|
|
napi_call_function(env, argv[0], start, 0, NULL, NULL);
|
|
|
|
for (uint32_t idx = 0; idx < array_length; idx++) {
|
|
NODE_API_CALL(napi_define_properties(env, native_array[idx], 1, &desc));
|
|
}
|
|
|
|
// Conclude the benchmark.
|
|
NODE_API_CALL(
|
|
napi_call_function(env, argv[0], end, 1, &js_array_length, NULL));
|
|
|
|
free(native_array);
|
|
|
|
return undefined;
|
|
}
|
|
|
|
static napi_value RunFastPath(napi_env env, napi_callback_info info) {
|
|
return Runner(env, info, napi_writable | napi_enumerable | napi_configurable);
|
|
}
|
|
|
|
static napi_value RunSlowPath(napi_env env, napi_callback_info info) {
|
|
return Runner(env, info, napi_writable | napi_enumerable);
|
|
}
|
|
|
|
NAPI_MODULE_INIT() {
|
|
napi_property_descriptor props[] = {
|
|
{"runFastPath",
|
|
NULL,
|
|
RunFastPath,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
napi_writable | napi_configurable | napi_enumerable,
|
|
NULL},
|
|
{"runSlowPath",
|
|
NULL,
|
|
RunSlowPath,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
napi_writable | napi_configurable | napi_enumerable,
|
|
NULL},
|
|
};
|
|
|
|
NODE_API_CALL(napi_define_properties(
|
|
env, exports, sizeof(props) / sizeof(*props), props));
|
|
return exports;
|
|
}
|