src: add option to set cpu affinity of mainthread

src/node_main_instance.cc

@@ -19,6 +19,10 @@
 #if HAVE_INSPECTOR
 #include "inspector/worker_inspector.h"  // ParentInspectorHandle
 #endif
+#if defined(__linux__)
+#include <sched.h>
+#include <unistd.h>
+#endif
 
 namespace node {
 
@@ -134,6 +138,18 @@ int NodeMainInstance::Run() {
 
 void NodeMainInstance::Run(int* exit_code, Environment* env) {
   if (*exit_code == 0) {
+    if (per_process::cli_options->set_mainthread_cpu_affinity) {
+#if defined(__linux__)
+      cpu_set_t mask;
+      int c = sched_getcpu();
+      if (c >= 0) {
+        CPU_ZERO(&mask);
+        CPU_SET(c, &mask);
+        sched_setaffinity(0, sizeof(cpu_set_t), &mask);
+      }
+#endif
+    }
+
     LoadEnvironment(env, StartExecutionCallback{});
 
     *exit_code = SpinEventLoop(env).FromMaybe(1);

src/node_options.cc

@@ -908,6 +908,13 @@ PerProcessOptionsParser::PerProcessOptionsParser(
             &PerProcessOptions::trace_sigint,
             kAllowedInEnvironment);
 
+  AddOption("--set-mainthread-cpu-affinity",
+            "enable setting mainthread's cpu affinity to the currently "
+            "running cpu before entering event loop, implemented on "
+            "Linux OS only",
+            &PerProcessOptions::set_mainthread_cpu_affinity,
+            kAllowedInEnvironment);
+
   Insert(iop, &PerProcessOptions::get_per_isolate_options);
 
   AddOption("--node-memory-debug",

src/node_options.h

@@ -283,6 +283,7 @@ class PerProcessOptions : public Options {
   // TODO(addaleax): Some of these could probably be per-Environment.
   std::string use_largepages = "off";
   bool trace_sigint = false;
+  bool set_mainthread_cpu_affinity = false;
   std::vector<std::string> cmdline;
 
   inline PerIsolateOptions* get_per_isolate_options();