rexport more tokio stuff

rama/src/lib.rs

@@ -10,3 +10,5 @@ pub mod state;
 pub mod stream;
 
 pub use tokio::main;
+
+pub use tokio::{select, spawn};

rama/src/server/tcp/listener.rs

@@ -215,7 +215,7 @@ impl<L> TcpListener<L> {
 
         loop {
             let guard = guard.clone();
-            tokio::select! {
+            crate::select! {
                 _ = guard.cancelled() => {
                     tracing::info!("signal received: initiate graceful shutdown");
                     break Ok(());

rama/src/service/spawn.rs

@@ -33,7 +33,7 @@ where
                 }
             });
         } else {
-            tokio::spawn(async move {
+            crate::spawn(async move {
                 if let Err(err) = service.call(request).await {
                     let err = err.into();
                     tracing::error!(error = err, "service error");

rama/src/stream/service/echo.rs

@@ -9,7 +9,7 @@ use crate::{service::Service, stream::Stream};
 /// ```rust
 /// use rama::{service::Service, stream::service::EchoService};
 ///
-/// # #[tokio::main]
+/// # #[rama::main]
 /// # async fn main() -> Result<(), Box<dyn std::error::Error>> {
 /// # let stream = tokio_test::io::Builder::new().read(b"hello world").write(b"hello world").build();
 /// let mut service = EchoService::new();

rama/src/stream/service/tracker/bytes.rs

@@ -19,8 +19,9 @@ use std::{
     task::{Context, Poll},
 };
 
+use crate::stream::{AsyncRead, AsyncWrite, ReadBuf};
+
 use pin_project_lite::pin_project;
-use tokio::io::{AsyncRead, AsyncWrite, ReadBuf};
 
 pin_project! {
     /// A wrapper around a [`AsyncRead`] and/or [`AsyncWrite`] that tracks the number
@@ -283,7 +284,7 @@ mod tests {
         let (check_tx, mut check_rx) = tokio::sync::broadcast::channel(1);
         let check_rx_2 = check_tx.subscribe();
 
-        let task_1 = tokio::spawn(async move {
+        let task_1 = crate::spawn(async move {
             let mut tracker = tracker;
             let mut buf = [0u8; 3];
 
@@ -315,7 +316,7 @@ mod tests {
         let task_2 = {
             let handle = handle.clone();
             let mut check_rx = check_rx_2;
-            tokio::spawn(async move {
+            crate::spawn(async move {
                 check_rx.recv().await.unwrap();
 
                 assert_eq!(handle.read(), 3);