php 无限级分类处理

/**
     * 无限级分类处理
     * @param array $data 数据源
     * @param string $idName 主键
     * @param string $fieldName 父级字段
     * @param string $childrenKey 子级字段名
     * @return array
     */
    public function formatCategory(array $data, string $idName = "id", string $fieldName = 'pid', $childrenKey = 'children')
    {
        $items = [];
        foreach ($data as $item) {
            $items[$item[$idName]] = $item;
        }
        $result = array();
        foreach ($items as $item) {
            if (isset($items[$item[$fieldName]])) {
                $items[$item[$fieldName]][$childrenKey][] = &$items[$item[$idName]];
            } else if ($item[$fieldName] == 0) {
                $result[] = &$items[$item[$idName]];
            }
        }
        return $result;
    }