如何在不修改Substrate Get trait的前提下动态设置返回值？

需求说明 #

需要修改给定的Rust/Substrate代码，实现以下目标：

• 在业务模块中将最大值设为500
• 在单元测试的mock runtime中将最大值设为1000
• 不能修改Substrate原生的Get trait
• 希望配套set_value函数，但受限于Get::get是静态方法，无法传递&self，需要可行的替代方案

原始代码 #

// SUBSTRATE BEGIN
//
/// A trait for querying a single value from a type.
///
/// It is not required that the value is constant.
//
pub trait Get<T> {
    /// Return the current value.
    fn get() -> T;
}

impl<T: Default> Get<T> for () {
    fn get() -> T {
        T::default()
    }
}
// SUBSTRATE END

pub struct MaxRecordsTrait {
    value: u32,
}

impl MaxRecordsTrait {
    pub fn get_value() -> u32 {
        return 8000;
    }
}

impl Get<Option<u32>> for MaxRecordsTrait {
    fn get() -> Option<u32> {
        return Some(Self::get_value());
    }
}

可行解决方案 #

方案1：静态全局变量+线程同步 #

利用静态全局变量存储最大值，通过线程安全的锁（如Mutex）实现读写操作，既满足Get trait的静态方法要求，又支持动态修改值。

// SUBSTRATE BEGIN
pub trait Get<T> {
    fn get() -> T;
}

impl<T: Default> Get<T> for () {
    fn get() -> T {
        T::default()
    }
}
// SUBSTRATE END

use std::sync::Mutex;

// 静态全局变量存储最大值，初始值为8000
static MAX_RECORDS: Mutex<u32> = Mutex::new(8000);

pub struct MaxRecordsTrait;

impl MaxRecordsTrait {
    // 静态方法：设置最大值
    pub fn set_value(v: u32) {
        *MAX_RECORDS.lock().unwrap() = v;
    }

    // 静态方法：获取当前最大值
    pub fn get_value() -> u32 {
        *MAX_RECORDS.lock().unwrap()
    }
}

impl Get<Option<u32>> for MaxRecordsTrait {
    fn get() -> Option<u32> {
        Some(Self::get_value())
    }
}

使用方式：

• 业务模块初始化时调用MaxRecordsTrait::set_value(500)
• 单元测试的mock runtime中调用MaxRecordsTrait::set_value(1000)

注意：测试环境中建议使用expect替代unwrap，避免锁异常导致测试崩溃。

方案2：编译时宏生成不同实现 #

通过自定义宏，为不同场景（业务模块/测试）生成独立的Get实现，编译时就确定对应的值，无运行时开销。

// SUBSTRATE BEGIN
pub trait Get<T> {
    fn get() -> T;
}

impl<T: Default> Get<T> for () {
    fn get() -> T {
        T::default()
    }
}
// SUBSTRATE END

// 定义宏：生成指定最大值的结构体及Get实现
macro_rules! define_max_records {
    ($struct_name:ident, $max_value:expr) => {
        pub struct $struct_name;

        impl $struct_name {
            pub fn get_value() -> u32 {
                $max_value
            }
        }

        impl Get<Option<u32>> for $struct_name {
            fn get() -> Option<u32> {
                Some(Self::get_value())
            }
        }
    };
}

// 业务模块使用：最大值设为500
define_max_records!(ModuleMaxRecords, 500);

// 测试环境使用：最大值设为1000（仅编译测试代码时生效）
#[cfg(test)]
define_max_records!(TestMaxRecords, 1000);

使用方式：

• 业务模块中直接使用ModuleMaxRecords::get()获取值
• 测试代码中使用TestMaxRecords::get()获取测试专用值

方案3：Substrate框架配置Trait方案（推荐用于Frame模块） #

如果你的代码是Substrate Frame模块的一部分，可以通过模块的Config trait将最大值作为配置项，不同runtime（业务/测试）实现不同的配置值。

// SUBSTRATE BEGIN
pub trait Get<T> {
    fn get() -> T;
}

impl<T: Default> Get<T> for () {
    fn get() -> T {
        T::default()
    }
}

// 模拟Frame模块的Config trait结构
pub trait Config {
    type MaxRecords: Get<Option<u32>>;
}
// SUBSTRATE END

// 业务模块的Config实现：绑定最大值为500的Get实现
pub struct ModuleConfig;
impl Config for ModuleConfig {
    type MaxRecords = ModuleMaxRecords;
}

pub struct ModuleMaxRecords;
impl Get<Option<u32>> for ModuleMaxRecords {
    fn get() -> Option<u32> {
        Some(500)
    }
}

// 测试Mock Runtime的Config实现：绑定最大值为1000的Get实现
#[cfg(test)]
pub struct MockConfig;
#[cfg(test)]
impl Config for MockConfig {
    type MaxRecords = TestMaxRecords;
}

#[cfg(test)]
pub struct TestMaxRecords;
#[cfg(test)]
impl Get<Option<u32>> for TestMaxRecords {
    fn get() -> Option<u32> {
        Some(1000)
    }
}

优势：符合Substrate框架的设计规范，通过配置隔离不同环境的参数，是Frame模块的标准做法。

方案选择建议 #

• 需要动态调整最大值：选择方案1
• 仅需不同编译环境的静态配置：选择方案2
• 基于Substrate Frame开发模块：优先选择方案3

内容的提问来源于stack exchange，提问作者Robert La Ferla