java - Java 或 SQL 中的树遍历函数

Question

我有以下场景需要一些帮助。

1. user_profile 表(user_id、姓名、地址、城市)
2. parent_user 表(parent_id、user_id)

如果我传递一个parent_id作为输入，它应该为我提供所有子/子用户的个人资料信息，直到特定深度(例如，直到第5级)。

例如(parent_id -> user_id):1 -> 2, 2 -> 3, 3 -> 4, 4 -> 5, 2 -> 6

因此考虑到这个值，当我传递 1 作为输入时，它应该检索用户 ID 2、3、4、5 和 6 的用户配置文件。

我可以使用 SQL 查询本身来实现此目的吗？

或者使用代码？

这是我到目前为止所尝试过的。但我不认为它是一种高效且能正常工作的方法。

//Still not clear with idea of level or depth
void getAllChild(int parent_id, int depth) {
 int count = 0, user_id = 0;
 List listChild = runSQL("SELECT user_id FROM parent_user WHERE parent_id = "+parent_id);
 List<Map> subList;
 for (int i = 0; i < listChild.size(); i++) {
  user_id = listChild.get(i));
  Map userProfile = runSQL("SELECT * FROM user_profile WHERE user_id = "+user_id);
  subList.add(userProfile);
  getAllChild(user_id, 2);
 }
}

请帮我解决这个问题

Answer 1

SQL 在查询递归数据方面相当糟糕。你的方法基本没问题。您的代码的一个明显问题是您将每个递归调用的结果存储在本地子列表中，该结果不会返回给调用者。另一个问题是，你没有检查深度

有几种方法可以解决这个问题:要么将 subList 作为该方法的参数，然后将其传递给每个递归调用。

void getAllChild(List<Map> subList, int parent_id, int depth) {

 // check whether maximum depth is reached
 if (depth < 0) {
   return;
 }
 depth--;

 int count = 0, user_id = 0;
 List listChild = runSQL("SELECT user_id FROM parent_user WHERE parent_id = "+parent_id);
 for (int i = 0; i < listChild.size(); i++) {
  user_id = listChild.get(i));
  Map userProfile = runSQL("SELECT * FROM user_profile WHERE user_id = "+user_id);
  subList.add(userProfile);
  getAllChild(subList, user_id, depth);
 }
}

或者返回subList作为结果，并将其添加到调用方法的subList中:

List<Map> getAllChild(int parent_id, int depth) {

 // check whether maximum depth is reached
 if (depth == 0) {
   return Collections.EMPTY_LIST;
 }
 depth--;

 int count = 0, user_id = 0;
 List listChild = runSQL("SELECT user_id FROM parent_user WHERE parent_id = "+parent_id);
 List<Map> subList;
 for (int i = 0; i < listChild.size(); i++) {
  user_id = listChild.get(i));
  Map userProfile = runSQL("SELECT * FROM user_profile WHERE user_id = "+user_id);
  subList.add(userProfile);
  subList.addAll(getAllChild(user_id, depth));
 }
 return subList;
}