Function action

test/unit/db-concurrency.spec.js:56–117

(dbName, op, ...args)

  }

  async function action (dbName, op, ...args) {
    if (op === 'compactDatafile') return
    if (!tables.includes(dbName)) {
      throw new Error(`Table ${dbName} does not exist`)
    }
    const db = getDatabase(dbName)
    if (op === 'find') {
      const stmt = db.prepare(`SELECT * FROM \`${dbName}\``)
      const rows = stmt.all()
      return (rows || []).map(row => {
        const r = JSON.parse(row.data || '{}')
        return { ...r, _id: row._id }
      })
    } else if (op === 'findOne') {
      const query = args[0] || {}
      const stmt = db.prepare(`SELECT * FROM \`${dbName}\` WHERE _id = ? LIMIT 1`)
      const row = stmt.get(query._id)
      if (!row) return null
      const r = JSON.parse(row.data || '{}')
      return { ...r, _id: row._id }
    } else if (op === 'insert') {
      const inserts = Array.isArray(args[0]) ? args[0] : [args[0]]
      const inserted = []
      for (const doc of inserts) {
        const _id = doc._id || doc.id || uid()
        const copy = { ...doc }
        delete copy._id
        delete copy.id
        const data = JSON.stringify(copy)
        const stmt = db.prepare(`INSERT OR REPLACE INTO \`${dbName}\` (_id, data) VALUES (?, ?)`)
        stmt.run(_id, data)
        inserted.push({ ...doc, _id })
      }
      return Array.isArray(args[0]) ? inserted : inserted[0]
    } else if (op === 'remove') {
      const query = args[0] || {}
      const stmt = db.prepare(`DELETE FROM \`${dbName}\` WHERE _id = ?`)
      const res = stmt.run(query._id)
      return res.changes
    } else if (op === 'update') {
      const query = args[0]
      const updateObj = args[1]
      const options = args[2] || {}
      const { upsert = false } = options
      const qid = query._id || query.id
      const newData = updateObj.$set || updateObj
      const _id = qid
      const copy = { ...newData }
      delete copy._id
      delete copy.id
      const data = JSON.stringify(copy)
      let res
      if (upsert) {
        const stmt = db.prepare(`REPLACE INTO \`${dbName}\` (_id, data) VALUES (?, ?)`)
        res = stmt.run(_id, data)
      } else {
        const stmt = db.prepare(`UPDATE \`${dbName}\` SET data = ? WHERE _id = ?`)
        res = stmt.run(data, qid)