Songbird: Tokio 1.0 (#36)

Migrates to the new version of tokio, requiring channel and sleep changes in a few locations. Additionally points to the in-tree v0.3 version of twilight.

src/driver/connection/mod.rs

@@ -18,7 +18,7 @@ use crate::{
 use discortp::discord::{IpDiscoveryPacket, IpDiscoveryType, MutableIpDiscoveryPacket};
 use error::{Error, Result};
 use flume::Sender;
-use std::{net::IpAddr, str::FromStr};
+use std::{net::IpAddr, str::FromStr, sync::Arc};
 use tokio::net::UdpSocket;
 use tracing::{debug, info, instrument};
 use url::Url;
@@ -97,7 +97,7 @@ impl Connection {
             return Err(Error::CryptoModeUnavailable);
         }
 
-        let mut udp = UdpSocket::bind("0.0.0.0:0").await?;
+        let udp = UdpSocket::bind("0.0.0.0:0").await?;
         udp.connect((ready.ip, ready.port)).await?;
 
         // Follow Discord's IP Discovery procedures, in case NAT tunnelling is needed.
@@ -161,7 +161,9 @@ impl Connection {
         let (ws_msg_tx, ws_msg_rx) = flume::unbounded();
         let (udp_sender_msg_tx, udp_sender_msg_rx) = flume::unbounded();
         let (udp_receiver_msg_tx, udp_receiver_msg_rx) = flume::unbounded();
-        let (udp_rx, udp_tx) = udp.split();
+
+        let udp_rx = Arc::new(udp);
+        let udp_tx = Arc::clone(&udp_rx);
 
         let ssrc = ready.ssrc;
 

src/driver/tasks/udp_rx.rs

@@ -20,8 +20,8 @@ use discortp::{
     PacketSize,
 };
 use flume::Receiver;
-use std::collections::HashMap;
-use tokio::net::udp::RecvHalf;
+use std::{collections::HashMap, sync::Arc};
+use tokio::net::UdpSocket;
 use tracing::{error, info, instrument, warn};
 use xsalsa20poly1305::XSalsa20Poly1305 as Cipher;
 
@@ -236,7 +236,7 @@ struct UdpRx {
     config: Config,
     packet_buffer: [u8; VOICE_PACKET_MAX],
     rx: Receiver<UdpRxMessage>,
-    udp_socket: RecvHalf,
+    udp_socket: Arc<UdpSocket>,
 }
 
 impl UdpRx {
@@ -391,7 +391,7 @@ pub(crate) async fn runner(
     rx: Receiver<UdpRxMessage>,
     cipher: Cipher,
     config: Config,
-    udp_socket: RecvHalf,
+    udp_socket: Arc<UdpSocket>,
 ) {
     info!("UDP receive handle started.");
 

src/driver/tasks/udp_tx.rs

@@ -2,14 +2,15 @@ use super::message::*;
 use crate::constants::*;
 use discortp::discord::MutableKeepalivePacket;
 use flume::Receiver;
+use std::sync::Arc;
 use tokio::{
-    net::udp::SendHalf,
-    time::{timeout_at, Elapsed, Instant},
+    net::UdpSocket,
+    time::{timeout_at, Instant},
 };
 use tracing::{error, info, instrument, trace};
 
 #[instrument(skip(udp_msg_rx))]
-pub(crate) async fn runner(udp_msg_rx: Receiver<UdpTxMessage>, ssrc: u32, mut udp_tx: SendHalf) {
+pub(crate) async fn runner(udp_msg_rx: Receiver<UdpTxMessage>, ssrc: u32, udp_tx: Arc<UdpSocket>) {
     info!("UDP transmit handle started.");
 
     let mut keepalive_bytes = [0u8; MutableKeepalivePacket::minimum_packet_size()];
@@ -22,7 +23,7 @@ pub(crate) async fn runner(udp_msg_rx: Receiver<UdpTxMessage>, ssrc: u32, mut ud
     loop {
         use UdpTxMessage::*;
         match timeout_at(ka_time, udp_msg_rx.recv_async()).await {
-            Err(Elapsed { .. }) => {
+            Err(_) => {
                 trace!("Sending UDP Keepalive.");
                 if let Err(e) = udp_tx.send(&keepalive_bytes[..]).await {
                     error!("Fatal UDP keepalive send error: {:?}.", e);

src/driver/tasks/ws.rs

@@ -57,7 +57,7 @@ impl AuxNetwork {
             let mut ws_error = false;
             let mut should_reconnect = false;
 
-            let hb = time::delay_until(next_heartbeat);
+            let hb = time::sleep_until(next_heartbeat);
 
             tokio::select! {
                 _ = hb => {

src/tracks/command.rs

@@ -1,7 +1,7 @@
 use super::*;
 use crate::events::EventData;
+use flume::Sender;
 use std::time::Duration;
-use tokio::sync::oneshot::Sender as OneshotSender;
 
 /// A request from external code using a [`TrackHandle`] to modify
 /// or act upon an [`Track`] object.
@@ -27,7 +27,7 @@ pub enum TrackCommand {
     /// Run some closure on this track, with direct access to the core object.
     Do(Box<dyn FnOnce(&mut Track) + Send + Sync + 'static>),
     /// Request a read-only view of this track's state.
-    Request(OneshotSender<Box<TrackState>>),
+    Request(Sender<Box<TrackState>>),
     /// Change the loop count/strategy of this track.
     Loop(LoopState),
     /// Prompts a track's input to become live and usable, if it is not already.

src/tracks/handle.rs

@@ -3,8 +3,9 @@ use crate::{
     events::{Event, EventData, EventHandler},
     input::Metadata,
 };
+use flume::Sender;
 use std::{fmt, sync::Arc, time::Duration};
-use tokio::sync::{mpsc::UnboundedSender, oneshot, RwLock};
+use tokio::sync::RwLock;
 use typemap_rev::TypeMap;
 use uuid::Uuid;
 
@@ -25,7 +26,7 @@ pub struct TrackHandle {
 }
 
 struct InnerHandle {
-    command_channel: UnboundedSender<TrackCommand>,
+    command_channel: Sender<TrackCommand>,
     seekable: bool,
     uuid: Uuid,
     metadata: Box<Metadata>,
@@ -50,7 +51,7 @@ impl TrackHandle {
     ///
     /// [`Input`]: crate::input::Input
     pub fn new(
-        command_channel: UnboundedSender<TrackCommand>,
+        command_channel: Sender<TrackCommand>,
         seekable: bool,
         uuid: Uuid,
         metadata: Box<Metadata>,
@@ -159,10 +160,10 @@ impl TrackHandle {
 
     /// Request playback information and state from the audio context.
     pub async fn get_info(&self) -> TrackResult<Box<TrackState>> {
-        let (tx, rx) = oneshot::channel();
+        let (tx, rx) = flume::bounded(1);
         self.send(TrackCommand::Request(tx))?;
 
-        rx.await.map_err(|_| TrackError::Finished)
+        rx.recv_async().await.map_err(|_| TrackError::Finished)
     }
 
     /// Set an audio track to loop indefinitely.

src/tracks/mod.rs

@@ -25,8 +25,8 @@ mod state;
 pub use self::{command::*, error::*, handle::*, looping::*, mode::*, queue::*, state::*};
 
 use crate::{constants::*, driver::tasks::message::*, events::EventStore, input::Input};
+use flume::{Receiver, TryRecvError};
 use std::time::Duration;
-use tokio::sync::mpsc::{self, error::TryRecvError, UnboundedReceiver};
 use uuid::Uuid;
 
 /// Control object for audio playback.
@@ -102,7 +102,7 @@ pub struct Track {
     /// Track commands are sent in this manner to ensure that access
     /// occurs in a thread-safe manner, without allowing any external
     /// code to lock access to audio objects and block packet generation.
-    pub(crate) commands: UnboundedReceiver<TrackCommand>,
+    pub(crate) commands: Receiver<TrackCommand>,
 
     /// Handle for safe control of this audio track from other threads.
     ///
@@ -124,11 +124,7 @@ impl Track {
     /// In general, you should probably use [`create_player`].
     ///
     /// [`create_player`]: fn.create_player.html
-    pub fn new_raw(
-        source: Input,
-        commands: UnboundedReceiver<TrackCommand>,
-        handle: TrackHandle,
-    ) -> Self {
+    pub fn new_raw(source: Input, commands: Receiver<TrackCommand>, handle: TrackHandle) -> Self {
         let uuid = handle.uuid();
 
         Self {
@@ -310,7 +306,7 @@ impl Track {
                         MakePlayable => self.make_playable(),
                     }
                 },
-                Err(TryRecvError::Closed) => {
+                Err(TryRecvError::Disconnected) => {
                     // this branch will never be visited.
                     break;
                 },
@@ -389,7 +385,7 @@ pub fn create_player(source: Input) -> (Track, TrackHandle) {
 /// [`Track`]: Track
 /// [`TrackHandle`]: TrackHandle
 pub fn create_player_with_uuid(source: Input, uuid: Uuid) -> (Track, TrackHandle) {
-    let (tx, rx) = mpsc::unbounded_channel();
+    let (tx, rx) = flume::unbounded();
     let can_seek = source.is_seekable();
     let metadata = source.metadata.clone();
     let handle = TrackHandle::new(tx, can_seek, uuid, metadata);