diff --git a/src/node_external_reference.h b/src/node_external_reference.h index 4ebaa8d27feae8..f51999e996fd06 100644 --- a/src/node_external_reference.h +++ b/src/node_external_reference.h @@ -81,6 +81,7 @@ class ExternalReferenceRegistry { V(types) \ V(uv) \ V(v8) \ + V(zlib) \ V(worker) #if NODE_HAVE_I18N_SUPPORT diff --git a/src/node_zlib.cc b/src/node_zlib.cc index b8733229b0b06b..ec0a8e90f1cbcf 100644 --- a/src/node_zlib.cc +++ b/src/node_zlib.cc @@ -25,6 +25,7 @@ #include "async_wrap-inl.h" #include "env-inl.h" +#include "node_external_reference.h" #include "threadpoolwork-inl.h" #include "util-inl.h" @@ -1266,6 +1267,16 @@ struct MakeClass { env->SetConstructorFunction(target, name, z); } + + static void Make(ExternalReferenceRegistry* registry) { + registry->Register(Stream::New); + registry->Register(Stream::template Write); + registry->Register(Stream::template Write); + registry->Register(Stream::Close); + registry->Register(Stream::Init); + registry->Register(Stream::Params); + registry->Register(Stream::Reset); + } }; void Initialize(Local target, @@ -1283,6 +1294,12 @@ void Initialize(Local target, FIXED_ONE_BYTE_STRING(env->isolate(), ZLIB_VERSION)).Check(); } +void RegisterExternalReferences(ExternalReferenceRegistry* registry) { + MakeClass::Make(registry); + MakeClass::Make(registry); + MakeClass::Make(registry); +} + } // anonymous namespace void DefineZlibConstants(Local target) { @@ -1408,3 +1425,4 @@ void DefineZlibConstants(Local target) { } // namespace node NODE_MODULE_CONTEXT_AWARE_INTERNAL(zlib, node::Initialize) +NODE_MODULE_EXTERNAL_REFERENCE(zlib, node::RegisterExternalReferences)