budibase/packages/server/src/threads/index.js

const workerFarm = require("worker-farm")
const env = require("../environment")

const ThreadType = {
  QUERY: "query",
  AUTOMATION: "automation",
}

function typeToFile(type) {
  let filename = null
  switch (type) {
    case ThreadType.QUERY:
      filename = "./query"
      break
    case ThreadType.AUTOMATION:
      filename = "./automation"
      break
    default:
      throw "Unknown thread type"
  }
  return require.resolve(filename)
}

class Thread {
  constructor(type, opts = { timeoutMs: null, count: 1 }) {
    this.type = type
    this.count = opts.count ? opts.count : 1
    this.disableThreading =
      env.isTest() ||
      env.DISABLE_THREADING ||
      this.count === 0 ||
      env.isInThread()
    if (!this.disableThreading) {
      const workerOpts = {
        autoStart: true,
        maxConcurrentWorkers: this.count,
      }
      if (opts.timeoutMs) {
        workerOpts.maxCallTime = opts.timeoutMs
      }
      this.workers = workerFarm(workerOpts, typeToFile(type))
    }
  }

  run(data) {
    return new Promise((resolve, reject) => {
      let fncToCall
      // if in test then don't use threading
      if (this.disableThreading) {
        fncToCall = require(typeToFile(this.type))
      } else {
        fncToCall = this.workers
      }
      fncToCall(data, (err, response) => {
        if (err) {
          reject(err)
        } else {
          resolve(response)
        }
      })
    })
  }
}

module.exports.Thread = Thread
module.exports.ThreadType = ThreadType