Rust运算符重载中&T、&mut T等类型匹配问题求助

当你添加了impl AddAssign<Vec3> for &mut Vec3之后，Rust的类型推断系统在处理&mut Vec3 += &mut Vec3时陷入了匹配困境：

1. 原本没有这个新impl时，编译器会自动将右侧的&mut Vec3隐式转换为&Vec3（因为可变引用可以安全转为不可变引用），从而匹配你已实现的AddAssign<&Vec3> for &mut Vec3。
2. 但新增的AddAssign<Vec3> for &mut Vec3让编译器尝试另一条路径：能否将右侧的&mut Vec3转换为Vec3？这显然不可能——因为&mut Vec3是引用，无法直接移动所有权得到Vec3。此时编译器不会再回退尝试之前的隐式转换路径，而是直接报错说找不到AddAssign<&mut Vec3> for &mut Vec3的实现。

更符合Rust最佳实践的方式是只为Vec3本身实现AddAssign，让引用类型通过自动解引用（Deref Coercion）来复用这些实现，避免为引用类型重复编写代码。

步骤1：重构AddAssign实现 #

首先移除针对&mut Vec3的AddAssign实现，改为为Vec3实现两种AddAssign：

// 实现 Vec3 += Vec3
impl AddAssign<Vec3> for Vec3 {
    fn add_assign(&mut self, rhs: Vec3) {
        // 复用下面的引用实现
        *self += &rhs;
    }
}

// 实现 Vec3 += &Vec3（自动兼容&mut Vec3，因为可变引用可转为不可变引用）
impl AddAssign<&Vec3> for Vec3 {
    fn add_assign(&mut self, rhs: &Vec3) {
        for i in 0..3 {
            self[i] += rhs[i];
        }
    }
}

步骤2：处理引用的运算符场景 #

现在，无论是Vec3 += Vec3、&mut Vec3 += Vec3、Vec3 += &Vec3还是&mut Vec3 += &mut Vec3，编译器都会自动通过解引用匹配到上面的实现：

• 对于&mut Vec3类型的左值，编译器会自动解引用为Vec3的可变引用，调用Vec3的add_assign方法。
• 对于&mut Vec3类型的右值，编译器会自动转为&Vec3，匹配AddAssign<&Vec3> for Vec3。

额外优化：简化Neg实现 #

你的Neg实现可以更简洁高效，不需要修改原对象，直接返回新的Vec3：

impl Neg for Vec3 {
    type Output = Self;
    fn neg(self) -> Self::Output {
        Vec3([-self.0[0], -self.0[1], -self.0[2]])
    }
}

测试验证 #

修改后的测试代码可以正常运行：

#[cfg(test)]
mod tests {
    use super::Vec3;
    #[test]
    fn test_add() {
        let mut v = -Vec3([1f32, 2f32, 3f32]);
        let mut w = -Vec3([1f32, 2f32, 3f32]);
        {
            let mut vv = &mut v;
            let ww = &mut w;
            vv += ww;
            println!("{:?}", vv); // 输出 Vec3([-2.0, -4.0, -6.0])
        }
    }
}

• 减少重复代码：不需要为引用类型单独实现运算符，复用值类型的实现。
• 符合Rust的设计哲学：利用自动解引用让代码更简洁，同时保持类型安全。
• 避免类型推断歧义：不会因为新增实现导致原有代码的匹配逻辑被破坏。

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