PHP实现万级用户关系树的高效构建与优化

人亦已歌

PHP实现万级用户关系树的高效构建与优化

2. // mysql查询结果集类似于这样的二维数组， p_user_id 标记为上级用户ID, 无限级
3. array: [
4. 0 => array:3 [
5. "id" => "0"
6. "name" => "小赵"
7. "p_user_id" => ""
8. ],
9. 1 => array:3 [
10. "id" => "100"
11. "name" => "小钱"
12. "p_user_id" => "0"
13. ],
14. 2 => array:3 [
15. "id" => "1000"
16. "name" => "小孙"
17. "p_user_id" => "100"
18. ],
19. 3 => array:3 [
20. "id" => "10000"
21. "name" => "小李"
22. "p_user_id" => "1000"
23. ],
24. ……
25. ]

复制代码

2. // 将ID作为key，重组数组
3. $users = array_column($res,null,'id');
4. $result = [];
5. // 找出每个用户的下级
6. foreach($users as $id => $item)
7. {
8. $result[$item['p_user_id']][] = $item;
9. }
10. // 遍历成树状结构数组
11. $tree = $this->createTree($result, [$users[0]]);
12. // 如果需要用到类似于echart中树状结构图表，则移除不需要的key
13. $data = $this->removeKeys($tree);
14. // 最终echart中需要的json数组
15. $treeJson = json_encode($data);

复制代码

2. /
3. * 二维数组转树状结构数组
4. */
5. public function createTree(&$list, $parent){
6. $tree = array();
7. foreach ($parent as $k=>$l){
8. if(isset($list[$l['id']])){
9. $l['children'] = $this->createTree($list, $list[$l['id']]);
10. }
11. $tree[] = $l;
12. }
13. return $tree;
14. }

复制代码

2. /
3. * 移除不需要的key
4. */
5. public function removeKeys(&$arr)
6. {
7. foreach($arr as $key => &$item)
8. {
9. if(is_array($item))
10. {
11. unset($item['id']);
12. unset($item['p_user_id']);
13. if(isset($item['children']))
14. {
15. $this->removeKeys($item['children']);
16. }
17. }
18. }
19. return $arr;
20. }

复制代码

本例中，万级用户关系，mysql一次找出所有用户字段，再PHP遍历，再移除多余字段，总耗时也就是1秒左右，不过echart展示这么大的结构太慢了，瓶颈在前台渲染上面。

如果针对指定用户异步查询上下级，则还是用ajax异步直接查询mysql比较合理。