如何使用 `index_mut` 获取可变引用?

How can I use `index_mut` to get a mutable reference?

即使我为我的结构实现 IndexMut,我也无法获得对结构内部向量元素的可变引用。

use std::ops::{Index, IndexMut};

struct Test<T> {
    data: Vec<T>,
}

impl<T> Index<usize> for Test<T> {
    type Output = T;
    fn index<'a>(&'a self, idx: usize) -> &'a T {
        return &self.data[idx];
    }
}

impl<T> IndexMut<usize> for Test<T> {
    fn index_mut<'a>(&'a mut self, idx: usize) -> &'a mut T {
        // even here I cannot get mutable reference to self.data[idx]
        return self.data.index_mut(idx);
    }
}

fn main() {
    let mut a: Test<i32> = Test { data: Vec::new() };
    a.data.push(1);
    a.data.push(2);
    a.data.push(3);
    let mut b = a[1];
    b = 10;

    // will print `[1, 2, 3]` instead of [1, 10, 3]
    println!("[{}, {}, {}]", a.data[0], a.data[1], a.data[2]);
}

如何使用 index_mut 获取可变引用?可能吗?

你快到了。改变这个:

let mut b = a[1];
b = 10;

对此:

let b = &mut a[1];
*b = 10;

索引语法。您的代码从向量中提取一个 i32 并修改变量 - 自然地,它不会影响向量本身。为了通过索引获取引用,需要显式写出来。

这很自然:当您使用索引访问切片或数组的元素时,您得到的是元素的值,而不是对它们的引用,为了获得引用,您需要显式地编写它.