wxpay-rs 2.0.1

WeChat Pay API v3 Rust SDK
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

//! 凭证管理模块
//!
//! 提供微信支付凭证的管理功能。

use std::sync::Arc;
use tokio::sync::RwLock;

use crate::config::WxPayConfig;

/// 微信支付凭证
///
/// 封装了商户号、应用 ID 等凭证信息。
#[derive(Debug, Clone)]
pub struct Credentials {
    /// 应用 ID
    app_id: String,

    /// 商户号
    merchant_id: String,

    /// API v3 密钥
    api_v3_key: String,

    /// 证书序列号
    cert_serial_number: String,
}

impl Credentials {
    /// 从配置创建凭证
    pub fn from_config(config: &WxPayConfig) -> Self {
        Self {
            app_id: config.app_id.clone(),
            merchant_id: config.merchant_id.clone(),
            api_v3_key: config.api_v3_key.clone(),
            cert_serial_number: config.cert_serial_number.clone(),
        }
    }

    /// 获取应用 ID
    pub fn app_id(&self) -> &str {
        &self.app_id
    }

    /// 获取商户号
    pub fn merchant_id(&self) -> &str {
        &self.merchant_id
    }

    /// 获取 API v3 密钥
    pub fn api_v3_key(&self) -> &str {
        &self.api_v3_key
    }

    /// 获取证书序列号
    pub fn cert_serial_number(&self) -> &str {
        &self.cert_serial_number
    }
}

/// 凭证提供者
///
/// 提供动态获取凭证的能力，支持凭证刷新。
#[derive(Debug)]
pub struct CredentialProvider {
    /// 凭证
    credentials: Arc<RwLock<Credentials>>,
}

impl CredentialProvider {
    /// 创建新的凭证提供者
    pub fn new(config: &WxPayConfig) -> Self {
        let credentials = Credentials::from_config(config);
        Self {
            credentials: Arc::new(RwLock::new(credentials)),
        }
    }

    /// 获取凭证
    pub async fn get_credentials(&self) -> Credentials {
        self.credentials.read().await.clone()
    }

    /// 更新凭证
    pub async fn update_credentials(&self, credentials: Credentials) {
        let mut current = self.credentials.write().await;
        *current = credentials;
    }
}

impl Clone for CredentialProvider {
    fn clone(&self) -> Self {
        Self {
            credentials: self.credentials.clone(),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn create_test_config() -> WxPayConfig {
        WxPayConfig::builder()
            .app_id("wx88888888")
            .merchant_id("1900000109")
            .api_v3_key("abcdefghijklmnopqrstuvwxyz123456")
            .private_key(vec![1, 2, 3, 4])
            .cert_serial_number("CERT123456")
            .build()
            .unwrap()
    }

    #[test]
    fn test_credentials_from_config() {
        let config = create_test_config();
        let credentials = Credentials::from_config(&config);

        assert_eq!(credentials.app_id(), "wx88888888");
        assert_eq!(credentials.merchant_id(), "1900000109");
        assert_eq!(credentials.api_v3_key(), "abcdefghijklmnopqrstuvwxyz123456");
        assert_eq!(credentials.cert_serial_number(), "CERT123456");
    }

    #[tokio::test]
    async fn test_credential_provider() {
        let config = create_test_config();
        let provider = CredentialProvider::new(&config);

        let credentials = provider.get_credentials().await;
        assert_eq!(credentials.app_id(), "wx88888888");
    }

    #[tokio::test]
    async fn test_credential_provider_update() {
        let config = create_test_config();
        let provider = CredentialProvider::new(&config);

        let new_credentials = Credentials {
            app_id: "wx99999999".to_string(),
            merchant_id: "1900000209".to_string(),
            api_v3_key: "abcdefghijklmnopqrstuvwxyz123456".to_string(),
            cert_serial_number: "CERT999999".to_string(),
        };

        provider.update_credentials(new_credentials).await;

        let credentials = provider.get_credentials().await;
        assert_eq!(credentials.app_id(), "wx99999999");
        assert_eq!(credentials.merchant_id(), "1900000209");
    }
}