From 9a3b8a8d6b7014e7ce642564487eda69baa5003e Mon Sep 17 00:00:00 2001
From: cjihrig <cjihrig@gmail.com>
Date: Fri, 16 Dec 2022 22:15:23 -0500
Subject: [PATCH] os: add availableParallelism()

This commit exposes uv_available_parallelism() as an alternative
to cpus().length. uv_available_parallelism() is inspired by
Rust's available_parallelism().
---
 doc/api/os.md            | 11 +++++++++++
 lib/os.js                |  3 +++
 src/node_os.cc           |  7 +++++++
 test/parallel/test-os.js |  4 ++++
 4 files changed, 25 insertions(+)

diff --git a/doc/api/os.md b/doc/api/os.md
index e2e702dcea5ef49..c472801f3425dea 100644
--- a/doc/api/os.md
+++ b/doc/api/os.md
@@ -26,6 +26,17 @@ The operating system-specific end-of-line marker.
 * `\n` on POSIX
 * `\r\n` on Windows
 
+## `os.availableParallelism()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* Returns: {integer}
+
+Returns an estimate of the default amount of parallelism a program should use.
+Always returns a non-zero value.
+
 ## `os.arch()`
 
 <!-- YAML
diff --git a/lib/os.js b/lib/os.js
index bff7b275810a4a4..055710f5620ee23 100644
--- a/lib/os.js
+++ b/lib/os.js
@@ -45,6 +45,7 @@ const {
 const { validateInt32 } = require('internal/validators');
 
 const {
+  getAvailableParallelism,
   getCPUs,
   getFreeMem,
   getHomeDirectory: _getHomeDirectory,
@@ -100,6 +101,7 @@ const getOSVersion = () => version;
  */
 const getMachine = () => machine;
 
+getAvailableParallelism[SymbolToPrimitive] = () => getAvailableParallelism();
 getFreeMem[SymbolToPrimitive] = () => getFreeMem();
 getHostname[SymbolToPrimitive] = () => getHostname();
 getOSVersion[SymbolToPrimitive] = () => getOSVersion();
@@ -364,6 +366,7 @@ function userInfo(options) {
 
 module.exports = {
   arch,
+  availableParallelism: getAvailableParallelism,
   cpus,
   endianness,
   freemem: getFreeMem,
diff --git a/src/node_os.cc b/src/node_os.cc
index 57ba05f25e945f7..cf1928521b138e4 100644
--- a/src/node_os.cc
+++ b/src/node_os.cc
@@ -380,6 +380,10 @@ static void GetPriority(const FunctionCallbackInfo<Value>& args) {
   args.GetReturnValue().Set(priority);
 }
 
+static void GetAvailableParallelism(const FunctionCallbackInfo<Value>& args) {
+  unsigned int parallelism = uv_available_parallelism();
+  args.GetReturnValue().Set(parallelism);
+}
 
 void Initialize(Local<Object> target,
                 Local<Value> unused,
@@ -397,6 +401,8 @@ void Initialize(Local<Object> target,
   SetMethod(context, target, "getUserInfo", GetUserInfo);
   SetMethod(context, target, "setPriority", SetPriority);
   SetMethod(context, target, "getPriority", GetPriority);
+  SetMethod(
+      context, target, "getAvailableParallelism", GetAvailableParallelism);
   SetMethod(context, target, "getOSInformation", GetOSInformation);
   target
       ->Set(context,
@@ -417,6 +423,7 @@ void RegisterExternalReferences(ExternalReferenceRegistry* registry) {
   registry->Register(GetUserInfo);
   registry->Register(SetPriority);
   registry->Register(GetPriority);
+  registry->Register(GetAvailableParallelism);
   registry->Register(GetOSInformation);
 }
 
diff --git a/test/parallel/test-os.js b/test/parallel/test-os.js
index cceebe1a6f19d03..f0f55fbe261946b 100644
--- a/test/parallel/test-os.js
+++ b/test/parallel/test-os.js
@@ -255,6 +255,8 @@ if (!common.isIBMi) {
   is.number(os.uptime(), 'uptime');
 }
 
+is.number(+os.availableParallelism, 'availableParallelism');
+is.number(os.availableParallelism(), 'availableParallelism');
 is.number(+os.freemem, 'freemem');
 is.number(os.freemem(), 'freemem');
 
@@ -264,3 +266,5 @@ if (common.isWindows) {
 } else {
   assert.strictEqual(devNull, '/dev/null');
 }
+
+assert.ok(os.availableParallelism() > 0);