bajo-extra 0.2.9 → 0.2.11

package/bajo/helper/fetch-and-save.js

@@ -1,47 +1,52 @@
-async function fetchAndSave ({ source = {}, converter, coll, current = {}, options = {} } = {}) {
-  const { setImmediate, importPkg, spinner, print } = this.bajo.helper
-  const { isEmpty, isFunction } = await importPkg('lodash-es')
-  const { fetch } = this.bajoExtra.helper
-  const { recordCreate, recordFind, recordUpdate, validationErrorMessage } = this.bajoDb.helper
-  const spin = spinner({ showCounter: true }).start('Fetching starts...')
-  const resp = await fetch(source.url, source.options ?? {})
-  if (isEmpty(resp)) spin.fatal('No result from server, aborted!')
-  if (source.abort) {
-    const aborted = await source.abort.call(this, resp)
-    if (aborted) spin.fatal(aborted)
+async function handler (rec, bulk) {
+  const { importPkg } = this.bajo.helper
+  const { isFunction, set } = await importPkg('lodash-es')
+  const { recordCreate, recordFind, recordUpdate } = this.bajoDb.helper
+  const save = bulk.save ?? {}
+  const current = save.current ?? {}
+  let existing
+  let record
+  let method
+  save.checkUnique = save.checkUnique ?? 'id'
+  if (['unique', 'upsert'].includes(save.mode)) {
+    const query = isFunction(save.checkUnique) ? await save.checkUnique.call(this, rec, save) : set({}, save.checkUnique, rec[save.checkUnique])
+    const resp = await recordFind(save.coll, { query, limit: 1 }, { skipCache: true })
+    if (resp.length > 0) existing = resp[0]
   }
-  let count = 0
-  const iterator = isFunction(source.iterator) ? await source.iterator.call(this, resp) : resp[source.iterator]
-  spin.setText('Got %d records, processing...', iterator.length)
-  for (let r of iterator) {
-    await setImmediate()
-    if (converter) r = await converter.call(this, r, options)
-    if (isEmpty(r)) continue
-    try {
-      await recordCreate(coll, r)
-      if (current.coll && current.query) {
-        const query = await current.query.call(this, r)
-        const recs = await recordFind(current.coll, { query }, { skipCache: true })
-        const rc = current.converter ? await current.converter.call(this, r, options) : r
-        if (rc) {
-          if (recs.length > 0) {
-            const id = recs[0].id
-            await recordUpdate(current.coll, id, rc)
-          } else {
-            await recordCreate(current.coll, rc)
-          }
-        }
+  if (existing) {
+    if (save.mode === 'upsert') {
+      const body = save.updateConverter ? await save.updateConverter.call(this, rec, save) : rec
+      record = await recordUpdate(save.coll, existing.id, body)
+      method = 'updated'
+    } else {
+      method = 'skipped'
+    }
+  } else {
+    record = await recordCreate(save.coll, rec)
+    method = 'created'
+  }
+  if (record && current.coll && current.query) {
+    const query = await current.query.call(this, { body: rec, record, opts: save })
+    const recs = await recordFind(current.coll, { query }, { skipCache: true })
+    const rc = current.converter ? await current.converter.call(this, { body: rec, record, opts: save }) : rec
+    if (rc) {
+      if (recs.length > 0) {
+        const id = recs[0].id
+        await recordUpdate(current.coll, id, rc)
+      } else {
+        await recordCreate(current.coll, rc)
       }
-      if (options.printCount && (count % options.printCount === 0)) print.succeed(`[${spin.getElapsed()}] Batch line %d/%d`, count, iterator.length)
-      else if (!spin.opts.isLog) spin.setText('Record %d/%d...', count, iterator.length)
-      count++
-    } catch (err) {
-      console.log(err)
-      spin.setText(validationErrorMessage(err) + ', continue')
     }
   }
-  spin.info(`${count}/${iterator.length} records processed`)
-  spin.succeed('Done!')
+  return method
+}
+
+async function fetchAndSave ({ url, bulk, save = {}, opts = {} } = {}) {
+  const { importPkg } = this.bajo.helper
+  const { fetchBulk } = this.bajoExtra.helper
+  const { merge } = await importPkg('lodash-es')
+  merge(bulk, { handler, save })
+  await fetchBulk(url, bulk, opts)
 }
 
 export default fetchAndSave

package/bajo/helper/fetch-bulk.js

@@ -0,0 +1,77 @@
+async function fetching ({ url, opts, bulk, spin }) {
+  const { setImmediate, importPkg, print } = this.bajo.helper
+  const { isEmpty, isFunction, has } = await importPkg('lodash-es')
+  const { validationErrorMessage } = this.bajoDb.helper
+  const { fetch } = this.bajoExtra.helper
+  const resp = await fetch(url, opts ?? {})
+  if (isEmpty(resp)) {
+    spin.fatal('No result from server, aborted!')
+    return -1
+  }
+  if (bulk.abort) {
+    const aborted = await bulk.abort.call(this, resp)
+    if (aborted) {
+      spin.fatal(aborted)
+      return -1
+    }
+  }
+  let count = 0
+  const stat = { created: 0, updated: 0, skipped: 0, error: 0 }
+  bulk.dataKey = bulk.dataKey ?? 'data'
+  const data = isFunction(bulk.dataKey) ? await bulk.dataKey.call(this, resp) : resp[bulk.dataKey]
+  if (data.length === 0) {
+    print.warn('No records to process, abort')
+    return 0
+  }
+  spin.setText('Got %d records, processing...', data.length)
+  for (let r of data) {
+    await setImmediate()
+    if (bulk.converter) r = await bulk.converter.call(this, r, bulk)
+    if (isEmpty(r)) {
+      stat.skipped++
+      continue
+    }
+    try {
+      const result = await bulk.handler.call(this, r, bulk)
+      if (result && has(stat, result)) stat[result]++
+      if (bulk.printCount && bulk.printCount < count && (count % bulk.printCount === 0)) print.succeed('[%s] Processed %d/%d', spin.getElapsed(), count, data.length)
+      else if (!spin.opts.isLog) spin.setText('Record %d/%d...', count, data.length)
+      count++
+    } catch (err) {
+      console.log(err)
+      spin.setText(validationErrorMessage(err) + ', continue')
+    }
+  }
+  print.succeed('[%s] %d/%d records processed', spin.getElapsed(), count, data.length)
+  print.succeed('[%s] Created: %d, Updated: %d, Skipped: %d', spin.getElapsed(), stat.created, stat.updated, stat.skipped)
+  return data.length
+}
+
+async function fetchBulk (url, bulk = {}, opts = {}) {
+  const { print, spinner, importPkg, error } = this.bajo.helper
+  const { isFunction } = await importPkg('lodash-es')
+  opts.params = opts.params ?? {}
+  bulk.maxStep = bulk.maxStep ?? 0
+  if (!isFunction(bulk.handler)) throw error('A function handler must be provided')
+  if (bulk.paramsIncFn && isFunction(bulk.ParamsFn)) {
+    print.info('Bulk fetch starting')
+    const spin = spinner({ showCounter: true }).start('Fetching starts...')
+    let step = 1
+    for (;;) {
+      print.info('[%s] Batch #%d', spin.getElapsed(), step)
+      const newOpts = await bulk.paramsIncFn.call(this, { url, bulk, opts })
+      if (newOpts) opts = newOpts
+      const length = await fetching.call(this, { url, bulk, opts, spin })
+      if (length === 0 || (bulk.maxStep > 0 && step >= bulk.maxStep)) {
+        print.info('All done!')
+        break
+      }
+      step++
+    }
+  } else {
+    const spin = spinner({ showCounter: true }).start('Fetching starts...')
+    await fetching.call(this, { url, bulk, opts, spin })
+  }
+}
+
+export default fetchBulk

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bajo-extra",
-  "version": "0.2.9",
+  "version": "0.2.11",
   "description": "Extra package for Bajo Framework",
   "main": "index.js",
   "scripts": {
@@ -32,6 +32,7 @@
     "email-addresses": "^5.0.0",
     "fast-jwt": "^3.2.0",
     "fast-xml-parser": "^4.3.2",
+    "littlehash": "^1.0.1",
     "ndjson-csv-xlsx": "^1.1.1",
     "performant-array-to-tree": "^1.11.0",
     "query-string": "^8.1.0",