src: use DoTryWrite() for not-all-Buffer writev()s too

PR-URL: #18019 Reviewed-By: James M Snell <jasnell@gmail.com>
nodejs · Jan 14, 2018 · 02fef8a · 02fef8a
1 parent 20f6aae
commit 02fef8a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/stream_base.cc b/src/stream_base.cc
@@ -192,6 +192,13 @@ int StreamBase::Writev(const FunctionCallbackInfo<Value>& args) {
       offset += str_size;
       bytes += str_size;
     }
+
+    err = DoTryWrite(&buf_list, &count);
+    if (err != 0 || count == 0) {
+      req_wrap->Dispatched();
+      req_wrap->Dispose();
+      goto done;
+    }
   }
 
   err = DoWrite(req_wrap, buf_list, count, nullptr);