如何为含不同类型变体的Rust枚举实现Index trait？

你的代码存在三个核心问题：

1. 错误的SliceIndex绑定：你将Idx绑定到了SliceIndex<[usize]>，但实际需要的是能对[u8]和[u32]进行切片的索引类型（比如Range<usize>、RangeFull等）。
2. 不可变引用下的可变切片错误：index方法接收的是&self（不可变引用），但你试图从中获取&mut [u8]类型的切片——不可变引用无法访问内部可变引用的可变权限，这会触发编译错误。
3. Index trait的Output类型不匹配：Index要求返回现有数据的引用，但你试图返回临时创建的GlkArray实例的引用，这会导致生命周期悬空，因为临时实例会在函数结束后销毁。

方案一：实现自定义slice方法（推荐） #

既然你需要返回新的GlkArray实例而非引用，直接实现自定义方法更符合语义，也能避开Index trait的限制：

use std::ops::SliceIndex;

pub enum GlkArray<'a> {
    U8(&'a mut [u8]),
    U32(&'a mut [u32]),
}

impl<'a> GlkArray<'a> {
    pub fn slice<I>(&mut self, index: I) -> Self
    where
        I: SliceIndex<[u8], Output = [u8]> + SliceIndex<[u32], Output = [u32]>,
    {
        match self {
            GlkArray::U8(buf) => GlkArray::U8(&mut buf[index]),
            GlkArray::U32(buf) => GlkArray::U32(&mut buf[index]),
        }
    }
}

// 使用示例
fn main() {
    let mut u8_buf = [1u8, 2, 3, 4];
    let mut u32_buf = [10u32, 20, 30];
    
    let mut arr_u8 = GlkArray::U8(&mut u8_buf);
    let sliced_u8 = arr_u8.slice(1..3); // 得到GlkArray::U8(&mut [2,3])
    
    let mut arr_u32 = GlkArray::U32(&mut u32_buf);
    let sliced_u32 = arr_u32.slice(..2); // 得到GlkArray::U32(&mut [10,20])
}

该方案优势：

• 完全匹配你的需求，返回同变体的GlkArray实例
• 支持所有标准切片索引类型（范围、单个索引等）
• 规避了Index trait的引用语义限制

方案二：实现IndexMut trait（仅当需要索引语法时） #

如果坚持要用array[index]的语法，需要实现IndexMut（因为要操作可变切片），但需要调整设计——IndexMut的Output必须是单一类型，所以需要封装辅助枚举，且存在一定安全风险：

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

// 可变辅助枚举，作为IndexMut的Output
pub enum GlkArraySliceMut<'a> {
    U8(&'a mut [u8]),
    U32(&'a mut [u32]),
}

pub enum GlkArray<'a> {
    U8(&'a mut [u8]),
    U32(&'a mut [u32]),
}

impl<'a, I> IndexMut<I> for GlkArray<'a>
where
    I: SliceIndex<[u8]> + SliceIndex<[u32]>,
{
    type Output = GlkArraySliceMut<'a>;
    
    fn index_mut(&mut self, index: I) -> &mut Self::Output {
        // 使用Box::leak延长临时实例生命周期，存在内存泄漏风险，仅作演示
        match self {
            GlkArray::U8(buf) => Box::leak(Box::new(GlkArraySliceMut::U8(&mut buf[index]))),
            GlkArray::U32(buf) => Box::leak(Box::new(GlkArraySliceMut::U32(&mut buf[index]))),
        }
    }
}

⚠️ 注意：该方案使用Box::leak绕过生命周期限制，会导致内存泄漏，不推荐在生产代码中使用。

• SliceIndex<[u8]> + SliceIndex<[u32]>的绑定确保索引类型能同时作用于u8和u32切片，所有标准索引类型（如Range<usize>）都满足这个条件。
• Index/IndexMut的语义是引用现有数据，而你的需求是创建新的容器实例，因此自定义slice方法是更合理的选择。

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