Rust no_std crate中基于feature简化泛型默认类型声明

不需要编写过程宏，用Rust内置的cfg属性和默认泛型参数就能简洁实现你的需求。

解决方案代码 #

// 假设StdAllocator在std_alloc特性下定义并实现Allocator
#[cfg(feature = "std_alloc")]
pub struct StdAllocator;
#[cfg(feature = "std_alloc")]
impl Allocator for StdAllocator {
    // 实现Allocator trait的必要方法
}

pub trait Allocator {
    // 定义Allocator trait的核心接口
}

pub struct DynamicArray<T, TAllocator = #[cfg(feature = "std_alloc")] StdAllocator>
where
    TAllocator: Allocator,
{
    // 结构体核心字段示例
    data: *mut T,
    allocator: TAllocator,
    len: usize,
    cap: usize,
}

impl<T, TAllocator> DynamicArray<T, TAllocator>
where
    TAllocator: Allocator,
{
    pub fn new(allocator: TAllocator) -> Self {
        // 初始化逻辑
        unsafe {
            DynamicArray {
                data: std::ptr::null_mut(),
                allocator,
                len: 0,
                cap: 0,
            }
        }
    }

    // 启用std_alloc时提供简化构造函数
    #[cfg(feature = "std_alloc")]
    pub fn new_default() -> Self {
        Self::new(StdAllocator)
    }

    pub fn push(&mut self, item: T) -> Result<(), ()> {
        // 带分配失败处理的push逻辑，返回Result
        Ok(())
    }
}

工作原理 #

• 当启用std_alloc特性时，#[cfg(feature = "std_alloc")]标记的默认参数会保留，结构体泛型等效于TAllocator = StdAllocator，用户可直接写DynamicArray<i32>自动使用默认分配器。
• 当未启用std_alloc特性时，该默认参数会被移除，用户必须显式指定分配器类型（如DynamicArray<i32, MyCustomAllocator>），否则触发编译错误。

额外优化 #

在std_alloc特性下提供无需传入分配器的构造函数（如示例中的new_default），可以进一步降低用户的使用成本。

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