u

src/store/category/getTree.js

@@ -0,0 +1,38 @@
+import sql from 'mssql';
+import { GET_TREE } from '../../sql/category.js';
+
+function buildTree(rows, rootId) {
+  const byId = new Map();
+  for (const row of rows) {
+    byId.set(row.id, {
+      id: row.id,
+      parent_id: row.parent_id,
+      sort: row.sort,
+      children: [],
+    });
+  }
+
+  let root = null;
+  for (const node of byId.values()) {
+    if (node.id === rootId) {
+      root = node;
+      continue;
+    }
+    const parent = byId.get(node.parent_id);
+    if (parent) parent.children.push(node);
+  }
+
+  return root;
+}
+
+export default function getTree(pool) {
+  return async (rootId) => {
+    const result = await pool
+      .request()
+      .input('rootId', sql.Int, rootId)
+      .query(GET_TREE);
+
+    if (result.recordset.length === 0) return null;
+    return buildTree(result.recordset, rootId);
+  };
+}