autobee 2.0.0-rc.2 → 2.0.0-rc.4

package/index.js

@@ -105,6 +105,8 @@ module.exports = class Autobee extends ReadyResource {
     this._hasApply = !!handlers.apply
     this._hasUpdate = !!handlers.update
     this._needsUpdate = false
+    this._ackRequired = false
+    this._ackedHeads = new Map()
     this._updateLocalCore = null
     this._host = new ApplyCalls(this)
     this._notifyHandler = null
@@ -484,6 +486,8 @@ module.exports = class Autobee extends ReadyResource {
     // tracked across drain iteration
     this.system.shared = null
 
+    this._ackRequired = false
+
     const changes = this._hasUpdate ? new UpdateChanges(this) : null
     if (changes) changes.track()
 
@@ -503,7 +507,12 @@ module.exports = class Autobee extends ReadyResource {
             break // revaluate conditions...
           }
 
-          if (!(await this._bumpPendingWriters())) break
+          if (await this._bumpPendingWriters()) {
+            this._needsUpdate = true
+            continue
+          }
+
+          if (!this._appendAck()) break
           this._needsUpdate = true
         }
 
@@ -744,6 +753,39 @@ module.exports = class Autobee extends ReadyResource {
     }
   }
 
+  // append a null value node to ack writer
+  _appendAck() {
+    if (!this._ackRequired) return false
+    if (!this.writers.writable) return false
+
+    if (this.writers.localWriter.pending !== null) return false
+
+    const links = this.system.getLinks(this.local.key)
+    const t = Math.max(this._now(), this.system.timestamp)
+
+    let unlinked = false
+    for (const { key, length } of links) {
+      const hex = b4a.toString(key, 'hex')
+      const acked = this._ackedHeads.get(hex) || 0
+      if (length > acked) {
+        unlinked = true
+        break
+      }
+    }
+    if (!unlinked) {
+      this._ackRequired = false
+      return false
+    }
+
+    for (const { key, length } of links) {
+      this._ackedHeads.set(b4a.toString(key, 'hex'), length)
+    }
+
+    this.writers.appendLocal(null, t, { start: 0, end: 0 }, links, false, null)
+    this._ackRequired = false
+    return true
+  }
+
   async _bumpPendingWriters() {
     if (this._catchupMigratedNodes !== null) {
       await this._bumpMigratedWriters()

package/lib/apply-calls.js

@@ -1,4 +1,5 @@
 const ID = require('hypercore-id-encoding')
+const b4a = require('b4a')
 
 class ApplyCalls {
   constructor(auto) {
@@ -39,6 +40,10 @@ class ApplyCalls {
 
   ackWriter(key) {
     if (typeof key === 'string') key = ID.decode(key)
+    // acking an optimistic node means we owe it a linking node
+    if (this.applying && this.applying[0].optimistic && !b4a.equals(key, this.auto.local.key)) {
+      this.auto._ackRequired = true
+    }
     return this.auto.system.ackWriter(key)
   }
 

package/lib/system.js

@@ -13,6 +13,27 @@ const INDEXER_LT = b4a.from([3])
 const EMPTY_HEAD = { length: 0, key: null }
 const EMPTY = b4a.from([])
 
+class SystemSnapshot {
+  constructor(bee) {
+    this.bee = bee
+  }
+
+  async get(key, { timeout } = {}) {
+    const node = await this.bee.get(encoding.encodeSystemWriterKey(key), { timeout })
+    return node !== null ? encoding.decodeSystemWriter(node.key, node.value) : null
+  }
+
+  async has(link) {
+    const node = await this.get(link.key)
+    if (!node || node.length < link.length) return false
+    return true
+  }
+
+  close() {
+    return this.bee.close()
+  }
+}
+
 module.exports = class Systembee {
   constructor(store, name, opts = {}) {
     this.name = name // for debuggin
@@ -230,6 +251,14 @@ module.exports = class Systembee {
     return info ? !info.isRemoved : !!optimistic
   }
 
+  snapshot(head = null) {
+    const bee =
+      head === null
+        ? this.bee.snapshot()
+        : this.bee.checkout({ length: head.length, key: head.key })
+    return new SystemSnapshot(bee)
+  }
+
   async get(key, { unflushed = false, timeout } = {}) {
     const node = await this.bee.get(encoding.encodeSystemWriterKey(key), { timeout })
     const info = node !== null ? encoding.decodeSystemWriter(node.key, node.value) : null

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "autobee",
-  "version": "2.0.0-rc.2",
+  "version": "2.0.0-rc.4",
   "description": "Bee based autobase",
   "license": "Apache-2.0",
   "author": "Holepunch Inc.",