Laravel嵌套集合指定属性平均值计算功能故障排查

原代码的核心问题是递归调用时仅返回计算结果，无法传递有效元素计数，导致平均值计算只能硬编码除数。要解决这个问题，需要让递归逻辑同时返回总和和有效元素数量，再在顶层根据操作类型计算最终结果。

修改后的完整代码 #

/**
 * calculate the sum/AVG based on the chosen property on a single relation or nested relations.
 * 
 * @var \Illuminate\Support\Collection | array $entity
 * @var string $path
 * @var string $property
 * @var 'sum'|'avg' $operation
 * @return float
 */
public static function compute($entity, string $path, string $property, string $operation = 'sum'): float
{
    [$sum, $count] = self::computeRaw($entity, $path, $property);
    
    if ($operation === 'avg') {
        return $count > 0 ? $sum / $count : 0.0;
    }
    
    return $sum;
}

/**
 * 递归计算总和与有效元素数量
 * 
 * @param \Illuminate\Support\Collection|array|object $entity
 * @param string $path
 * @param string $property
 * @return array [sum, count]
 */
private static function computeRaw($entity, string $path, string $property): array
{
    $parts    = explode('.', $path, 2);
    $relation = $parts[0];
    $newPath  = $parts[1] ?? '';
    $sum      = 0.0;
    $count    = 0;

    // 处理集合/数组类型的实体，遍历每个元素递归计算
    if (is_iterable($entity)) {
        foreach ($entity as $item) {
            [$itemSum, $itemCount] = self::computeRaw($item, $path, $property);
            $sum += $itemSum;
            $count += $itemCount;
        }
        return [$sum, $count];
    }

    // 实体不存在对应关联，直接返回空结果
    if (!isset($entity->{$relation})) {
        return [$sum, $count];
    }

    $relationValue = $entity->{$relation};

    // 处理关联是集合/数组的情况
    if (is_iterable($relationValue)) {
        foreach (collect($relationValue) as $item) {
            if ($newPath === '') {
                // 已到最后一层路径，直接取属性值并转成数值类型
                $value = $item->{$property} ?? 0;
                $sum += is_numeric($value) ? (float)$value : 0;
                $count++;
            } else {
                // 还有嵌套路径，继续递归
                [$itemSum, $itemCount] = self::computeRaw($item, $newPath, $property);
                $sum += $itemSum;
                $count += $itemCount;
            }
        }
    } else {
        // 关联是对象的情况
        if ($newPath === '') {
            $value = $relationValue->{$property} ?? 0;
            $sum += is_numeric($value) ? (float)$value : 0;
            $count++;
        } else {
            [$itemSum, $itemCount] = self::computeRaw($relationValue, $newPath, $property);
            $sum += $itemSum;
            $count += $itemCount;
        }
    }

    return [$sum, $count];
}

关键修改说明 #

1. 拆分递归逻辑：新增computeRaw私有方法，专门负责递归遍历嵌套结构，返回包含总和和有效元素数量的数组，解决了递归过程中计数无法传递的问题。
2. 顶层处理运算：原compute方法作为入口，仅根据computeRaw的结果计算最终的总和或平均值，避免在递归过程中提前计算平均值导致的嵌套场景错误。
3. 类型兼容处理：增加了字符串转数值的逻辑，确保测试数据中类似"990"的字符串数字能正确参与计算。
4. 边界场景处理：新增关联不存在的判断，避免因属性不存在导致的报错。

测试验证 #

• 单路径测试：sum结果为73258+73235+73143+73127=292763，avg结果为292763/4=73190.75，完全符合预期。
• 嵌套路径测试：sum结果为-990+990+1250+690=1940，avg结果为1940/4=485，完全符合预期。

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