Skip to content

Commit e07122b

Browse files
legendecaslegendecas
committed
src: add worker per-isolate binding initialization
Create worker binding templates in the per-isolate initialization.
1 parent ee2d80d commit e07122b

File tree

9 files changed

+80
-23
lines changed

9 files changed

+80
-23
lines changed

‎src/async_wrap-inl.h

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,12 @@ inline v8::MaybeLocal<v8::Value> AsyncWrap::MakeCallback(
8080
return MakeCallback(cb_v.As<v8::Function>(), argc, argv);
8181
}
8282

83+
// static
84+
inline v8::Local<v8::FunctionTemplate> AsyncWrap::GetConstructorTemplate(
85+
Environment* env) {
86+
return GetConstructorTemplate(env->isolate_data());
87+
}
88+
8389
} // namespace node
8490

8591
#endif // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS

‎src/async_wrap.cc

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -334,18 +334,20 @@ void AsyncWrap::SetCallbackTrampoline(const FunctionCallbackInfo<Value>& args) {
334334
}
335335
}
336336

337-
Local<FunctionTemplate> AsyncWrap::GetConstructorTemplate(Environment* env) {
338-
Local<FunctionTemplate> tmpl = env->async_wrap_ctor_template();
337+
Local<FunctionTemplate> AsyncWrap::GetConstructorTemplate(
338+
IsolateData* isolate_data) {
339+
Local<FunctionTemplate> tmpl = isolate_data->async_wrap_ctor_template();
339340
if (tmpl.IsEmpty()) {
340-
Isolate* isolate = env->isolate();
341+
Isolate* isolate = isolate_data->isolate();
341342
tmpl = NewFunctionTemplate(isolate, nullptr);
342-
tmpl->SetClassName(FIXED_ONE_BYTE_STRING(env->isolate(), "AsyncWrap"));
343-
tmpl->Inherit(BaseObject::GetConstructorTemplate(env));
343+
tmpl->SetClassName(
344+
FIXED_ONE_BYTE_STRING(isolate_data->isolate(), "AsyncWrap"));
345+
tmpl->Inherit(BaseObject::GetConstructorTemplate(isolate_data));
344346
SetProtoMethod(isolate, tmpl, "getAsyncId", AsyncWrap::GetAsyncId);
345347
SetProtoMethod(isolate, tmpl, "asyncReset", AsyncWrap::AsyncReset);
346348
SetProtoMethod(
347349
isolate, tmpl, "getProviderType", AsyncWrap::GetProviderType);
348-
env->set_async_wrap_ctor_template(tmpl);
350+
isolate_data->set_async_wrap_ctor_template(tmpl);
349351
}
350352
return tmpl;
351353
}

‎src/async_wrap.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,8 @@ class AsyncWrap : public BaseObject {
138138
static constexpr double kInvalidAsyncId = -1;
139139

140140
static v8::Local<v8::FunctionTemplate> GetConstructorTemplate(
141+
IsolateData* isolate_data);
142+
inline static v8::Local<v8::FunctionTemplate> GetConstructorTemplate(
141143
Environment* env);
142144

143145
static void RegisterExternalReferences(ExternalReferenceRegistry* registry);

‎src/inspector_js_api.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
#include "async_wrap-inl.h"
12
#include "base_object-inl.h"
23
#include "inspector_agent.h"
34
#include "inspector_io.h"

‎src/node_binding.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,6 @@
8282
V(wasi) \
8383
V(wasm_web_api) \
8484
V(watchdog) \
85-
V(worker) \
8685
V(zlib)
8786

8887
#define NODE_BUILTIN_BINDINGS(V) \

‎src/node_binding.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@ static_assert(static_cast<int>(NM_F_LINKED) ==
2424
static_cast<int>(node::ModuleFlags::kLinked),
2525
"NM_F_LINKED != node::ModuleFlags::kLinked");
2626

27-
#define NODE_BUILTIN_PER_ISOLATE_BINDINGS(V) V(builtins)
27+
#define NODE_BUILTIN_PER_ISOLATE_BINDINGS(V) \
28+
V(builtins) \
29+
V(worker)
2830

2931
#define NODE_BINDING_CONTEXT_AWARE_CPP(modname, regfunc, priv, flags) \
3032
static node::node_module _module = { \

‎src/node_worker.cc

Lines changed: 19 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ using v8::MaybeLocal;
3434
using v8::Null;
3535
using v8::Number;
3636
using v8::Object;
37+
using v8::ObjectTemplate;
3738
using v8::ResourceConstraints;
3839
using v8::SealHandleScope;
3940
using v8::String;
@@ -882,19 +883,16 @@ void GetEnvMessagePort(const FunctionCallbackInfo<Value>& args) {
882883
}
883884
}
884885

885-
void InitWorker(Local<Object> target,
886-
Local<Value> unused,
887-
Local<Context> context,
888-
void* priv) {
889-
Environment* env = Environment::GetCurrent(context);
890-
Isolate* isolate = env->isolate();
886+
void InitWorker(IsolateData* isolate_data, Local<FunctionTemplate> target) {
887+
Isolate* isolate = isolate_data->isolate();
888+
Local<ObjectTemplate> proto = target->PrototypeTemplate();
891889

892890
{
893891
Local<FunctionTemplate> w = NewFunctionTemplate(isolate, Worker::New);
894892

895893
w->InstanceTemplate()->SetInternalFieldCount(
896894
Worker::kInternalFieldCount);
897-
w->Inherit(AsyncWrap::GetConstructorTemplate(env));
895+
w->Inherit(AsyncWrap::GetConstructorTemplate(isolate_data));
898896

899897
SetProtoMethod(isolate, w, "startThread", Worker::StartThread);
900898
SetProtoMethod(isolate, w, "stopThread", Worker::StopThread);
@@ -906,23 +904,32 @@ void InitWorker(Local<Object> target,
906904
SetProtoMethod(isolate, w, "loopIdleTime", Worker::LoopIdleTime);
907905
SetProtoMethod(isolate, w, "loopStartTime", Worker::LoopStartTime);
908906

909-
SetConstructorFunction(context, target, "Worker", w);
907+
SetConstructorFunction(isolate, proto, "Worker", w);
910908
}
911909

912910
{
913911
Local<FunctionTemplate> wst = NewFunctionTemplate(isolate, nullptr);
914912

915913
wst->InstanceTemplate()->SetInternalFieldCount(
916914
WorkerHeapSnapshotTaker::kInternalFieldCount);
917-
wst->Inherit(AsyncWrap::GetConstructorTemplate(env));
915+
wst->Inherit(AsyncWrap::GetConstructorTemplate(isolate_data));
918916

919917
Local<String> wst_string =
920918
FIXED_ONE_BYTE_STRING(isolate, "WorkerHeapSnapshotTaker");
921919
wst->SetClassName(wst_string);
922-
env->set_worker_heap_snapshot_taker_template(wst->InstanceTemplate());
920+
isolate_data->set_worker_heap_snapshot_taker_template(
921+
wst->InstanceTemplate());
923922
}
924923

925-
SetMethod(context, target, "getEnvMessagePort", GetEnvMessagePort);
924+
SetMethod(isolate, proto, "getEnvMessagePort", GetEnvMessagePort);
925+
}
926+
927+
void InitWorker(Local<Object> target,
928+
Local<Value> unused,
929+
Local<Context> context,
930+
void* priv) {
931+
Environment* env = Environment::GetCurrent(context);
932+
Isolate* isolate = env->isolate();
926933

927934
target
928935
->Set(env->context(),
@@ -976,5 +983,6 @@ void RegisterExternalReferences(ExternalReferenceRegistry* registry) {
976983
} // namespace node
977984

978985
NODE_BINDING_CONTEXT_AWARE_INTERNAL(worker, node::worker::InitWorker)
986+
NODE_BINDING_PER_ISOLATE_INIT(worker, node::worker::InitWorker)
979987
NODE_BINDING_EXTERNAL_REFERENCE(worker,
980988
node::worker::RegisterExternalReferences)

‎src/util.cc

Lines changed: 27 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,9 +56,13 @@ static std::atomic_int seq = {0}; // Sequence number for diagnostic filenames.
5656
namespace node {
5757

5858
using v8::ArrayBufferView;
59+
using v8::Context;
60+
using v8::FunctionTemplate;
5961
using v8::Isolate;
6062
using v8::Local;
63+
using v8::Object;
6164
using v8::String;
65+
using v8::Template;
6266
using v8::Value;
6367

6468
template <typename T>
@@ -482,14 +486,33 @@ void SetConstructorFunction(Local<v8::Context> context,
482486
context, that, OneByteString(isolate, name), tmpl, flag);
483487
}
484488

485-
void SetConstructorFunction(Local<v8::Context> context,
486-
Local<v8::Object> that,
487-
Local<v8::String> name,
488-
Local<v8::FunctionTemplate> tmpl,
489+
void SetConstructorFunction(Local<Context> context,
490+
Local<Object> that,
491+
Local<String> name,
492+
Local<FunctionTemplate> tmpl,
489493
SetConstructorFunctionFlag flag) {
490494
if (LIKELY(flag == SetConstructorFunctionFlag::SET_CLASS_NAME))
491495
tmpl->SetClassName(name);
492496
that->Set(context, name, tmpl->GetFunction(context).ToLocalChecked()).Check();
493497
}
494498

499+
void SetConstructorFunction(Isolate* isolate,
500+
Local<Template> that,
501+
const char* name,
502+
Local<FunctionTemplate> tmpl,
503+
SetConstructorFunctionFlag flag) {
504+
SetConstructorFunction(
505+
isolate, that, OneByteString(isolate, name), tmpl, flag);
506+
}
507+
508+
void SetConstructorFunction(Isolate* isolate,
509+
Local<Template> that,
510+
Local<String> name,
511+
Local<FunctionTemplate> tmpl,
512+
SetConstructorFunctionFlag flag) {
513+
if (LIKELY(flag == SetConstructorFunctionFlag::SET_CLASS_NAME))
514+
tmpl->SetClassName(name);
515+
that->Set(name, tmpl);
516+
}
517+
495518
} // namespace node

‎src/util.h

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -927,6 +927,20 @@ void SetConstructorFunction(v8::Local<v8::Context> context,
927927
SetConstructorFunctionFlag flag =
928928
SetConstructorFunctionFlag::SET_CLASS_NAME);
929929

930+
void SetConstructorFunction(v8::Isolate* isolate,
931+
v8::Local<v8::Template> that,
932+
const char* name,
933+
v8::Local<v8::FunctionTemplate> tmpl,
934+
SetConstructorFunctionFlag flag =
935+
SetConstructorFunctionFlag::SET_CLASS_NAME);
936+
937+
void SetConstructorFunction(v8::Isolate* isolate,
938+
v8::Local<v8::Template> that,
939+
v8::Local<v8::String> name,
940+
v8::Local<v8::FunctionTemplate> tmpl,
941+
SetConstructorFunctionFlag flag =
942+
SetConstructorFunctionFlag::SET_CLASS_NAME);
943+
930944
} // namespace node
931945

932946
#endif // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS

0 commit comments

Comments
 (0)