From f09b838408f6c775755daf05572b4b9712342148 Mon Sep 17 00:00:00 2001
From: theanarkh <theratliter@gmail.com>
Date: Wed, 25 Jan 2023 08:47:08 +0800
Subject: [PATCH] src,lib: add constrainedMemory API for process

PR-URL: https://github.com/nodejs/node/pull/46218
Reviewed-By: Yagiz Nizipli <yagiz@nizipli.com>
Reviewed-By: Darshan Sen <raisinten@gmail.com>
---
 doc/api/process.md                             | 18 ++++++++++++++++++
 lib/internal/bootstrap/node.js                 |  1 +
 src/node_process_methods.cc                    |  9 +++++++++
 .../test-process-constrained-memory.js         | 12 ++++++++++++
 4 files changed, 40 insertions(+)
 create mode 100644 test/parallel/test-process-constrained-memory.js

diff --git a/doc/api/process.md b/doc/api/process.md
index 6f81cf33d6a9ae..5b6ae09eda95b1 100644
--- a/doc/api/process.md
+++ b/doc/api/process.md
@@ -1107,6 +1107,23 @@ and [Cluster][] documentation), the `process.connected` property will return
 Once `process.connected` is `false`, it is no longer possible to send messages
 over the IPC channel using `process.send()`.
 
+## `process.constrainedMemory()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+> Stability: 1 - Experimental
+
+* {number|undefined}
+
+Gets the amount of memory available to the process (in bytes) based on
+limits imposed by the OS. If there is no such constraint, or the constraint
+is unknown, `undefined` is returned.
+
+See [`uv_get_constrained_memory`][uv_get_constrained_memory] for more
+information.
+
 ## `process.cpuUsage([previousValue])`
 
 <!-- YAML
@@ -3893,6 +3910,7 @@ cases:
 [process_warning]: #event-warning
 [report documentation]: report.md
 [terminal raw mode]: tty.md#readstreamsetrawmodemode
+[uv_get_constrained_memory]: https://docs.libuv.org/en/v1.x/misc.html#c.uv_get_constrained_memory
 [uv_rusage_t]: https://docs.libuv.org/en/v1.x/misc.html#c.uv_rusage_t
 [wikipedia_major_fault]: https://en.wikipedia.org/wiki/Page_fault#Major
 [wikipedia_minor_fault]: https://en.wikipedia.org/wiki/Page_fault#Minor
diff --git a/lib/internal/bootstrap/node.js b/lib/internal/bootstrap/node.js
index 81b441a554e345..1156bcb3150fb5 100644
--- a/lib/internal/bootstrap/node.js
+++ b/lib/internal/bootstrap/node.js
@@ -200,6 +200,7 @@ const rawMethods = internalBinding('process_methods');
   process.cpuUsage = wrapped.cpuUsage;
   process.resourceUsage = wrapped.resourceUsage;
   process.memoryUsage = wrapped.memoryUsage;
+  process.constrainedMemory = rawMethods.constrainedMemory;
   process.kill = wrapped.kill;
   process.exit = wrapped.exit;
 
diff --git a/src/node_process_methods.cc b/src/node_process_methods.cc
index ba983dccfdf38c..eecd61d0647f5a 100644
--- a/src/node_process_methods.cc
+++ b/src/node_process_methods.cc
@@ -206,6 +206,13 @@ static void MemoryUsage(const FunctionCallbackInfo<Value>& args) {
           : static_cast<double>(array_buffer_allocator->total_mem_usage());
 }
 
+static void GetConstrainedMemory(const FunctionCallbackInfo<Value>& args) {
+  uint64_t value = uv_get_constrained_memory();
+  if (value != 0) {
+    args.GetReturnValue().Set(static_cast<double>(value));
+  }
+}
+
 void RawDebug(const FunctionCallbackInfo<Value>& args) {
   CHECK(args.Length() == 1 && args[0]->IsString() &&
         "must be called with a single string");
@@ -577,6 +584,7 @@ static void Initialize(Local<Object> target,
 
   SetMethod(context, target, "umask", Umask);
   SetMethod(context, target, "memoryUsage", MemoryUsage);
+  SetMethod(context, target, "constrainedMemory", GetConstrainedMemory);
   SetMethod(context, target, "rss", Rss);
   SetMethod(context, target, "cpuUsage", CPUUsage);
   SetMethod(context, target, "resourceUsage", ResourceUsage);
@@ -607,6 +615,7 @@ void RegisterExternalReferences(ExternalReferenceRegistry* registry) {
   registry->Register(Umask);
   registry->Register(RawDebug);
   registry->Register(MemoryUsage);
+  registry->Register(GetConstrainedMemory);
   registry->Register(Rss);
   registry->Register(CPUUsage);
   registry->Register(ResourceUsage);
diff --git a/test/parallel/test-process-constrained-memory.js b/test/parallel/test-process-constrained-memory.js
new file mode 100644
index 00000000000000..2bd94a6749e56b
--- /dev/null
+++ b/test/parallel/test-process-constrained-memory.js
@@ -0,0 +1,12 @@
+'use strict';
+require('../common');
+const assert = require('assert');
+const { Worker } = require('worker_threads');
+const constrainedMemory = process.constrainedMemory();
+if (constrainedMemory !== undefined) {
+  assert(constrainedMemory > 0);
+}
+if (!process.env.isWorker) {
+  process.env.isWorker = true;
+  new Worker(__filename);
+}