@nxtedition/lib 26.0.26 → 26.0.27

package/app.js

@@ -38,6 +38,9 @@ import makeUnderPressure from './under-pressure.js'
 import undici from '@nxtedition/undici'
 import { isPrimary } from 'node:cluster'
 import { nice } from '@napi-rs/nice'
+import path from 'node:path'
+
+import { sched_setaffinity, sched_getaffinity } from '@nxtedition/sched'
 
 export function makeApp(appConfig, onTerminate) {
   let ds
@@ -310,6 +313,55 @@ export function makeApp(appConfig, onTerminate) {
     }
   }
 
+  if (appConfig.affinity != null && process.platform === 'linux') {
+    try {
+      const nodes = []
+      for (const entry of fs.readdirSync('/sys/devices/system/node')) {
+        if (!entry.startsWith('node')) {
+          continue
+        }
+
+        const cpulist = fs
+          .readFileSync(path.join('/sys/devices/system/node', entry, 'cpulist'), 'utf8')
+          .trim()
+        const cpus = []
+        for (const part of cpulist.split(',')) {
+          if (part.includes('-')) {
+            const [start, end] = part.split('-').map(Number)
+            for (let i = start; i <= end; i++) cpus.push(i)
+          } else {
+            cpus.push(Number(part))
+          }
+        }
+        nodes.push(cpus)
+      }
+
+      let indices
+      if (typeof appConfig.affinity === 'number') {
+        indices = [appConfig.affinity]
+      } else if (Array.isArray(appConfig.affinity)) {
+        indices = appConfig.affinity
+      } else {
+        indices = [Math.floor(Math.random() * nodes.length)]
+      }
+
+      if (indices?.length > 0) {
+        let affinity = []
+        for (const index of indices) {
+          affinity = fp.union(affinity, nodes[index % nodes.length])
+        }
+
+        const next = affinity
+        const prev = sched_getaffinity(0)
+
+        sched_setaffinity(0, fp.intersection(prev, next))
+        logger.debug({ data: { prev, next, nodes, affinity } }, 'sched_setaffinity succeeded')
+      }
+    } catch (err) {
+      logger.error({ err }, 'sched_setaffinity failed')
+    }
+  }
+
   if (appConfig.toobusy) {
     const resolution = appConfig.toobusy.resolution ?? 10
     const interval = appConfig.toobusy.interval ?? 500

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nxtedition/lib",
-  "version": "26.0.26",
+  "version": "26.0.27",
   "license": "MIT",
   "author": "Robert Nagy <robert.nagy@boffins.se>",
   "type": "module",
@@ -66,7 +66,9 @@
     "@aws-sdk/client-s3": "3.873.0",
     "@elastic/elasticsearch": "^8.17.1",
     "@elastic/transport": "^8.9.3",
+    "@napi-rs/nice": "^1.1.1",
     "@nxtedition/nxt-undici": "^6.4.0",
+    "@nxtedition/sched": "^1.0.1",
     "@smithy/node-http-handler": "^4.1.1",
     "acorn": "^8.15.0",
     "astring": "^1.9.0",