Rust中如何持久化非对象安全的dyn Distribution trait对象？

核心问题原因 #

rand的Distribution trait无法被用作 trait 对象（dyn），是因为其sample方法带有泛型参数R: Rng——这违反了Rust的对象安全规则：带泛型参数的方法无法被放入虚函数表（vtable），因此无法动态分发。

最优解决方案：自定义对象安全的Wrapper Trait #

通过创建一个对象安全的包装trait，将泛型方法转换为接受trait 对象的方法，就能实现持久化任意Distribution实例的需求。

步骤1：定义对象安全的Wrapper Trait

use rand::distributions::Distribution;
use rand::Rng;

// 自定义对象安全的分发 trait
pub trait DynDistribution<T> {
    fn sample_dyn(&self, rng: &mut dyn Rng) -> T;
}

// 为所有实现了Distribution的类型自动实现DynDistribution
impl<D, T> DynDistribution<T> for D
where
    D: Distribution<T>,
{
    fn sample_dyn(&self, rng: &mut dyn Rng) -> T {
        self.sample(rng)
    }
}

步骤2：修改结构体使用Wrapper Trait

pub struct Bandit<const K: usize> {
    levers: [f32; K],
    // 使用自定义的对象安全trait存储分布
    dist: Box<dyn DynDistribution<f32>>,
}

步骤3：使用示例

use rand_distr::{Normal, Uniform};
use rand::thread_rng;

fn main() {
    // 存储正态分布
    let normal_dist = Normal::new(0.0, 1.0).unwrap();
    let mut bandit_normal = Bandit {
        levers: [0.0; 3],
        dist: Box::new(normal_dist),
    };

    // 存储均匀分布
    let uniform_dist = Uniform::new(-1.0, 1.0);
    let mut bandit_uniform = Bandit {
        levers: [0.0; 3],
        dist: Box::new(uniform_dist),
    };

    // 采样示例
    let mut rng = thread_rng();
    let _sample1 = bandit_normal.dist.sample_dyn(&mut rng);
    let _sample2 = bandit_uniform.dist.sample_dyn(&mut rng);
}

方案优势 #

• 避免重复创建分布对象：一次性初始化后持久化存储，消除了重复创建的性能开销；
• 支持任意Distribution类型：只要是实现了Distribution的类型，都能被包装存储，无需修改枚举或添加新类型分支；
• 性能开销极小：仅增加一层trait方法的动态调用，远小于重复初始化分布的开销。

备选方案：固定Rng类型 #

如果你的场景只需要特定的Rng实现（比如ThreadRng），可以直接将sample方法的泛型参数固定为具体类型，简化实现：

use rand::distributions::Distribution;
use rand::rngs::ThreadRng;

pub trait FixedRngDistribution<T> {
    fn sample_fixed(&self, rng: &mut ThreadRng) -> T;
}

impl<D, T> FixedRngDistribution<T> for D
where
    D: Distribution<T>,
{
    fn sample_fixed(&self, rng: &mut ThreadRng) -> T {
        self.sample(rng)
    }
}

这种方案性能略优，但灵活性不如通用的dyn Rng版本。

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