chore: support CN endpoints 2

longportapp · Dec 14, 2023 · 9e1613e · 9e1613e
1 parent be4df67
commit 9e1613e
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/rust/src/config.rs b/rust/src/config.rs
@@ -52,8 +52,18 @@ impl Config {
  ) -> Self {
  Self {
  http_cli_config: HttpClientConfig::new(app_key, app_secret, access_token),
- quote_ws_url: QUOTE_WS_URL.to_string(),
- trade_ws_url: TRADE_WS_URL.to_string(),
+ quote_ws_url: if is_cn() {
+ CN_QUOTE_WS_URL
+ } else {
+ QUOTE_WS_URL
+ }
+ .to_string(),
+ trade_ws_url: if is_cn() {
+ CN_TRADE_WS_URL
+ } else {
+ TRADE_WS_URL
+ }
+ .to_string(),
  language: Language::EN,
  }
  }