datafolder 1.0.17 → 1.0.19

package/src/QyCache.js

@@ -5,52 +5,50 @@ import {
 import r from "node:events";
 
 import {
-    QyDir as p
+    QyDir as l
 } from "./QyDir.js";
 
 import {
-    QyFileLogger as u
+    QyFileLogger as g
 } from "./QyFileLogger.js";
 
 import {
-    QyKVData as d
+    QyKVData as h
 } from "./QyKVData.js";
 
 import {
-    QyTriggerNode as f,
+    QyTriggerNode as R,
     updateTriggerNodes as s
 } from "./QyTrigger.js";
 
 import {
-    isString as x,
-    lockExclusiveFilePath as C,
-    unlockExclusiveFilePath as i
+    isString as C,
+    lockExclusiveFilePath as x,
+    unlockExclusiveFilePath as o,
+    ensureDir as q,
+    SpecialFilePaths as e,
+    SpecialOperators as t
 } from "./QyUtils.js";
 
 import {
-    dumpStructure as q
+    dumpStructure as O
 } from "./QyStructure.js";
 
 import {
     getIndexPropPaths as n,
-    parseNamePathAndQuery as g
+    parseNamePathAndQuery as d
 } from "./QyIndex.js";
 
 import {
-    getDefaultOptions as h
+    getDefaultOptions as u
 } from "./QyDefaultOptions.js";
 
 import {
-    ensureDir as R,
-    SpecialFilePaths as e
-} from "./QyUtils.js";
-
-import {
-    QueryCounter as l
+    QueryCounter as p
 } from "./QyUnionNode.js";
 
 import {
-    QyKVDataLoader as O
+    QyKVDataLoader as P
 } from "./QyKVDataLoader.js";
 
 import {
@@ -58,19 +56,28 @@ import {
 } from "./QyMod.js";
 
 import {
-    logger as Q
+    logger as b
 } from "./QyLogger.js";
 
+import {
+    QyEmbeddings as D
+} from "./QyEmbeddings.js";
+
+var {} = t;
+
 let {
-    HiddenFolderName: D,
+    HiddenFolderName: y,
     IndexDirPath: F,
-    TriggerDirPath: P,
-    RpcDirPath: S,
+    TriggerDirPath: w,
+    RpcDirPath: E,
     ExpireDirPath: m,
-    MetaFilePath: t
+    MetaFilePath: i,
+    HNSWDirPath: f,
+    InfoDirPath: Q,
+    RecycleDirPath: S
 } = e;
 
-class o extends r {
+class a extends r {
     rootFolder;
     options;
     qyDB;
@@ -81,35 +88,35 @@ class o extends r {
     rpcRootDir;
     indexRootDir;
     expireRootDir;
+    hnswRootDir;
+    infoRootDir;
+    recycleRootDir;
     exclusiveFilePath;
     qyKVData;
     qyFileLogger;
+    qyEmbeddings;
     rpcFunMap = {};
     constructor(r, e, t) {
-        super(), e = {
-            ...h("QyCache"),
+        super(), this.rootFolder = r, e = this.options = {
+            ...u("QyCache"),
             ...e
-        }, o = new p("", !0, void 0, i = this), a = m.at(-1), s = F.at(-1), l = new p(D, !0, o), 
-        n = new p(a, !0, l), g = new p(s, !0, l), o._subdirMap = {
-            [D]: l
-        }, l._subdirMap = {
-            [a]: n,
-            [s]: g
-        }, Object.assign(i, {
-            rootDir: o,
-            expireRootDir: n,
-            indexRootDir: g
-        });
-        var i, o, a, s, n, g, l = v(r, "qykv");
+        }, this.qyDB = t;
+        var i, o = this, t = o.rootDir = new l("", !0, void 0, o), a = new l(y, !0, t);
+        t._subdirMap = {
+            [y]: a
+        }, a._subdirMap = {};
+        for ([ s, i ] of [ [ F, "indexRootDir" ], [ m, "expireRootDir" ], [ f, "hnswRootDir" ], [ Q, "infoRootDir" ], [ S, "recycleRootDir" ] ]) {
+            var s = s.at(-1), n = new l(s, !0, a);
+            a._subdirMap[s] = n, o[i] = n;
+        }
+        t = v(r, "qykv");
         Object.assign(this, {
-            rootFolder: r,
-            options: e,
-            qyDB: t,
-            kvFolder: l,
+            kvFolder: t,
             rpcFolder: v(r, "rpc"),
             exclusiveFilePath: v(r, ".opened"),
-            qyKVData: new d(l, e, this),
-            qyFileLogger: new u(v(r, "qylog"), e)
+            qyKVData: new h(t, e, this),
+            qyFileLogger: new g(v(r, "qylog"), e),
+            qyEmbeddings: new D(this, e)
         });
     }
     async start() {
@@ -120,20 +127,21 @@ class o extends r {
             qyKVData: i,
             qyFileLogger: o,
             exclusiveFilePath: a,
-            rootDir: s
+            rootDir: s,
+            qyEmbeddings: n
         } = this;
-        Q.log(`Starting at ${r}...`);
+        b.log(`Starting at ${r}...`);
         try {
-            await R(e), o.start();
+            await q(e), o.start();
             var [ {
-                maxChangeId: n,
+                maxChangeId: l,
                 loadedMaxAclNum: g,
-                qySnapshots: l,
-                aclKeyValueMap: p
-            } ] = await Promise.all([ ((r, e) => (r = new O(r, e)).load())(e, t), C(a) ]);
-            i.start(n, g, l, p);
+                qySnapshots: h,
+                aclKeyValueMap: d
+            } ] = await Promise.all([ ((r, e) => (r = new P(r, e)).load())(e, t), x(a) ]);
+            i.start(l, g, h, d), await n.start();
         } catch (r) {
-            throw o.stop(), r;
+            throw o.stop(), n.stop(), r;
         }
         t.preloadFoldersAtStart && !function r(e) {
             let {
@@ -143,45 +151,49 @@ class o extends r {
             for (var o of t) r(o);
         }(s);
         var u = this, e = u.rootDir;
-        e.triggerNodes = [ new f() ];
-        for (d of (e = u.triggerRootDir = e.searchOrCreateSubdir(P)).fileList) {
+        e.triggerNodes = [ new R() ];
+        for (p of (e = u.triggerRootDir = e.searchOrCreateSubdir(w)).fileList) {
             var {
-                name: d,
-                fileContent: h
-            } = d, [ h, c, D ] = h;
-            u.insertTriggerToTree(d, h, c, D);
+                name: p,
+                fileContent: c
+            } = p, [ c, D, y ] = c;
+            u.insertTriggerToTree(p, c, D, y);
         }
         var F = this;
-        F.rpcRootDir = F.rootDir.searchOrCreateSubdir(S);
+        F.rpcRootDir = F.rootDir.searchOrCreateSubdir(E);
         for (m of F.rpcRootDir.fileList) {
             var {
                 name: m,
-                fileContent: y
+                fileContent: f
             } = m;
-            x(y) && F.insertRpc(m, y);
+            C(f) && F.insertRpc(m, f);
         }
-        return t.dumpStructureAtStart && await q(v(r, "qystruct"), s), Q.log("Started."), 
+        return t.dumpStructureAtStart && await O(v(r, "qystruct"), s), b.log("Started."), 
         this;
     }
     async stop() {
-        Q.log("Stopping...");
+        b.log("Stopping...");
         var {
             qyKVData: r,
-            qyFileLogger: e,
-            exclusiveFilePath: t
+            qyEmbeddings: e,
+            qyFileLogger: t,
+            exclusiveFilePath: i
         } = this;
-        return await r.stop(), Q.log("QyFileLogger stopped."), await e.stop(), i(t), 
-        Object.assign(this, {
+        return await r.stop(), await e.stop(), b.log("QyFileLogger stopped."), await t.stop(), 
+        o(i), Object.assign(this, {
             rootDir: void 0,
             indexRootDir: void 0,
             expireRootDir: void 0,
             triggerRootDir: void 0,
             rpcRootDir: void 0,
+            hnswRootDir: void 0,
+            infoRootDir: void 0,
+            recycleRootDir: void 0,
             qyKVData: void 0,
             rpcFunMap: void 0,
             qyDB: void 0,
             qyFileLogger: void 0
-        }), Q.log("Stopped."), this;
+        }), b.log("Stopped."), this;
     }
     increaseChangeId() {
         return this.qyKVData.increaseChangeId();
@@ -197,22 +209,22 @@ class o extends r {
         return t && s.save(r, t), a.save(r, e, i, o);
     }
     getFile(r, e = this.rootDir) {
-        return r.isFile ? r : x(r) ? e.getFile(r) : e.searchFile(r);
-    }
-    getDir(r, e = this.rootDir) {
-        return r.isDir ? r : x(r) ? e.getSubdir(r) : e.searchSubdir(r);
+        return r.isFile ? r : C(r) ? e.getFile(r) : e.searchFile(r);
     }
     getOrCreateFile(r, e = this.rootDir) {
-        return r.isFile ? r : x(r) ? e.getOrCreateFile(r) : e.searchOrCreateFile(r);
+        return r.isFile ? r : C(r) ? e.getOrCreateFile(r) : e.searchOrCreateFile(r);
+    }
+    getDir(r, e = this.rootDir) {
+        return r.isDir ? r : C(r) ? e.getSubdir(r) : e.searchSubdir(r);
     }
     getOrCreateDir(r, e = this.rootDir) {
-        return r.isDir ? r : x(r) ? e.getOrCreateSubdir(r) : e.searchOrCreateSubdir(r);
+        return r.isDir ? r : C(r) ? e.getOrCreateSubdir(r) : e.searchOrCreateSubdir(r);
     }
     getIndexDir(r) {
-        return this.indexRootDir.searchSubdir(y(r.fullPathHash));
+        return this.indexRootDir.searchSubdir(K(r.fullPathHash));
     }
     getOrCreateIndexDir(r) {
-        return this.indexRootDir.searchOrCreateSubdir(y(r.fullPathHash));
+        return this.indexRootDir.searchOrCreateSubdir(K(r.fullPathHash));
     }
     getTriggerFile(r) {
         return this.triggerRootDir.getFile(r);
@@ -221,10 +233,36 @@ class o extends r {
         return this.triggerRootDir.getOrCreateFile(r);
     }
     getExpireInfoFile(r, e = "c") {
-        return this.expireRootDir.searchFile(a(r, e));
+        return this.expireRootDir.searchFile(L(r, e));
     }
     getOrCreateExpireInfoFile(r, e = "c") {
-        return this.expireRootDir.searchOrCreateFile(a(r, e));
+        return this.expireRootDir.searchOrCreateFile(L(r, e));
+    }
+    getInfoFile(r) {
+        return this.infoRootDir.searchFile(I(r.fullPathHash));
+    }
+    getOrCreateInfoFile(r) {
+        return this.infoRootDir.searchOrCreateFile(I(r.fullPathHash));
+    }
+    hToSrcFile(r) {
+        r = this.infoRootDir.searchFile(I(r));
+        if (r && r.fileContent) {
+            r = r.fileContent.namePath;
+            if (r) return this.getOrCreateFile(r);
+        }
+    }
+    getEmbedding(r) {
+        var e, t = this.getInfoFile(r);
+        if (t && t.fileContent) return {
+            length: t,
+            idx: e
+        } = t.fileContent, this.qyEmbeddings.getEmbedding(r.fullPathHash, t, e);
+    }
+    getOrCreateRecycleFile(r) {
+        return this.recycleRootDir.getOrCreateFile(r);
+    }
+    getRecycleFile(r) {
+        return this.recycleRootDir.getFile(r);
     }
     get triggerFileList() {
         return this.triggerRootDir.fileList;
@@ -239,7 +277,7 @@ class o extends r {
         return this.rpcRootDir.fileList;
     }
     get metaFile() {
-        return this.rootDir.searchOrCreateFile(t);
+        return this.rootDir.searchOrCreateFile(i);
     }
     insertTriggerToTree(r, e, t, i) {
         var {
@@ -263,21 +301,21 @@ class o extends r {
         delete this.rpcFunMap[r];
     }
     queryFiles(r, e, t, i, o = this.rootDir) {
-        var a = [], t = (x(e) && !t && (t = 1), new l(t, i)), i = this.options.queryAutoCreateIndex;
-        return w(r, e, a, t, o, i), a;
+        var a = [], t = (C(e) && !t && (t = 1), new p(t, i)), i = this.options.queryAutoCreateIndex;
+        return T(r, e, a, t, o, i), a;
     }
     queryFilesMulti(r, e, t, i = this.rootDir) {
-        var o, a, s = [], n = new l(e, t), g = this.options.queryAutoCreateIndex;
-        for ([ o, a ] of r) w(o, a, s, n, i, g);
+        var o, a, s = [], n = new p(e, t), l = this.options.queryAutoCreateIndex;
+        for ([ o, a ] of r) T(o, a, s, n, i, l);
         return s;
     }
     async callRpc(e, r, t) {
         var i, o = this.rpcFunMap[e];
-        if (!o) return Q.warn(i = `RPC ${e} not implemented.`), new Error(i);
+        if (!o) return b.warn(i = `RPC ${e} not implemented.`), new Error(i);
         try {
             return await o(r, t);
         } catch (r) {
-            return Q.error(`Calling RPC ${e} failed:`, r), r;
+            return b.error(`Calling RPC ${e} failed:`, r), r;
         }
     }
     monitorItemExpire(r, e) {
@@ -294,18 +332,22 @@ class o extends r {
     }
 }
 
-function a(r, e) {
+function I(r) {
+    return [ r.substring(0, 2), r ];
+}
+
+function L(r, e) {
     return [ e, r.substring(0, 2), r ];
 }
 
-function y(r) {
+function K(r) {
     return [ r[0], r ];
 }
 
-function w(r, e, t, i, o, a) {
-    e && (a = a ? n(e) : void 0, g(r, o, a, e, t, i));
+function T(r, e, t, i, o, a) {
+    e && (a = a ? n(e) : void 0, d(r, o, a, e, t, i));
 }
 
 export {
-    o as QyCache
+    a as QyCache
 };

package/src/QyDB.js

@@ -33,7 +33,7 @@ import {
     logger as m
 } from "./QyLogger.js";
 
-let p = e(n, "../data/sampleDataFolder");
+let d = e(n, "../data/sampleDataFolder");
 
 class l extends t {
     rootFolder;
@@ -42,15 +42,12 @@ class l extends t {
     batch;
     _immediate;
     creationTime;
-    constructor(e = p, t) {
-        s(e) && (t = e, e = p), super(), this.options = {
+    constructor(e = d, t) {
+        s(e) && (t = e, e = d), super(), this.options = {
             ...o("QyDB"),
             ...t
         }, this.rootFolder = e;
-        for (let i of [ "createIndex", "delIndex", "insert", "put", "remove", "delDir", "delFile", "delContent", "insertTrigger", "removeTrigger", "insertRpc", "removeRpc", "expireFile", "expireDir", "expireContent", "expireFileM", "expireDirM", "expireContentM", "removeFileExpire", "removeDirExpire", "removeContentExpire" ]) this[i] = (...e) => {
-            var t = this.currentBatch;
-            return t[i].apply(t, e).run();
-        };
+        for (let t of [ "createIndex", "delIndex", "insert", "put", "remove", "delDir", "delFile", "delContent", "insertTrigger", "removeTrigger", "insertRpc", "removeRpc", "insertEmbedding", "delEmbedding", "insertWithEmbedding", "expireFile", "expireDir", "expireContent", "expireFileM", "expireDirM", "expireContentM", "removeFileExpire", "removeDirExpire", "removeContentExpire" ]) this[t] = (...e) => this.currentBatch[t](...e).run();
     }
     async start(e) {
         var t;

package/src/QyDefaultOptions.js

@@ -1,5 +1,6 @@
 var e = {
     append: !1,
+    read: !1,
     retryIntervalFactor: 2,
     maxRetryInterval: 60,
     emitSaveEvent: !1,
@@ -57,6 +58,13 @@ let t = {
     },
     QyAclWriter: {
         minBatchSize: 2
+    },
+    QyEmbeddingWriter: {
+        append: !0,
+        read: !0
+    },
+    QyEmbeddings: {
+        appendOnly: !1
     }
 };
 

package/src/QyEmbeddingWriter.d.ts

@@ -0,0 +1,12 @@
+import { QyBinWriter } from "./QyBinWriter.ts";
+export declare class QyEmbeddingWriter extends QyBinWriter {
+    length: any;
+    byteLength: any;
+    totalCount: number;
+    constructor(filePath: any | undefined, length: any);
+    start(): Promise<void>;
+    append(embedding: Float32Array, needPromise?: any): any;
+    pWrite(embedding: Float32Array, idx: any, needPromise?: any): any;
+    _op_pWrite(buffer: any, idx: any): any;
+    readSync(idx: any): Float32Array<ArrayBuffer> | undefined;
+}

package/src/QyEmbeddingWriter.js

@@ -0,0 +1,55 @@
+import {
+    readBufferFromFD as s
+} from "./QyUtils.js";
+
+import {
+    getDefaultOptions as r
+} from "./QyDefaultOptions.js";
+
+import {
+    QyBinWriter as t
+} from "./QyBinWriter.js";
+
+class e extends t {
+    length;
+    byteLength;
+    totalCount = 0;
+    constructor(t = void 0, e) {
+        super(r("QyEmbeddingWriter"), t), this.length = e, this.byteLength = e * Float32Array.BYTES_PER_ELEMENT;
+    }
+    async start() {
+        await this.pushAndRunWithPromise({
+            opFunName: "_op_start"
+        }), 0 == this.totalCount && (this.totalCount = this.totalSavedSize / this.byteLength);
+    }
+    append(t, e = !1) {
+        return ++this.totalCount, this.save(0, i(t), e);
+    }
+    pWrite(t, e, r = !1) {
+        t = {
+            opFunName: "_op_pWrite",
+            args: [ i(t), e ]
+        };
+        return r ? this.pushAndRunWithPromise(t) : this.pushAndRun(t);
+    }
+    _op_pWrite(t, e) {
+        return this.fileHandle.write(t, 0, t.byteLength, e * this.byteLength);
+    }
+    readSync(t) {
+        var {
+            filePath: e,
+            fileHandle: r,
+            length: i,
+            byteLength: n
+        } = this;
+        return (r = r && s(r.fd, e, t * n, n)) ? new Float32Array(r.buffer, r.byteOffset, i) : void 0;
+    }
+}
+
+function i(t) {
+    return Buffer.from(t.buffer, t.byteOffset, t.byteLength);
+}
+
+export {
+    e as QyEmbeddingWriter
+};

package/src/QyEmbeddings.d.ts

@@ -0,0 +1,15 @@
+export declare class QyEmbeddings {
+    qyCache: any;
+    embeddingFolder: any;
+    options: any;
+    qyDB: any;
+    qyEmbeddingWriters: any;
+    hashToEmbeddingMap: {};
+    constructor(qyCache: any, options?: any);
+    start(): Promise<void>;
+    stop(): Promise<void>;
+    addEmbedding(fullPathHash: any, embedding: any): any;
+    getEmbedding(fullPathHash: any, length: any, idx: any): any;
+    updateEmbedding(fullPathHash: any, embedding: any, idx: any): any;
+    delEmbedding(fullPathHash: any, length: any, idx: any): void;
+}

package/src/QyEmbeddings.js

@@ -0,0 +1,128 @@
+import {
+    readdir as t
+} from "node:fs/promises";
+
+import {
+    folderOrFileExists as i,
+    SpecialOperators as e
+} from "./QyUtils.js";
+
+import {
+    getDefaultOptions as r
+} from "./QyDefaultOptions.js";
+
+import {
+    logger as a
+} from "./QyLogger.js";
+
+import {
+    basename as n,
+    join as s
+} from "node:path";
+
+import {
+    QyEmbeddingWriter as d
+} from "./QyEmbeddingWriter.js";
+
+let o = e.$del;
+
+class l {
+    qyCache;
+    embeddingFolder;
+    options;
+    qyDB;
+    qyEmbeddingWriters = {};
+    hashToEmbeddingMap = {};
+    constructor(e, t = void 0) {
+        var i = s(e.rootFolder, "qyembedding");
+        t = {
+            ...r("QyEmbeddings"),
+            ...t
+        }, Object.assign(this, {
+            qyCache: e,
+            embeddingFolder: i,
+            options: t,
+            qyDB: e.qyDB
+        });
+    }
+    async start() {
+        var e = this.embeddingFolder;
+        await i(e) && await Promise.all((await t(e)).filter(e => /[0-9]+\.bin$/.test(e)).map(async e => {
+            await h(this, n(e, ".bin")).start();
+        }));
+    }
+    async stop() {
+        var e = this.qyEmbeddingWriters;
+        e && (await Promise.all(Object.values(e).map(e => e.stop())), this.qyEmbeddingWriters = void 0);
+    }
+    addEmbedding(e, t) {
+        var {
+            hashToEmbeddingMap: i,
+            options: r
+        } = this;
+        if (i[e] = t, !r.appendOnly) {
+            i = ((e, t) => {
+                var {
+                    qyCache: e,
+                    qyDB: i
+                } = e, r = e.getRecycleFile(t);
+                if (r) {
+                    var n, e = r.fileContent;
+                    for (n in e) return i.insert(r, {
+                        [n]: o
+                    }), n;
+                }
+            })(this, t.length);
+            if (null != i) return p(this, e, t, i);
+        }
+        return m(this, e, t);
+    }
+    getEmbedding(e, t, i) {
+        var r = this.hashToEmbeddingMap, r = r[e];
+        return r || (null != t && null != i ? this.hashToEmbeddingMap[e] = g(this, t).readSync(i) : void 0);
+    }
+    updateEmbedding(e, t, i) {
+        var {
+            options: r,
+            hashToEmbeddingMap: n
+        } = this;
+        return n[e] = t, r.appendOnly ? m(this, e, t) : p(this, e, t, i);
+    }
+    delEmbedding(e, t, i) {
+        var {
+            hashToEmbeddingMap: r,
+            qyCache: n,
+            qyDB: a
+        } = this;
+        delete r[t], null != t && null != i && a.insert(n.getOrCreateRecycleFile(t), {
+            [i]: 1
+        });
+    }
+}
+
+function m(e, t, i) {
+    var r = i.length, e = g(e, r), n = e.totalCount;
+    return e.append(i), a.info(`Append ${t} ${r} ` + n), n;
+}
+
+function p(e, t, i, r) {
+    var n = i.length;
+    return g(e, i.length).pWrite(i, r), a.info(`pWrite ${t} ${n} ` + r), r;
+}
+
+function h(e, t) {
+    var {
+        qyEmbeddingWriters: e,
+        embeddingFolder: i
+    } = e;
+    return e[t] = new d(s(i, t + ".bin"), t);
+}
+
+function g(e, t) {
+    let i = e.qyEmbeddingWriters[t];
+    return i || (i = h(e, t)).start(), i;
+}
+
+export {
+    l as QyEmbeddings
+};