合并递归函数消除代码重复：计算矢量图层几何差

问题背景 #

你当前有两个用于计算矢量图层布尔差的递归函数：

• difference_recursive：接收&Geometry引用，返回Some(input_geom.clone())
• inner_difference_recursive：接收Geometry自有实例，返回Some(input_geom)
  首次调用时输入是FeatureIterator返回的引用（无法直接移动，解引用会触发编译错误），后续递归调用则使用自有数据，需要合并这两个函数消除重复代码。

解决方案：使用Cow<Geometry>实现克隆-on-write兼容 #

std::borrow::Cow是Rust中专门处理“仅在需要时克隆”的类型，它可以同时容纳引用和自有实例，完美适配你的场景。

合并后的递归函数示例

use std::borrow::Cow;
use your_crate::Geometry; // 替换为实际的Geometry类型路径

fn difference_recursive<'a>(
    mut input_geom: Cow<'a, Geometry>,
    other_geoms: &[Geometry],
    index: usize
) -> Option<Geometry> {
    // 递归终止条件：处理完所有待差集的几何
    if index >= other_geoms.len() {
        return Some(input_geom.into_owned());
    }

    // 执行布尔差计算（假设你有一个difference_with方法）
    match input_geom.difference_with(&other_geoms[index]) {
        // 差集结果为空，直接返回None
        None => None,
        // 差集结果存在，递归处理下一个几何
        Some(result) => {
            difference_recursive(Cow::Owned(result), other_geoms, index + 1)
        }
    }
}

// 对外暴露的入口函数，处理首次调用的引用场景
pub fn difference(input_geom: &Geometry, other_geoms: &[Geometry]) -> Option<Geometry> {
    difference_recursive(Cow::Borrowed(input_geom), other_geoms, 0)
}

关键逻辑说明 #

• 参数类型兼容：递归函数参数用Cow<'a, Geometry>，既能接收首次调用的&Geometry（通过Cow::Borrowed转换），也能接收后续递归的自有Geometry（通过Cow::Owned转换）。
• 返回值统一：递归终止时，用into_owned()将Cow转为自有Geometry，不管原本是引用还是自有实例，统一输出自有数据。
• 性能优化：只有当需要修改或返回自有实例时才会触发克隆，首次调用的引用在未修改前不会被克隆，避免不必要的性能开销。

备选方案：泛型参数配合特质约束 #

如果你不想使用Cow，也可以通过泛型约束让函数同时接受引用和自有类型：

fn difference_recursive<T>(input_geom: T, other_geoms: &[Geometry], index: usize) -> Option<Geometry>
where
    T: AsRef<Geometry> + Into<Option<Geometry>>,
{
    if index >= other_geoms.len() {
        return input_geom.into();
    }

    match input_geom.as_ref().difference_with(&other_geoms[index]) {
        None => None,
        Some(result) => difference_recursive(result, other_geoms, index + 1),
    }
}

// 入口函数
pub fn difference(input_geom: &Geometry, other_geoms: &[Geometry]) -> Option<Geometry> {
    // 对引用手动转为Option<Geometry>（克隆后传入）
    difference_recursive(Some(input_geom.clone()), other_geoms, 0)
}

不过这种方案需要在首次调用时主动克隆引用，性能不如Cow方案高效。

核心思路总结 #

无论是用Cow还是泛型特质，核心都是统一处理“引用”和“自有实例”两种输入类型，避免为两种类型编写重复的递归逻辑。Cow是Rust中处理这类场景的惯用方式，兼顾简洁性与性能。

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