child_process: spawn().ref() and spawn().unref()

nodejs · Jun 7, 2012 · 3116522 · 3116522
1 parent c45522d
commit 3116522
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 2 deletions.
diff --git a/lib/child_process.js b/lib/child_process.js
@@ -845,3 +845,13 @@ ChildProcess.prototype.kill = function(sig) {
     }
   }
 };
+
+
+ChildProcess.prototype.ref = function() {
+  if (this._handle) this._handle.ref();
+};
+
+
+ChildProcess.prototype.unref = function() {
+  if (this._handle) this._handle.unref();
+};
diff --git a/src/handle_wrap.cc b/src/handle_wrap.cc
@@ -50,8 +50,18 @@ void HandleWrap::Initialize(Handle<Object> target) {
 }
 
 
-// This function is used only for process.stdout. It's put here instead of
-// in TTYWrap because here we have access to the Close binding.
+Handle<Value> HandleWrap::Ref(const Arguments& args) {
+  HandleScope scope;
+
+  UNWRAP(HandleWrap)
+
+  uv_ref(wrap->handle__);
+  wrap->unref_ = false;
+
+  return v8::Undefined();
+}
+
+
 Handle<Value> HandleWrap::Unref(const Arguments& args) {
   HandleScope scope;
 

diff --git a/src/handle_wrap.h b/src/handle_wrap.h
@@ -50,6 +50,7 @@ class HandleWrap {
   public:
     static void Initialize(v8::Handle<v8::Object> target);
     static v8::Handle<v8::Value> Close(const v8::Arguments& args);
+    static v8::Handle<v8::Value> Ref(const v8::Arguments& args);
     static v8::Handle<v8::Value> Unref(const v8::Arguments& args);
 
   protected:

diff --git a/src/process_wrap.cc b/src/process_wrap.cc
@@ -67,6 +67,9 @@ class ProcessWrap : public HandleWrap {
     NODE_SET_PROTOTYPE_METHOD(constructor, "spawn", Spawn);
     NODE_SET_PROTOTYPE_METHOD(constructor, "kill", Kill);
 
+    NODE_SET_PROTOTYPE_METHOD(constructor, "ref", HandleWrap::Ref);
+    NODE_SET_PROTOTYPE_METHOD(constructor, "unref", HandleWrap::Unref);
+
     target->Set(String::NewSymbol("Process"), constructor->GetFunction());
   }