基于用户关联关系构建成员树：查询user=1对应ID1-5的思路

现有数据库表结构及数据 #

需求：指定user=1时，获取关联的ID为1、2、3、4、5的记录。

思路一：数据库递归CTE查询 #

现代数据库（MySQL 8.0+、PostgreSQL、SQL Server等）支持递归公共表表达式，可直接在数据库层完成链式关联查询，效率较高。

以MySQL为例，编写递归查询语句：

WITH RECURSIVE member_chain AS (
    -- 初始节点：获取user=1的记录
    SELECT ID, user, user2, side, date
    FROM your_table_name
    WHERE user = 1
    UNION ALL
    -- 递归遍历：用前一轮的user2作为当前查询的user，获取关联记录
    SELECT t.ID, t.user, t.user2, t.side, t.date
    FROM your_table_name t
    JOIN member_chain mc ON t.user = mc.user2
)
SELECT * FROM member_chain;

该查询会先抓取user=1的记录，再递归查找所有以之前结果中user2为user的记录，直到无匹配项，最终返回ID1-5的关联记录。

思路二：应用层迭代遍历 #

若数据库不支持递归CTE（如MySQL 5.x），可在应用层通过循环迭代实现：

1. 先查询user=1的记录，加入结果集，记录该记录的user2值（即2）。
2. 以该user2值作为新的user，查询对应记录并加入结果集，再更新user2值（即3）。
3. 重复步骤2，直到查询不到对应记录为止。

Python伪代码示例：

def get_member_chain(db_connection, start_user):
    chain_records = []
    current_user = start_user
    while True:
        # 执行单条记录查询
        cursor = db_connection.cursor(dictionary=True)
        cursor.execute("SELECT * FROM your_table_name WHERE user = %s", (current_user,))
        record = cursor.fetchone()
        if not record:
            break
        chain_records.append(record)
        # 更新当前查询的user为当前记录的user2
        current_user = record["user2"]
    return chain_records

# 调用获取user=1的关联链
result = get_member_chain(your_db_conn, 1)

注意事项 #

• 需处理循环引用：若数据存在A关联B、B关联A的情况，需加入已访问节点的判断，避免无限循环。
• 性能对比：数据库递归CTE比应用层遍历更高效，可减少多次数据库查询的开销。

内容的提问来源于stack exchange，提问作者Mouner Mostafa