UNPKG

@convex-dev/aggregate

Version:

Convex component to calculate counts and sums of values for efficient aggregation.

116 lines 3.57 kB
import { v } from "convex/values"; import { query } from "./_generated/server.js"; import { getTree, p } from "./btree.js"; import schema from "./schema.js"; export const display = query({ args: { namespace: v.optional(v.any()) }, handler: async (ctx, args) => { const tree = await getTree(ctx.db, args.namespace); if (!tree) { return "empty"; } return await displayNode(ctx.db, tree.root); }, }); async function displayNode(db, node, depth = 0) { const n = (await db.get(node)); for (let i = 0; i < n.items.length; i++) { if (n.subtrees.length > 0) { await displayNode(db, n.subtrees[i], depth + 1); } console.log(" ".repeat(depth) + p(n.items[i].k)); } if (n.subtrees.length > 0) { await displayNode(db, n.subtrees[n.subtrees.length - 1], depth + 1); } } export const dump = query({ args: { namespace: v.optional(v.any()) }, returns: v.string(), handler: async (ctx, args) => { return await dumpTree(ctx.db, args.namespace); }, }); export async function dumpTree(db, namespace) { const t = (await getTree(db, namespace)); return dumpNode(db, t.root); } /// Prints keys in-order, with brackets for each node. async function dumpNode(db, node) { const n = (await db.get(node)); let s = "["; if (n.subtrees.length === 0) { s += n.items .map((i) => i.k) .map(p) .join(", "); } else { const subtrees = await Promise.all(n.subtrees.map((subtree) => dumpNode(db, subtree))); for (let i = 0; i < n.items.length; i++) { s += `${subtrees[i]}, ${p(n.items[i].k)}, `; } s += subtrees[n.items.length]; } s += "]"; return s; } export const inspectNode = query({ args: { node: v.optional(v.string()), namespace: v.optional(v.any()) }, returns: v.null(), handler: async (ctx, args) => { const tree = await getTree(ctx.db, args.namespace); if (!tree) { console.log("no tree"); return; } let n = await ctx.db.get(tree.root); if (args.node) { n = await ctx.db.get(args.node); } if (!n) { console.log("no node"); return; } console.log("btreeNode", n._id); console.log("aggregate", n.aggregate); for (let i = 0; i < n.items.length; i++) { if (n.subtrees.length > 0) { console.log("subtree", n.subtrees[i]); } console.log("item", n.items[i]); } if (n.subtrees.length > 0) { console.log("subtree", n.subtrees[n.subtrees.length - 1]); } }, }); export const listTrees = query({ args: { take: v.optional(v.number()), }, returns: v.array(v.object({ ...schema.tables.btree.validator.fields, _id: v.id("btree"), _creationTime: v.number(), })), handler: async (ctx, args) => { const values = await ctx.db.query("btree").take(args.take ?? 100); return values; }, }); export const listTreeNodes = query({ args: { take: v.optional(v.number()), }, returns: v.array(v.object({ ...schema.tables.btreeNode.validator.fields, _id: v.id("btreeNode"), _creationTime: v.number(), })), handler: async (ctx, args) => { const values = await ctx.db.query("btreeNode").take(args.take ?? 100); return values; }, }); //# sourceMappingURL=inspect.js.map