diff --git a/src/gateway/backend_tungstenite.rs b/src/gateway/backend_tungstenite.rs index 53b6982..f99424d 100644 --- a/src/gateway/backend_tungstenite.rs +++ b/src/gateway/backend_tungstenite.rs @@ -11,16 +11,17 @@ use super::GatewayMessage; use crate::errors::GatewayError; #[derive(Debug, Clone)] -pub struct WebSocketBackend; +pub struct TungsteniteBackend; // These could be made into inherent associated types when that's stabilized -pub type WsSink = SplitSink>, tungstenite::Message>; -pub type WsStream = SplitStream>>; +pub type TungsteniteSink = + SplitSink>, tungstenite::Message>; +pub type TungsteniteStream = SplitStream>>; -impl WebSocketBackend { +impl TungsteniteBackend { pub async fn connect( websocket_url: &str, - ) -> Result<(WsSink, WsStream), crate::errors::GatewayError> { + ) -> Result<(TungsteniteSink, TungsteniteStream), crate::errors::GatewayError> { let mut roots = rustls::RootCertStore::empty(); for cert in rustls_native_certs::load_native_certs().expect("could not load platform certs") { diff --git a/src/gateway/gateway.rs b/src/gateway/gateway.rs index 9e3410c..e2923c9 100644 --- a/src/gateway/gateway.rs +++ b/src/gateway/gateway.rs @@ -6,7 +6,7 @@ use tokio::task; use self::event::Events; use super::*; -use super::{WsSink, WsStream}; +use super::{Sink, Stream}; use crate::types::{ self, AutoModerationRule, AutoModerationRuleUpdate, Channel, ChannelCreate, ChannelDelete, ChannelUpdate, Guild, GuildRoleCreate, GuildRoleUpdate, JsonField, RoleObject, SourceUrlField, @@ -17,8 +17,8 @@ use crate::types::{ pub struct Gateway { events: Arc>, heartbeat_handler: HeartbeatHandler, - websocket_send: Arc>, - websocket_receive: WsStream, + websocket_send: Arc>, + websocket_receive: Stream, kill_send: tokio::sync::broadcast::Sender<()>, store: Arc>>>>, url: String, diff --git a/src/gateway/handle.rs b/src/gateway/handle.rs index 9a3c509..620faba 100644 --- a/src/gateway/handle.rs +++ b/src/gateway/handle.rs @@ -14,7 +14,7 @@ use crate::types::{self, Composite}; pub struct GatewayHandle { pub url: String, pub events: Arc>, - pub websocket_send: Arc>, + pub websocket_send: Arc>, /// Tells gateway tasks to close pub(super) kill_send: tokio::sync::broadcast::Sender<()>, pub(crate) store: Arc>>>>, diff --git a/src/gateway/heartbeat.rs b/src/gateway/heartbeat.rs index a5875a4..b8e4bec 100644 --- a/src/gateway/heartbeat.rs +++ b/src/gateway/heartbeat.rs @@ -27,7 +27,7 @@ pub(super) struct HeartbeatHandler { impl HeartbeatHandler { pub fn new( heartbeat_interval: Duration, - websocket_tx: Arc>, + websocket_tx: Arc>, kill_rc: tokio::sync::broadcast::Receiver<()>, ) -> Self { let (send, receive) = tokio::sync::mpsc::channel(32); @@ -49,7 +49,7 @@ impl HeartbeatHandler { /// Can be killed by the kill broadcast; /// If the websocket is closed, will die out next time it tries to send a heartbeat; pub async fn heartbeat_task( - websocket_tx: Arc>, + websocket_tx: Arc>, heartbeat_interval: Duration, mut receive: Receiver, mut kill_receive: tokio::sync::broadcast::Receiver<()>, diff --git a/src/gateway/mod.rs b/src/gateway/mod.rs index 8314999..03ac502 100644 --- a/src/gateway/mod.rs +++ b/src/gateway/mod.rs @@ -23,11 +23,11 @@ use std::sync::{Arc, RwLock}; use tokio::sync::Mutex; #[cfg(not(target_arch = "wasm32"))] -pub type WsSink = backend_tungstenite::WsSink; +pub type Sink = backend_tungstenite::TungsteniteSink; #[cfg(not(target_arch = "wasm32"))] -pub type WsStream = backend_tungstenite::WsStream; +pub type Stream = backend_tungstenite::TungsteniteStream; #[cfg(not(target_arch = "wasm32"))] -pub type WebSocketBackend = backend_tungstenite::WebSocketBackend; +pub type WebSocketBackend = backend_tungstenite::TungsteniteBackend; // Gateway opcodes /// Opcode received when the server dispatches a [crate::types::WebSocketEvent]