@@ -2,7 +2,9 @@ use std::time::Duration;
22
33use backoff:: { backoff:: Backoff , ExponentialBackoff } ;
44use futures_util:: StreamExt ;
5- use pyth_lazer_protocol:: api:: { SubscribeRequest , SubscriptionId , UnsubscribeRequest , WsRequest } ;
5+ use pyth_lazer_protocol:: subscription:: {
6+ Request , SubscribeRequest , SubscriptionId , UnsubscribeRequest ,
7+ } ;
68use tokio:: { pin, select, sync:: mpsc, time:: Instant } ;
79use tracing:: { error, info, warn} ;
810use url:: Url ;
@@ -16,7 +18,7 @@ use anyhow::{bail, Context, Result};
1618const BACKOFF_RESET_DURATION : Duration = Duration :: from_secs ( 10 ) ;
1719
1820pub struct PythLazerResilientWSConnection {
19- request_sender : mpsc:: Sender < WsRequest > ,
21+ request_sender : mpsc:: Sender < Request > ,
2022}
2123
2224impl PythLazerResilientWSConnection {
@@ -51,17 +53,15 @@ impl PythLazerResilientWSConnection {
5153
5254 pub async fn subscribe ( & mut self , request : SubscribeRequest ) -> Result < ( ) > {
5355 self . request_sender
54- . send ( WsRequest :: Subscribe ( request) )
56+ . send ( Request :: Subscribe ( request) )
5557 . await
5658 . context ( "Failed to send subscribe request" ) ?;
5759 Ok ( ( ) )
5860 }
5961
6062 pub async fn unsubscribe ( & mut self , subscription_id : SubscriptionId ) -> Result < ( ) > {
6163 self . request_sender
62- . send ( WsRequest :: Unsubscribe ( UnsubscribeRequest {
63- subscription_id,
64- } ) )
64+ . send ( Request :: Unsubscribe ( UnsubscribeRequest { subscription_id } ) )
6565 . await
6666 . context ( "Failed to send unsubscribe request" ) ?;
6767 Ok ( ( ) )
@@ -95,7 +95,7 @@ impl PythLazerResilientWSConnectionTask {
9595 pub async fn run (
9696 & mut self ,
9797 response_sender : mpsc:: Sender < AnyResponse > ,
98- request_receiver : & mut mpsc:: Receiver < WsRequest > ,
98+ request_receiver : & mut mpsc:: Receiver < Request > ,
9999 ) -> Result < ( ) > {
100100 loop {
101101 let start_time = Instant :: now ( ) ;
@@ -128,7 +128,7 @@ impl PythLazerResilientWSConnectionTask {
128128 pub async fn start (
129129 & mut self ,
130130 sender : mpsc:: Sender < AnyResponse > ,
131- request_receiver : & mut mpsc:: Receiver < WsRequest > ,
131+ request_receiver : & mut mpsc:: Receiver < Request > ,
132132 ) -> Result < ( ) > {
133133 let mut ws_connection =
134134 PythLazerWSConnection :: new ( self . endpoint . clone ( ) , self . access_token . clone ( ) ) ?;
@@ -137,7 +137,7 @@ impl PythLazerResilientWSConnectionTask {
137137
138138 for subscription in self . subscriptions . clone ( ) {
139139 ws_connection
140- . send_request ( WsRequest :: Subscribe ( subscription) )
140+ . send_request ( Request :: Subscribe ( subscription) )
141141 . await ?;
142142 }
143143 loop {
@@ -167,10 +167,10 @@ impl PythLazerResilientWSConnectionTask {
167167 }
168168 Some ( request) = request_receiver. recv( ) => {
169169 match request {
170- WsRequest :: Subscribe ( request) => {
170+ Request :: Subscribe ( request) => {
171171 self . subscribe( & mut ws_connection, request) . await ?;
172172 }
173- WsRequest :: Unsubscribe ( request) => {
173+ Request :: Unsubscribe ( request) => {
174174 self . unsubscribe( & mut ws_connection, request) . await ?;
175175 }
176176 }
0 commit comments