Rust 如何使用宏生成编译时唯一整数

发布于03月14日

我需要一个程序的几个部分，在不同的模块中，有一个唯一的整数.

例如:

pub fn foo() -> u64 {
    unique_integer!()
}

pub fn bar() -> u64 {
    unique_integer!()
}

(foo()永远不应该返回与bar()相同的值，但值本身是无意义的，不需要跨构建保持稳定.foo()的所有调用都必须返回相同的值，对bar()的所有调用都必须返回相同的值.这些值最好是连续的，但不是必需的.)

有没有办法使用宏来实现这一点？

推荐答案

使用静态原子变量作为计数器，可以对once_cell执行类似操作:

use core::sync::atomic::{Ordering, AtomicU64};
use once_cell::sync::Lazy;

static COUNTER: AtomicU64 = AtomicU64::new(0);

fn foo() -> u64 {
    static LOCAL_COUNTER: Lazy<u64> = Lazy::new(|| COUNTER.fetch_add(1, Ordering::Relaxed));
    *LOCAL_COUNTER
}

fn bar() -> u64 {
    static LOCAL_COUNTER: Lazy<u64> = Lazy::new(|| COUNTER.fetch_add(1, Ordering::Relaxed));
    *LOCAL_COUNTER
}

fn main() {
    dbg!(foo()); // 0
    dbg!(foo()); // still 0
    dbg!(bar()); // 1
    dbg!(foo()); // unchanged - 0
    dbg!(bar()); // unchanged - 1
}

Playground

是的，重复的代码可以像往常一样包装在宏中:

macro_rules! unique_integer {
    () => {{
        static LOCAL_COUNTER: Lazy<u64> = Lazy::new(|| COUNTER.fetch_add(1, Ordering::Relaxed));
        *LOCAL_COUNTER
    }}
}

fn foo() -> u64 {
    unique_integer!()
}

fn bar() -> u64 {
    unique_integer!()
}