once_cell项目中OnceNonZeroUsize的性能优化探讨

在并发编程中，once_cell库提供的OnceNonZeroUsize类型是一个非常有用的工具，它允许线程安全地初始化并存储一个非零的usize值。本文将深入分析该类型的性能特点，并探讨如何通过添加get_unchecked()方法来进一步优化性能。

性能现状分析

通过对OnceNonZeroUsize与类似类型spin::Once<NonZeroUsize>的对比测试，我们发现：

1. 初始化性能优势：OnceNonZeroUsize::get_or_init()比spin::Once的对应方法更高效，因为它只需要一次内存加载而非两次。

2. 单次读取性能相当：单次调用OnceNonZeroUsize::get()与spin::Once::get_unchecked()性能相近。

3. 重复读取性能问题：多次调用OnceNonZeroUsize::get()时，由于使用原子读取操作，每次调用都会导致内存加载，而spin::Once::get_unchecked()则允许编译器优化掉冗余加载。

引入get_unchecked的必要性

在某些使用场景中，开发者可以确保get_or_init()已经被调用过，此时使用原子操作进行读取就成为了不必要的性能开销。为此，我们建议为OnceNonZeroUsize添加一个get_unchecked()方法。

该方法的设计考虑了两个关键点：

1. 安全性：该方法标记为unsafe，因为调用者必须确保值已经被初始化，否则会导致未定义行为。

2. 性能：该方法使用普通内存读取而非原子操作，避免了不必要的性能开销。

方法签名设计讨论

关于get_unchecked()的返回类型，存在两种观点：

1. 直接返回NonZeroUsize：这种设计更符合"unchecked"的语义，既移除了原子加载，也移除了零值检查。

2. 返回usize：这种设计更灵活，允许调用者自行决定如何处理可能的零值情况。

最终，项目维护者选择了第一种方案，认为应该通过精确的签名表达安全契约，而不是进行防御性编程。如果用户需要安全保证，应该继续使用get().unwrap()并承担原子加载的开销。

实现细节

get_unchecked()的实现核心是直接读取底层内存值，并假设该值已经被正确初始化：

pub unsafe fn get_unchecked(&self) -> NonZeroUsize {
    let p = self.inner.as_ptr();
    let value = unsafe { p.read() };
    unsafe { NonZeroUsize::new_unchecked(value) }
}

这种实现方式完全移除了原子操作的开销，为性能敏感的场景提供了更好的优化空间。

使用建议

开发者在使用get_unchecked()时应当注意：

1. 确保在调用前已经通过get_or_init()完成了初始化
2. 理解如果违反上述保证会导致未定义行为
3. 在性能关键路径上使用该方法，而非性能敏感区域仍应使用安全的get()

通过合理使用这一优化，可以在保证线程安全的同时，获得更好的性能表现。