node/test/node-api/test_async_context/binding.c at main · fetchapi/node · GitHub

History

129 lines (103 loc) · 4.3 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

#include <node_api.h>

#include <assert.h>

#include "../../js-native-api/common.h"

#define MAX_ARGUMENTS 10

#define RESERVED_ARGS 3

static napi_value MakeCallback(napi_env env, napi_callback_info info) {

size_t argc = MAX_ARGUMENTS;

size_t n;

napi_value args[MAX_ARGUMENTS];

// NOLINTNEXTLINE (readability/null_usage)

NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL));

NODE_API_ASSERT(env, argc > 0, "Wrong number of arguments");

napi_value async_context_wrap = args[0];

napi_value recv = args[1];

napi_value func = args[2];

napi_value argv[MAX_ARGUMENTS - RESERVED_ARGS];

for (n = RESERVED_ARGS; n < argc; n += 1) {

argv[n - RESERVED_ARGS] = args[n];

}

napi_valuetype func_type;

NODE_API_CALL(env, napi_typeof(env, func, &func_type));

napi_async_context context;

NODE_API_CALL(env, napi_unwrap(env, async_context_wrap, (void**)&context));

napi_value result;

if (func_type == napi_function) {

NODE_API_CALL(env, napi_make_callback(

env, context, recv, func, argc - RESERVED_ARGS, argv, &result));

} else {

NODE_API_ASSERT(env, false, "Unexpected argument type");

}

return result;

}

static void AsyncDestroyCb(napi_env env, void* data, void* hint) {

napi_status status = napi_async_destroy(env, (napi_async_context)data);

// We cannot use NODE_API_ASSERT_RETURN_VOID because we need to have a JS

// stack below in order to use exceptions.

assert(status == napi_ok);

}

#define CREATE_ASYNC_RESOURCE_ARGC 2

static napi_value CreateAsyncResource(napi_env env, napi_callback_info info) {

napi_value async_context_wrap;

NODE_API_CALL(env, napi_create_object(env, &async_context_wrap));

size_t argc = CREATE_ASYNC_RESOURCE_ARGC;

napi_value args[CREATE_ASYNC_RESOURCE_ARGC];

// NOLINTNEXTLINE (readability/null_usage)

NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL));

napi_value resource = args[0];

napi_value js_destroy_on_finalizer = args[1];

napi_valuetype resource_type;

NODE_API_CALL(env, napi_typeof(env, resource, &resource_type));

if (resource_type != napi_object) {

resource = NULL;

}

napi_value resource_name;

NODE_API_CALL(env, napi_create_string_utf8(

env, "test_async", NAPI_AUTO_LENGTH, &resource_name));

napi_async_context context;

NODE_API_CALL(env, napi_async_init(env, resource, resource_name, &context));

bool destroy_on_finalizer = true;

if (argc == 2) {

NODE_API_CALL(env, napi_get_value_bool(env, js_destroy_on_finalizer, &destroy_on_finalizer));

}

if (resource_type == napi_object && destroy_on_finalizer) {

NODE_API_CALL(env, napi_add_finalizer(

env, resource, (void*)context, AsyncDestroyCb, NULL, NULL));

}

NODE_API_CALL(env, napi_wrap(env, async_context_wrap, context, NULL, NULL, NULL));

return async_context_wrap;

}

#define DESTROY_ASYNC_RESOURCE_ARGC 1

static napi_value DestroyAsyncResource(napi_env env, napi_callback_info info) {

size_t argc = DESTROY_ASYNC_RESOURCE_ARGC;

napi_value args[DESTROY_ASYNC_RESOURCE_ARGC];

// NOLINTNEXTLINE (readability/null_usage)

NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL));

NODE_API_ASSERT(env, argc == 1, "Wrong number of arguments");

napi_value async_context_wrap = args[0];

napi_async_context async_context;

NODE_API_CALL(env,

napi_remove_wrap(env, async_context_wrap, (void**)&async_context));

NODE_API_CALL(env, napi_async_destroy(env, async_context));

return async_context_wrap;

}

static

napi_value Init(napi_env env, napi_value exports) {

napi_value fn;

NODE_API_CALL(env, napi_create_function(

// NOLINTNEXTLINE (readability/null_usage)

env, NULL, NAPI_AUTO_LENGTH, MakeCallback, NULL, &fn));

NODE_API_CALL(env, napi_set_named_property(env, exports, "makeCallback", fn));

NODE_API_CALL(env, napi_create_function(

// NOLINTNEXTLINE (readability/null_usage)

env, NULL, NAPI_AUTO_LENGTH, CreateAsyncResource, NULL, &fn));

NODE_API_CALL(env, napi_set_named_property(

env, exports, "createAsyncResource", fn));

NODE_API_CALL(env, napi_create_function(

// NOLINTNEXTLINE (readability/null_usage)

env, NULL, NAPI_AUTO_LENGTH, DestroyAsyncResource, NULL, &fn));

NODE_API_CALL(env, napi_set_named_property(

env, exports, "destroyAsyncResource", fn));

return exports;

}

NAPI_MODULE(NODE_GYP_MODULE_NAME, Init)

node/test/node-api/test_async_context/binding.c at main · fetchapi/node · GitHub

Expand file tree

Breadcrumbs

binding.c

Latest commit

History

Breadcrumbs

binding.c

File metadata and controls