From 50e33185fe0da8f4dc4d2f54ce72694fa44ea207 Mon Sep 17 00:00:00 2001
From: theanarkh <theratliter@gmail.com>
Date: Sat, 6 Aug 2022 14:15:57 +0800
Subject: [PATCH] dgram: add dgram send queue info

---
 doc/api/dgram.md                            | 17 ++++++++++++++++
 lib/dgram.js                                |  7 +++++++
 src/udp_wrap.cc                             | 22 ++++++++++++++++++++-
 src/udp_wrap.h                              |  4 ++++
 test/parallel/test-dgram-send-queue-info.js |  9 +++++++++
 5 files changed, 58 insertions(+), 1 deletion(-)
 create mode 100644 test/parallel/test-dgram-send-queue-info.js

diff --git a/doc/api/dgram.md b/doc/api/dgram.md
index df50a7b107513a..e3e03fe2d350ec 100644
--- a/doc/api/dgram.md
+++ b/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
diff --git a/lib/dgram.js b/lib/dgram.js
index 9a11a2287c6155..f3b8d485eb92d6 100644
--- a/lib/dgram.js
+++ b/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', {
diff --git a/src/udp_wrap.cc b/src/udp_wrap.cc
index a6911e2e921b98..2487039f4adb35 100644
--- a/src/udp_wrap.cc
+++ b/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,25 @@ 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
 
diff --git a/src/udp_wrap.h b/src/udp_wrap.h
index 75a123d8fa793e..341c361db9515e 100644
--- a/src/udp_wrap.h
+++ b/src/udp_wrap.h
@@ -150,6 +150,10 @@ 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;
diff --git a/test/parallel/test-dgram-send-queue-info.js b/test/parallel/test-dgram-send-queue-info.js
new file mode 100644
index 00000000000000..459a6269e3577a
--- /dev/null
+++ b/test/parallel/test-dgram-send-queue-info.js
@@ -0,0 +1,9 @@
+'use strict';
+require('../common');
+const assert = require('assert');
+const dgram = require('dgram');
+
+const sock = dgram.createSocket('udp4');
+assert.strictEqual(sock.getSendQueueSize(), 0);
+assert.strictEqual(sock.getSendQueueCount(), 0);
+sock.close();