diff --git a/src/pipe_wrap.cc b/src/pipe_wrap.cc index 2ae74ec23b..39b21a6fab 100644 --- a/src/pipe_wrap.cc +++ b/src/pipe_wrap.cc @@ -84,6 +84,7 @@ void PipeWrap::Initialize(Handle target) { NODE_SET_PROTOTYPE_METHOD(t, "close", HandleWrap::Close); NODE_SET_PROTOTYPE_METHOD(t, "unref", HandleWrap::Unref); + NODE_SET_PROTOTYPE_METHOD(t, "ref", HandleWrap::Ref); NODE_SET_PROTOTYPE_METHOD(t, "readStart", StreamWrap::ReadStart); NODE_SET_PROTOTYPE_METHOD(t, "readStop", StreamWrap::ReadStop); diff --git a/src/tcp_wrap.cc b/src/tcp_wrap.cc index 5e6f1c2ec1..481838a5a1 100644 --- a/src/tcp_wrap.cc +++ b/src/tcp_wrap.cc @@ -96,6 +96,9 @@ void TCPWrap::Initialize(Handle target) { NODE_SET_PROTOTYPE_METHOD(t, "close", HandleWrap::Close); + NODE_SET_PROTOTYPE_METHOD(t, "ref", HandleWrap::Ref); + NODE_SET_PROTOTYPE_METHOD(t, "unref", HandleWrap::Unref); + NODE_SET_PROTOTYPE_METHOD(t, "readStart", StreamWrap::ReadStart); NODE_SET_PROTOTYPE_METHOD(t, "readStop", StreamWrap::ReadStop); NODE_SET_PROTOTYPE_METHOD(t, "shutdown", StreamWrap::Shutdown); diff --git a/src/timer_wrap.cc b/src/timer_wrap.cc index 4332a68ba1..92964af7a3 100644 --- a/src/timer_wrap.cc +++ b/src/timer_wrap.cc @@ -52,6 +52,8 @@ class TimerWrap : public HandleWrap { constructor->SetClassName(String::NewSymbol("Timer")); NODE_SET_PROTOTYPE_METHOD(constructor, "close", HandleWrap::Close); + NODE_SET_PROTOTYPE_METHOD(constructor, "ref", HandleWrap::Ref); + NODE_SET_PROTOTYPE_METHOD(constructor, "unref", HandleWrap::Unref); NODE_SET_PROTOTYPE_METHOD(constructor, "start", Start); NODE_SET_PROTOTYPE_METHOD(constructor, "stop", Stop); diff --git a/src/udp_wrap.cc b/src/udp_wrap.cc index c5f0aa76da..2e9776b2e6 100644 --- a/src/udp_wrap.cc +++ b/src/udp_wrap.cc @@ -110,6 +110,9 @@ void UDPWrap::Initialize(Handle target) { NODE_SET_PROTOTYPE_METHOD(t, "setBroadcast", SetBroadcast); NODE_SET_PROTOTYPE_METHOD(t, "setTTL", SetTTL); + NODE_SET_PROTOTYPE_METHOD(t, "ref", HandleWrap::Ref); + NODE_SET_PROTOTYPE_METHOD(t, "unref", HandleWrap::Unref); + target->Set(String::NewSymbol("UDP"), Persistent::New(t)->GetFunction()); }