audit is an iterator

Author: mafintosh

lib/audit.js

@@ -1,29 +1,14 @@
-module.exports = async function audit (store, { dryRun = false } = {}) {
-  const stats = { cores: 0, skipped: 0, rootless: 0, dropped: 0 }
-
+module.exports = async function * audit (store, { dryRun = false } = {}) {
   for await (const { discoveryKey } of store.storage.createCoreStream()) {
-    stats.cores++
-
     const core = store.get({ discoveryKey, active: false })
     await core.ready()
 
-    const audit = await core.core.audit({ dryRun })
+    yield { discoveryKey, key: core.key, audit: await core.core.audit({ dryRun }) }
 
     try {
       await core.close()
     } catch {
       // ignore if failed, we are auditing...
     }
-
-    if (audit === null || audit.corrupt) {
-      stats.rootless++
-      continue
-    }
-
-    if (audit.droppedTreeNodes || audit.droppedBlocks || audit.droppedBits) {
-      stats.dropped++
-    }
   }
-
-  return stats
 }

test/basic.js

@@ -240,18 +240,21 @@ test('audit', async function (t) {
   const d = store.get({ name: 'another' })
 
   for (let i = 0; i < 100; i++) {
-    await a.append(i.toString())
-    await b.append(i.toString())
-    await c.append(i.toString())
+    if (i < 20) await a.append(i.toString())
+    if (i < 40) await b.append(i.toString())
+    if (i < 80) await c.append(i.toString())
     await d.append(i.toString())
   }
 
-  const result = await store.audit()
+  let n = 0
+  for await (const { discoveryKey, audit } of store.audit()) {
+    n++
+    if (audit.droppedBits || audit.droppedBlocks || audit.droppedTreeNodes || audit.corrupt) {
+      t.fail('bad core')
+    }
+  }
 
-  t.is(result.cores, 4)
-  t.is(result.skipped, 0)
-  t.is(result.rootless, 0)
-  t.is(result.dropped, 0)
+  t.is(n, 4)
 
   await a.close()
   await b.close()