mysql 递归查询树形展示

原创

mob64ca12e08acf 2023-10-13 10:20:25 ©著作权

文章标签 递归查询 MySQL 数据库 文章分类 MySQL 数据库

©著作权归作者所有：来自51CTO博客作者mob64ca12e08acf的原创作品，请联系作者获取转载授权，否则将追究法律责任

MySQL递归查询树形展示的实现步骤

引言

在数据库中，有时候需要查询具有层次结构的数据，并且以树形展示的形式呈现出来。MySQL自身并不支持递归查询，但是我们可以通过使用递归的方式来实现这个功能。本文将详细介绍如何使用MySQL实现递归查询树形展示。

整体流程

下面是整个过程的流程图：

stateDiagram
    [*] --> 开始
    开始 --> 连接数据库
    连接数据库 --> 执行递归查询
    执行递归查询 --> 输出结果
    输出结果 --> 结束
    结束 --> [*]

具体步骤

步骤1：连接数据库

首先，我们需要连接到MySQL数据库。可以使用以下代码来连接数据库：

import mysql.connector

# 建立与MySQL数据库的连接
mydb = mysql.connector.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="yourdatabase"
)

步骤2：执行递归查询

接下来，我们需要执行递归查询来获取树形展示的数据。MySQL中的递归查询可以通过使用WITH RECURSIVE语句来实现。以下是一个示例的递归查询代码：

# 定义递归查询
query = """
WITH RECURSIVE tree AS (
  SELECT id, name, parent_id
  FROM your_table
  WHERE parent_id IS NULL
  UNION ALL
  SELECT child.id, child.name, child.parent_id
  FROM your_table child
  INNER JOIN tree parent ON child.parent_id = parent.id
)
SELECT id, name, parent_id
FROM tree
"""

# 执行查询
cursor = mydb.cursor()
cursor.execute(query)
result = cursor.fetchall()

在上面的代码中，your_table是你的数据表的名称，id、name和parent_id是你的表中对应的字段名。

步骤3：输出结果

最后，我们可以将查询结果以树形展示的形式输出。以下是一个示例的输出代码：

# 输出结果
def print_tree(node, level=0):
  indent = "  " * level
  print(f"{indent}- {node[1]}")
  for child in result:
    if child[2] == node[0]:
      print_tree(child, level + 1)

# 从根节点开始输出
for node in result:
  if node[2] == None:
    print_tree(node)

上面的代码会递归遍历查询结果，根据层次关系以树形展示的形式输出。

完整代码

下面是整个过程的完整代码：

import mysql.connector

# 建立与MySQL数据库的连接
mydb = mysql.connector.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="yourdatabase"
)

# 定义递归查询
query = """
WITH RECURSIVE tree AS (
  SELECT id, name, parent_id
  FROM your_table
  WHERE parent_id IS NULL
  UNION ALL
  SELECT child.id, child.name, child.parent_id
  FROM your_table child
  INNER JOIN tree parent ON child.parent_id = parent.id
)
SELECT id, name, parent_id
FROM tree
"""

# 执行查询
cursor = mydb.cursor()
cursor.execute(query)
result = cursor.fetchall()

# 输出结果
def print_tree(node, level=0):
  indent = "  " * level
  print(f"{indent}- {node[1]}")
  for child in result:
    if child[2] == node[0]:
      print_tree(child, level + 1)

# 从根节点开始输出
for node in result:
  if node[2] == None:
    print_tree(node)