Go math/rand/v2 PCG.Uint64 怎样设置确定性随机源

如果测试用例、数据回放或本地实验需要“每次启动拿到同一串随机数”，不要依赖 math/rand/v2 的顶层函数。更稳定的做法是显式创建 rand.NewPCG(seed1, seed2)，再交给 rand.New；两个 uint64 种子固定后，Uint64() 的调用顺序也固定，序列就可以重复。

这里的 PCG 是统计用途的伪随机源，不适合令牌、密钥或密码学安全场景。文章只讨论本地复现、测试和状态恢复。

用两个种子创建本地随机源

PCG 的入口是两个 uint64 参数，而不是旧版 math/rand 常见的一个 int64。先把随机源放在局部对象里，避免测试之间共享全局状态：

package main

import (
	"fmt"
	"math/rand/v2"
)

func main() {
	// 两个固定种子共同决定这次 PCG 的初始状态。
	pcg := rand.NewPCG(0x123456789abcdef0, 0x0fedcba987654321)
	// rand.New 负责把 Source 包装成可调用的随机数生成器。
	r := rand.New(pcg)

	// 每次调用都会推进同一个 PCG 状态，得到序列中的下一个 uint64。
	fmt.Println(r.Uint64())
	fmt.Println(r.Uint64())
}

同一组种子配合同样的两次调用，每次运行都会得到同样的两项输出。关键不只是“种子相同”，还包括调用顺序相同：如果中间多调用一次 Uint64，后续所有位置都会向后错一格。

为什么固定种子后可以复现

PCG 内部有 128 位状态，NewPCG(seed1, seed2) 用两个种子初始化它；Uint64 每次读取一个均匀分布的 uint64，同时推进内部状态。因此复现条件可以拆成三项：

• 两个种子完全一致，不能只复制其中一个。
• 使用同一个 PCG 实例，且调用顺序一致。
• 不要把并发任务随机调用混在同一实例中，否则调度顺序会改变序列位置。

这也是为什么测试代码更适合每个用例各自创建 rand.Rand。如果需要不同场景，给它们分配不同的种子对；如果需要业务安全随机数，则切换到 crypto/rand，不要把 PCG 当作安全替代品。

需要暂停时保存当前状态

只保存初始种子只能从头重放。如果任务已经消费了若干个随机值，想从当前点继续，应保存 PCG 的二进制状态：

package main

import (
	"fmt"
	"math/rand/v2"
)

func main() {
	// 先创建可复现的源，并消耗一个值模拟任务已经执行了一部分。
	pcg := rand.NewPCG(11, 29)
	_ = pcg.Uint64()

	// 保存当前状态；错误需要保留，因为状态是后续恢复的输入。
	snapshot, err := pcg.MarshalBinary()
	if err != nil {
		panic(err)
	}

	// 恢复到独立实例，下一次 Uint64 会接着快照位置继续。
	var resumed rand.PCG
	if err := resumed.UnmarshalBinary(snapshot); err != nil {
		panic(err)
	}
	fmt.Println(resumed.Uint64())
}

MarshalBinary 保存的是“当前进度”，不是一组新的种子；恢复时用 UnmarshalBinary 载入同一份字节数据。快照要和代码版本、任务输入一起管理，避免把不同实验的状态混用。

复现失败时先查这四个地方

相关问题

PCG 能不能只设置一个种子？

PCG 的公开构造函数需要两个 uint64 值。可以约定其中一个为常量，但那仍然是两个输入共同定义初始状态，不能把它理解成旧版单整数种子接口。

为什么保存快照后还要保留调用上下文？

快照只描述生成器当前状态；业务如何解释下一个随机值、使用了哪组输入，仍由任务上下文决定。恢复时应同时保存任务版本和消费位置。

总的来说，确定性随机源的最小组合是“固定的两个种子 + 独立的 PCG 实例 + 固定的调用顺序”。需要中途续跑时，再把当前状态序列化；需要安全随机时，则换用密码学随机源。