|
1 | 1 | use crate::error::{HeliusError, Result}; |
2 | 2 | use crate::rpc_client::RpcClient; |
3 | 3 | use crate::types::{Cluster, HeliusEndpoints, MintApiAuthority}; |
4 | | -use crate::websocket::{EnhancedWebsocket, ENHANCED_WEBSOCKET_URL}; |
| 4 | +use crate::websocket::EnhancedWebsocket; |
5 | 5 | use crate::Helius; |
6 | 6 | use reqwest::Client; |
7 | 7 | use solana_client::nonblocking::rpc_client::RpcClient as AsyncSolanaRpcClient; |
@@ -108,7 +108,7 @@ impl Config { |
108 | 108 | let client: Client = Client::builder().build().map_err(HeliusError::ReqwestError)?; |
109 | 109 | let rpc_client: Arc<RpcClient> = Arc::new(self.rpc_client_with_reqwest_client(client.clone())?); |
110 | 110 |
|
111 | | - let wss: String = format!("{}{}", ENHANCED_WEBSOCKET_URL, self.api_key); |
| 111 | + let wss: String = EnhancedWebsocket::get_url(&self.cluster, &self.api_key)?; |
112 | 112 | let ws_client: Arc<EnhancedWebsocket> = |
113 | 113 | Arc::new(EnhancedWebsocket::new(&wss, ping_interval_secs, pong_timeout_secs).await?); |
114 | 114 |
|
@@ -144,7 +144,7 @@ impl Config { |
144 | 144 | let async_solana_client = Arc::new(AsyncSolanaRpcClient::new(rpc_url.to_string())); |
145 | 145 |
|
146 | 146 | // Setup websocket |
147 | | - let wss: String = format!("{}{}", ENHANCED_WEBSOCKET_URL, self.api_key); |
| 147 | + let wss: String = EnhancedWebsocket::get_url(&self.cluster, &self.api_key)?; |
148 | 148 | let ws_client: Arc<EnhancedWebsocket> = |
149 | 149 | Arc::new(EnhancedWebsocket::new(&wss, ping_interval_secs, pong_timeout_secs).await?); |
150 | 150 |
|
|
0 commit comments