prefix-hash-tree 0.0.3 → 0.0.5

package/index.d.ts

@@ -24,6 +24,6 @@ declare module 'prefix-hash-tree' {
 
     searchExact(key: Buffer): Promise<any>
 
-    prefixQuery(prefix: Buffer): Promise<[string, any][]>
+    prefixQuery(prefix: Buffer): Promise<[string, any][] | null>
   }
 }

package/index.js

@@ -27,11 +27,11 @@ class PrefixHashTree {
   _labelHash(nodeLabel) {
     const hash = b4a.allocUnsafe(32)
     sodium.crypto_generichash(hash, b4a.from(`${this.indexID}${nodeLabel}`))
-    return b4a.toBuffer(hash)
+    return hash
   }
 
   async _dhtLookup(nodeLabel) {
-    return await this.getFunc(this._labelHash(nodeLabel))
+    return this.getFunc(this._labelHash(nodeLabel))
   }
 
   async _linearSearch(nodeLabel) {
@@ -67,15 +67,13 @@ class PrefixHashTree {
   }
 
   async _fullSearch(nodeLabel) {
-    let { phtNode, metadata } = await this._binarySearch(nodeLabel)
+    const result = await this._binarySearch(nodeLabel)
 
-    if (phtNode === null) {
-      const result = await this._linearSearch(nodeLabel)
-      phtNode = result.phtNode
-      metadata = result.metadata
+    if (result.phtNode !== null) {
+      return result
     }
 
-    return { phtNode, metadata }
+    return this._linearSearch(nodeLabel)
   }
 
   unhashedKeyFrom(string) {
@@ -89,7 +87,7 @@ class PrefixHashTree {
   }
 
   async init() {
-    const { phtNode: rootNode, metadata: _ } = await this._dhtLookup('')
+    const { phtNode: rootNode } = await this._dhtLookup('')
 
     if (rootNode !== null) {
       return
@@ -100,7 +98,7 @@ class PrefixHashTree {
 
   async insert(key, val) {
     const nodeLabel = PrefixHashTree._bufferToBase2String(key)
-    const { phtNode: leaf, metadata: _ } = await this._fullSearch(nodeLabel)
+    const { phtNode: leaf } = await this._fullSearch(nodeLabel)
 
     // TODO: alert the caller upon replication failure
     if (leaf === null) {
@@ -122,7 +120,7 @@ class PrefixHashTree {
       const keysAsBase2Strings = pairs.map(([key, _]) => PrefixHashTree._bufferToBase2String(key))
 
       const lcp = PrefixHashTree._lcp(keysAsBase2Strings)
-      await this._split({ pairs: pairs, lcp: lcp, depth: label(leaf).length, leaf: leaf })
+      await this._split({ pairs, lcp, depth: label(leaf).length, leaf })
     }
   }
 
@@ -160,21 +158,22 @@ class PrefixHashTree {
       await this.putFunc(this._labelHash(label(parent)), parent)
 
       const next =
-        PrefixHashTree._lcp([lcp, label(child0)]) > PrefixHashTree._lcp([lcp, label(child1)])
+        PrefixHashTree._lcp([lcp, label(child0)]).length >
+        PrefixHashTree._lcp([lcp, label(child1)]).length
           ? child0
           : child1
 
-      await this._split({ pairs: pairs, lcp: lcp, depth: depth + 1, leaf: next })
+      await this._split({ pairs, lcp, depth: depth + 1, leaf: next })
     }
   }
 
   async searchLeaf(key) {
     const keyLabel = PrefixHashTree._bufferToBase2String(key)
-    return await this._fullSearch(keyLabel)
+    return this._fullSearch(keyLabel)
   }
 
   async searchExact(key) {
-    const { phtNode: leaf, metadata: _ } = await this.searchLeaf(key)
+    const { phtNode: leaf } = await this.searchLeaf(key)
 
     if (leaf === null) {
       return null
@@ -186,7 +185,7 @@ class PrefixHashTree {
 
   async prefixQuery(prefix) {
     const prefixLabel = PrefixHashTree._bufferToBase2String(prefix)
-    let { phtNode: startNode, metadata: _ } = await this._dhtLookup(prefixLabel)
+    let { phtNode: startNode } = await this._dhtLookup(prefixLabel)
 
     if (startNode === null) {
       const result = await this._fullSearch(prefixLabel)
@@ -198,7 +197,7 @@ class PrefixHashTree {
       return null
     }
 
-    return await this._doPrefixQuery({ phtNode: startNode, prefix: prefix })
+    return this._doPrefixQuery({ phtNode: startNode, prefix })
   }
 
   async _doPrefixQuery({ phtNode, prefix } = {}) {
@@ -209,38 +208,25 @@ class PrefixHashTree {
 
     const subtree0 = `${label(phtNode)}0`
     const subtree1 = `${label(phtNode)}1`
-    const prefixLabel = PrefixHashTree._bufferToBase2String(prefix)
-
-    const subtrieTraversals = []
 
-    if (prefixLabel.startsWith(subtree0) || subtree0.startsWith(prefixLabel)) {
-      subtrieTraversals.push(
-        (async () => {
-          // TODO: alert caller upon replication failure
-          const { phtNode: child, metadata: _ } = await this._dhtLookup(subtree0)
-          return child !== null ? this._doPrefixQuery({ phtNode: child, prefix }) : []
-        })()
-      )
-    }
-
-    if (prefixLabel.startsWith(subtree1) || subtree1.startsWith(prefixLabel)) {
-      subtrieTraversals.push(
-        (async () => {
-          // TODO: alert caller upon replication failure
-          const { phtNode: child, metadata: _ } = await this._dhtLookup(subtree1)
-          return child !== null ? this._doPrefixQuery({ phtNode: child, prefix }) : []
-        })()
-      )
-    }
+    const results = await Promise.all([
+      (async () => {
+        // TODO: alert caller upon replication failure
+        const { phtNode: child } = await this._dhtLookup(subtree0)
+        return child !== null ? this._doPrefixQuery({ phtNode: child, prefix }) : []
+      })(),
+      (async () => {
+        // TODO: alert caller upon replication failure
+        const { phtNode: child } = await this._dhtLookup(subtree1)
+        return child !== null ? this._doPrefixQuery({ phtNode: child, prefix }) : []
+      })()
+    ])
 
-    const results = await Promise.all(subtrieTraversals)
     return results.flat()
   }
 
   static _bufferToBase2String(buf) {
-    return Array.from(buf)
-      .map((byte) => byte.toString(2).padStart(8, '0'))
-      .join('')
+    return Array.from(buf, (byte) => byte.toString(2).padStart(8, '0')).join('')
   }
 
   static _lcp(strings = []) {

package/lib/pht-node.d.ts

@@ -22,7 +22,7 @@ declare module 'prefix-hash-tree' {
 
   export function size(node: PHTNode): number
 
-  export function setPointers(node: PHTNode, left: string, right: string): void
+  export function setPointers(node: PHTNode, left: string | null, right: string | null): void
 
   export function pointerLeft(node: PHTNode): string | null
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prefix-hash-tree",
-  "version": "0.0.3",
+  "version": "0.0.5",
   "description": "An extensible prefix hash tree for indexing over distributed hash tables",
   "main": "index.js",
   "exports": {