benchmarks/versus-external-pools/functions/function-to-bench.js

   1 /**
   2  * The worker function to execute during pools benchmarks.
   3  * NOTE: This function requires to be self-contained, thread-safe and re-entrant.
   4  * @param {*} data The worker data.
   5  * @returns {*} The result.
   6  */
   7 function functionToBench (data) {
   8   const crypto = require('crypto')
   9   const fs = require('fs')
  10   const TaskTypes = {
  11     CPU_INTENSIVE: 'CPU_INTENSIVE',
  12     IO_INTENSIVE: 'IO_INTENSIVE'
  13   }
  14   data = data || {}
  15   data.taskType = data.taskType || TaskTypes.CPU_INTENSIVE
  16   data.taskSize = data.taskSize || 5000
  17   const baseDirectory = `/tmp/poolifier-benchmarks/${crypto.randomInt(
  18     281474976710655
  19   )}`
  20   switch (data.taskType) {
  21     case TaskTypes.CPU_INTENSIVE:
  22       // CPU intensive task
  23       for (let i = 0; i < data.taskSize; i++) {
  24         const o = {
  25           a: i
  26         }
  27         JSON.stringify(o)
  28       }
  29       return { ok: 1 }
  30     case TaskTypes.IO_INTENSIVE:
  31       // IO intensive task
  32       if (fs.existsSync(baseDirectory) === true) {
  33         fs.rmSync(baseDirectory, { recursive: true })
  34       }
  35       fs.mkdirSync(baseDirectory, { recursive: true })
  36       for (let i = 0; i < data.taskSize; i++) {
  37         const filePath = `${baseDirectory}/${i}`
  38         fs.writeFileSync(filePath, i.toString(), {
  39           encoding: 'utf8',
  40           flag: 'a'
  41         })
  42         fs.readFileSync(filePath, 'utf8')
  43       }
  44       fs.rmSync(baseDirectory, { recursive: true })
  45       return { ok: 1 }
  46     default:
  47       throw new Error(`Unknown task type: ${data.taskType}`)
  48   }
  49 }
  50
  51 module.exports = functionToBench