dgram: add dgram send queue info

Author: theanarkh

doc/api/dgram.md

@@ -461,6 +461,23 @@ added: v8.7.0
 
 This method throws [`ERR_SOCKET_BUFFER_SIZE`][] if called on an unbound socket.
 
+### `socket.getSendQueueSize()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* Returns: {number} Number of bytes queued for sending.
+
+### `socket.getSendQueueCount()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* Returns: {number} Number of send requests currently in the queue awaiting
+  to be processed.
+
 ### `socket.ref()`
 
 <!-- YAML

lib/dgram.js

@@ -976,6 +976,13 @@ Socket.prototype.getSendBufferSize = function() {
   return bufferSize(this, 0, SEND_BUFFER);
 };
 
+Socket.prototype.getSendQueueSize = function() {
+  return this[kStateSymbol].handle.getSendQueueSize();
+};
+
+Socket.prototype.getSendQueueCount = function() {
+  return this[kStateSymbol].handle.getSendQueueCount();
+};
 
 // Deprecated private APIs.
 ObjectDefineProperty(Socket.prototype, '_handle', {

src/udp_wrap.cc

@@ -181,7 +181,8 @@ void UDPWrap::Initialize(Local<Object> target,
   SetProtoMethod(isolate, t, "setMulticastLoopback", SetMulticastLoopback);
   SetProtoMethod(isolate, t, "setBroadcast", SetBroadcast);
   SetProtoMethod(isolate, t, "setTTL", SetTTL);
-  SetProtoMethod(isolate, t, "bufferSize", BufferSize);
+  SetProtoMethod(isolate, t, "getSendQueueSize", GetSendQueueSize);
+  SetProtoMethod(isolate, t, "getSendQueueCount", GetSendQueueCount);
 
   t->Inherit(HandleWrap::GetConstructorTemplate(env));
 
@@ -783,6 +784,23 @@ MaybeLocal<Object> UDPWrap::Instantiate(Environment* env,
   return env->udp_constructor_function()->NewInstance(env->context());
 }
 
+void UDPWrap::GetSendQueueSize(const FunctionCallbackInfo<Value>& args) {
+  UDPWrap* wrap;
+  ASSIGN_OR_RETURN_UNWRAP(
+      &wrap, args.Holder(), args.GetReturnValue().Set(UV_EBADF));
+
+  size_t size = uv_udp_get_send_queue_size(&wrap->handle_);
+  args.GetReturnValue().Set(static_cast<double>(size));
+}
+
+void UDPWrap::GetSendQueueCount(const FunctionCallbackInfo<Value>& args) {
+  UDPWrap* wrap;
+  ASSIGN_OR_RETURN_UNWRAP(
+      &wrap, args.Holder(), args.GetReturnValue().Set(UV_EBADF));
+
+  size_t count = uv_udp_get_send_queue_count(&wrap->handle_);
+  args.GetReturnValue().Set(static_cast<double>(count));
+}
 
 }  // namespace node
 

src/udp_wrap.h

@@ -150,6 +150,9 @@ class UDPWrap final : public HandleWrap,
   static void SetBroadcast(const v8::FunctionCallbackInfo<v8::Value>& args);
   static void SetTTL(const v8::FunctionCallbackInfo<v8::Value>& args);
   static void BufferSize(const v8::FunctionCallbackInfo<v8::Value>& args);
+  static void GetSendQueueSize(const v8::FunctionCallbackInfo<v8::Value>& args);
+  static void GetSendQueueCount(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
 
   // UDPListener implementation
   uv_buf_t OnAlloc(size_t suggested_size) override;

test/parallel/test-dgram-send-queue-info.js

@@ -0,0 +1,33 @@
+'use strict';
+const common = require('../common');
+const assert = require('assert');
+const dgram = require('dgram');
+const cp = require('child_process');
+
+const socket = dgram.createSocket('udp4');
+assert.strictEqual(socket.getSendQueueSize(), 0);
+assert.strictEqual(socket.getSendQueueCount(), 0);
+
+const CODE = `
+  const server = dgram.createSocket('udp4');
+  const client = dgram.createSocket('udp4');
+  server.bind(0, () => {
+    client.connect(server.address().port, () => {
+      const data = "hello";
+      client.send(data);
+      client.send(data);
+      assert.strictEqual(client.getSendQueueSize(), data.length * 2);
+      assert.strictEqual(client.getSendQueueCount(), 2);
+      client.close();
+      server.close();
+    });
+  });
+`;
+
+const proc = cp.spawn(process.execPath,
+                      [ '--test-udp-no-try-send',
+                        '-e', CODE ]);
+
+proc.once('exit', common.mustCall((code) => {
+  assert(code === 0);
+}));