TypeORM中如何实现递归查询获取父ID？

霞舞 2025-02-21 00:00:00 次阅读

TypeORM递归查询：获取父ID及路径

TypeORM自身并不直接支持递归查询，需要借助数据库原生SQL语句或自定义逻辑实现。本文将介绍几种在TypeORM中实现递归查询的方法，以获取指定ID的父ID及完整路径。

场景：

假设数据库表结构如下，id为主键，pid为父ID：

+----+-----+
| id | pid |
+----+-----+
| 1  | NULL |
| 2  | 1   |
| 3  | 2   |
+----+-----+

目标：查询id=3的父ID路径（预期结果：[3, 2, 1]）。

方法一：使用数据库原生SQL (推荐)

这是最有效率的方法，直接利用数据库的递归查询能力。以下示例使用MySQL的公用表表达式 (CTE)：

WITH RECURSIVE ParentPath AS (
    SELECT id, pid, CAST(id AS CHAR) AS path
    FROM tree_table
    WHERE id = 3  -- 起始ID
    UNION ALL
    SELECT t.id, t.pid, CONCAT(p.path, ',', t.id)
    FROM tree_table t
    INNER JOIN ParentPath p ON t.id = p.pid
)
SELECT path
FROM ParentPath
ORDER BY LENGTH(path) DESC
LIMIT 1;

在TypeORM中，可以使用createQueryBuilder执行此SQL：

import { getConnection } from 'typeorm';

async function getParentPath(id: number): Promise {
  const result = await getConnection()
    .createQueryBuilder()
    .raw(`WITH RECURSIVE ParentPath AS (...) SELECT path FROM ParentPath ORDER BY LENGTH(path) DESC LIMIT 1;`,  // 将上面的SQL语句填入此处
    )
    .getRawOne();
  return result.path;
}