Rust数据校验库near-schema-checker-lib的使用：高效实现Near协议数据结构与模式验证

完整示例demo

以下是一个完整的NEAR合约示例，展示如何使用near-schema-checker-lib验证合约方法参数：

use near_sdk::borsh::{self, BorshDeserialize, BorshSerialize};
use near_sdk::{env, near_bindgen, serde_json};
use near_schema_checker_lib::{Schema, SchemaType, CustomValidator};
use lazy_static::lazy_static;

// 定义自定义验证器检查NEAR账户ID格式
fn validate_near_account(value: &serde_json::Value) -> Result<(), String> {
    if let Some(s) = value.as_str() {
        if s.ends_with(".near") {
            Ok(())
        } else {
            Err("账户ID必须以.near结尾".to_string())
        }
    } else {
        Err("账户ID必须是字符串".to_string())
    }
}

// 预编译常用Schema实例
lazy_static! {
    static ref TRANSFER_SCHEMA: Schema = Schema::new()
        .field("receiver_id", SchemaType::Custom(Box::new(CustomValidator::new(validate_near_account))))
        .field("amount", SchemaType::U128)
        .field("memo", SchemaType::Optional(Box::new(SchemaType::String)));
    
    static ref NFT_METADATA_SCHEMA: Schema = Schema::new()
        .field("title", SchemaType::String)
        .field("description", SchemaType::Optional(Box::new(SchemaType::String)))
        .field("media", SchemaType::String)
        .field("media_hash", SchemaType::Optional(Box::new(SchemaType::String)));
}

#[near_bindgen]
#[derive(BorshDeserialize, BorshSerialize)]
pub struct Contract {
    pub owner_id: String,
}

#[near_bindgen]
impl Contract {
    #[init]
    pub fn new(owner_id: String) -> Self {
        // 验证owner_id格式
        let owner_schema = Schema::new()
            .field("owner_id", SchemaType::Custom(Box::new(CustomValidator::new(validate_near_account))));
        
        owner_schema.validate(&serde_json::json!({"owner_id": owner_id.clone()}))
            .expect("无效的owner_id格式");

        Self { owner_id }
    }

    // 转账方法
    pub fn transfer(&mut self, params: serde_json::Value) {
        // 验证转账参数
        TRANSFER_SCHEMA.validate(&params)
            .unwrap_or_else(|e| env::panic_str(&format!("无效的转账参数: {}", e)));
        
        // 这里添加实际的转账逻辑...
        env::log_str("转账操作已验证通过");
    }

    // 创建NFT方法
    pub fn mint_nft(&mut self, metadata: serde_json::Value) {
        // 验证NFT元数据
        NFT_METADATA_SCHEMA.validate(&metadata)
            .unwrap_or_else(|e| env::panic_str(&format!("无效的NFT元数据: {}", e)));
        
        // 这里添加实际的NFT铸造逻辑...
        env::log_str("NFT元数据已验证通过");
    }
}

// 测试模块
#[cfg(test)]
mod tests {
    use super::*;
    use near_sdk::test_utils::{get_logs, VMContextBuilder};
    use near_sdk::{testing_env, AccountId, VMContext};

    fn get_context() -> VMContext {
        VMContextBuilder::new()
            .signer_account_id(AccountId::new_unchecked("alice.near".to_string()))
            .build()
    }

    #[test]
    fn test_transfer() {
        let context = get_context();
        testing_env!(context);
        
        let mut contract = Contract::new("owner.near".to_string());
        
        // 测试有效转账
        let valid_transfer = serde_json::json!({
            "receiver_id": "bob.near",
            "amount": "1000000000000000000000000"
        });
        contract.transfer(valid_transfer);
        assert_eq!(get_logs(), vec!["转账操作已验证通过"]);
        
        // 测试无效转账 - 会panic
        let invalid_transfer = serde_json::json!({
            "receiver_id": "invalid_account",
            "amount": "100"
        });
        let result = std::panic::catch_unwind(|| {
            contract.transfer(invalid_transfer);
        });
        assert!(result.is_err());
    }

    #[test]
    fn test_mint_nft() {
        let context = get_context();
        testing_env!(context);
        
        let mut contract = Contract::new("owner.near".to_string());
        
        // 测试有效NFT元数据
        let valid_metadata = serde_json::json!({
            "title": "My NFT",
            "media": "https://example.com/nft.jpg"
        });
        contract.mint_nft(valid_metadata);
        assert_eq!(get_logs(), vec!["NFT元数据已验证通过"]);
    }
}

这个完整示例展示了：

1. 如何使用自定义验证器验证NEAR账户ID格式
2. 如何使用预编译Schema实例提高性能
3. 如何在合约方法中验证输入参数
4. 如何编写测试用例验证数据校验功能
5. 如何处理验证失败的情况

您可以根据实际需求修改Schema定义和验证逻辑，以适应不同的NEAR合约开发场景。