autobee 2.5.0 → 2.5.1

package/index.js

@@ -1098,6 +1098,7 @@ module.exports = class Autobee extends ReadyResource {
 
     for await (const p of this._servableRequests(standing)) {
       approving.push(fetchApproval.call(this, p.key, p.amount, approvals))
+      if (approving.length === 32) break
     }
 
     if (!approving.length) return null
@@ -1179,6 +1180,10 @@ module.exports = class Autobee extends ReadyResource {
       return true
     }
 
+    // not added and not applicable optimistically: nothing can apply this
+    // batch now, so drop the writer for this drain instead of reselecting it
+    // (a refresh re-adds it once it has been added or has new blocks)
+    w.removePending()
     return true
   }
 
@@ -1190,19 +1195,19 @@ module.exports = class Autobee extends ReadyResource {
     if (t.view) this._workingBee.move(t.view)
 
     for (const b of t.tip) {
-      let failed = true
+      const optimistic = b[0].optimistic
+      let accepted = false
       try {
-        if (await this.system.canApply(b[0].key, true)) {
-          await this._applyBatch(b, true)
-          failed = false
-        }
-      } catch {}
+        accepted = await this._applyBatch(b, optimistic)
+      } catch (err) {
+        if (!optimistic) throw err
+      }
 
       // only check if batch was successful
       if (b !== batch) continue
 
-      const w = failed ? null : await this.system.get(b[0].key)
-      if (!w || w.length < b[0].length) {
+      // declined: apply threw, or never called addWriter/ackWriter for this writer
+      if (!accepted) {
         this._workingBee.move(rollbackView)
         this.system.bee.move(rollbackSystem)
         await this.system.reset()
@@ -1289,10 +1294,16 @@ module.exports = class Autobee extends ReadyResource {
     if (this._hasApply && (await this.system.canApply(batch[0].key, optimistic))) {
       this.stats.applies++
       this._host.applying = batch
-      await this._workingView.apply(userBatch)
-      this._host.applying = null
+      try {
+        await this._workingView.apply(userBatch)
+      } finally {
+        this._host.applying = null
+      }
     }
 
+    // read before flush clears it
+    const accepted = this.system.touched(batch[0].key)
+
     const changed = await this.system.flush(batch, this._workingBee)
 
     if (this.system.promotions.changed) this._prefetchApprovals().catch(safetyCatch)
@@ -1307,6 +1318,8 @@ module.exports = class Autobee extends ReadyResource {
       if (added) await this.writers.add(key)
       else await this.writers.remove(key)
     }
+
+    return accepted
   }
 
   async _storeBoot() {

package/lib/system.js

@@ -86,11 +86,14 @@ module.exports = class Systembee {
   }
 
   async ackWriter(key, { length = 0 } = {}) {
+    const info = await this.get(key, { unflushed: true })
     if (length === 0) {
-      const info = await this.get(key, { unflushed: false })
       length = info ? info.length : 0
     }
-    this.update(key, length, -1, -1, false, false, false, -1, null)
+    // acked but never granted: keep the op (record its length) but record the
+    // writer as removed so nothing else from it is applied
+    const isRemoved = !info || info.isAdded === false
+    this.update(key, length, -1, -1, false, false, isRemoved, -1, null)
   }
 
   async removeWriter(key, { length = 0 } = {}) {
@@ -224,6 +227,11 @@ module.exports = class Systembee {
     this.update(node.key, node.length, node.weight, -1, false, false, false, node.timestamp, null)
   }
 
+  // did this apply add or remove (ack) the writer
+  touched(key) {
+    return this.writers.has(b4a.toString(key, 'hex'))
+  }
+
   async canApply(key, optimistic) {
     const id = b4a.toString(key, 'hex')
     const w = this.writers.get(id)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "autobee",
-  "version": "2.5.0",
+  "version": "2.5.1",
   "description": "Bee based autobase",
   "license": "Apache-2.0",
   "author": "Holepunch Inc.",