xiaoxue_admin
/
dbx
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(sidebar): preserve Mongo collection children in search

This commit is contained in:
二丫讲梵 2026-07-08 23:44:25 +08:00 committed by GitHub
parent 843babf522
commit ddf6b3ed6b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 92 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/desktop/src/lib/sidebar/sidebarSearchTree.ts
View File

@ -1,7 +1,7 @@
import type { TreeNode, TreeNodeType } from "@/types/database";
import { createSidebarLabelMatcher, type SidebarLabelMatcher } from "@/lib/sidebar/sidebarSearch";
const preserveMatchedSubtreeTypes = new Set(["connection", "database", "schema", "table", "view"]);
const preserveMatchedSubtreeTypes = new Set(["connection", "database", "schema", "table", "view", "mongo-db", "mongo-collection"]);
const normalizedLabelCache = new WeakMap<TreeNode, { label: string; normalized: string }>();

91
packages/app-tests/sidebarSearchTree.test.ts
View File

@ -84,6 +84,97 @@ test("preserves loaded schema children when the database itself matches search",
assert.equal(filtered[0]?.children?.[0]?.label, "public");
});
test("preserves loaded MongoDB collection children when the database itself matches search", () => {
const nodes: TreeNode[] = [
{
id: "conn:analytics",
label: "analytics",
type: "mongo-db",
connectionId: "conn",
database: "analytics",
isExpanded: true,
children: [
{
id: "conn:analytics:__gridfs",
label: "tree.gridfs",
type: "mongo-gridfs",
connectionId: "conn",
database: "analytics",
isExpanded: false,
},
{
id: "conn:analytics:orders",
label: "orders",
type: "mongo-collection",
connectionId: "conn",
database: "analytics",
isExpanded: false,
},
],
},
];
const filtered = filterSidebarTree(nodes, "analytics", new Set());
assert.equal(filtered[0]?.label, "analytics");
assert.deepEqual(
filtered[0]?.children?.map((child) => child.type),
["mongo-gridfs", "mongo-collection"],
);
assert.equal(filtered[0]?.children?.[1]?.label, "orders");
});
test("preserves loaded MongoDB collection groups when the collection itself matches search", () => {
const nodes: TreeNode[] = [
{
id: "conn:analytics",
label: "analytics",
type: "mongo-db",
connectionId: "conn",
database: "analytics",
isExpanded: true,
children: [
{
id: "conn:analytics:orders",
label: "orders",
type: "mongo-collection",
connectionId: "conn",
database: "analytics",
isExpanded: true,
children: [
{
id: "conn:analytics:orders:__columns",
label: "tree.columns",
type: "group-columns",
connectionId: "conn",
database: "analytics",
tableName: "orders",
isExpanded: true,
children: [
{
id: "conn:analytics:orders:__columns:_id",
label: "_id",
type: "column",
connectionId: "conn",
database: "analytics",
tableName: "orders",
},
],
},
],
},
],
},
];
const filtered = filterSidebarTree(nodes, "orders", new Set());
const collection = filtered[0]?.children?.[0];
assert.equal(collection?.label, "orders");
assert.equal(collection?.children?.[0]?.label, "tree.columns");
assert.equal(collection?.children?.[0]?.children?.[0]?.label, "_id");
});
test("preserves loaded children when the connection itself matches search", () => {
const nodes: TreeNode[] = [
{