refactor: reorganize channel modules and implement async/sync structures

This commit is contained in:
Callum Leslie 2026-03-04 21:01:46 +00:00
parent 8366421ede
commit f62d8eebca
9 changed files with 40 additions and 24 deletions

View file

@ -1,4 +1,5 @@
pub mod traits;
pub mod channel;
#[cfg(feature = "async-tokio")]
pub mod tokio;

View file

@ -10,9 +10,17 @@ use crate::asynchronous::traits::{
use crate::types;
type TokioSucker<T> =
crate::AsyncSucker<T, TokioSender<types::Request>, TokioReceiver<types::Response<T>>>;
crate::asynchronous::channel::AsyncSucker<
T,
TokioSender<types::Request>,
TokioReceiver<types::Response<T>>,
>;
type TokioSourcer<T> =
crate::AsyncSourcer<T, TokioReceiver<types::Request>, TokioSender<types::Response<T>>>;
crate::asynchronous::channel::AsyncSourcer<
T,
TokioReceiver<types::Request>,
TokioSender<types::Response<T>>,
>;
pub struct TokioSender<T>(mpsc::UnboundedSender<T>);
pub struct TokioReceiver<T>(Mutex<mpsc::UnboundedReceiver<T>>);
@ -70,8 +78,8 @@ impl<T> TokioSuck<T> {
let (response_tx, response_rx) = TokioChannel::create_response_channel::<T>();
let state = ArcSwap::new(Arc::new(crate::types::ValueSource::None));
let sucker = crate::AsyncSucker::new(request_tx, response_rx);
let sourcer = crate::AsyncSourcer::new(request_rx, response_tx, state);
let sucker = crate::asynchronous::channel::AsyncSucker::new(request_tx, response_rx);
let sourcer = crate::asynchronous::channel::AsyncSourcer::new(request_rx, response_tx, state);
(sucker, sourcer)
}

View file

@ -1,10 +1,6 @@
#![doc = include_str!("../README.md")]
#![cfg_attr(docsrs, feature(doc_auto_cfg))]
#[cfg(feature = "async")]
pub mod async_channel;
#[cfg(feature = "sync")]
pub mod channel;
pub mod error;
#[cfg(feature = "async")]
@ -15,7 +11,7 @@ pub mod sync;
pub mod types;
#[cfg(feature = "async")]
pub use async_channel::{AsyncSourcer, AsyncSucker};
pub use asynchronous::channel::{AsyncSourcer, AsyncSucker};
#[cfg(feature = "sync")]
pub use channel::{Sourcer, Sucker};
pub use sync::channel::{Sourcer, Sucker};
pub use error::Error;

View file

@ -7,9 +7,17 @@ use arc_swap::ArcSwap;
use crossbeam_channel;
type CrossbeamSucker<T> =
crate::Sucker<T, CrossbeamSender<types::Request>, CrossbeamReceiver<types::Response<T>>>;
crate::sync::channel::Sucker<
T,
CrossbeamSender<types::Request>,
CrossbeamReceiver<types::Response<T>>,
>;
type CrossbeamSourcer<T> =
crate::Sourcer<T, CrossbeamReceiver<types::Request>, CrossbeamSender<types::Response<T>>>;
crate::sync::channel::Sourcer<
T,
CrossbeamReceiver<types::Request>,
CrossbeamSender<types::Response<T>>,
>;
/// Internal sender type for crossbeam backend
pub struct CrossbeamSender<T>(crossbeam_channel::Sender<T>);
@ -67,8 +75,8 @@ impl<T> CrossbeamSuck<T> {
let state = ArcSwap::new(Arc::new(crate::types::ValueSource::None));
let sucker = crate::Sucker::new(request_tx, response_rx);
let sourcer = crate::Sourcer::new(request_rx, response_tx, state);
let sucker = crate::sync::channel::Sucker::new(request_tx, response_rx);
let sourcer = crate::sync::channel::Sourcer::new(request_rx, response_tx, state);
(sucker, sourcer)
}

View file

@ -7,9 +7,9 @@ use arc_swap::ArcSwap;
use flume;
type FlumeSucker<T> =
crate::Sucker<T, FlumeSender<types::Request>, FlumeReceiver<types::Response<T>>>;
crate::sync::channel::Sucker<T, FlumeSender<types::Request>, FlumeReceiver<types::Response<T>>>;
type FlumeSourcer<T> =
crate::Sourcer<T, FlumeReceiver<types::Request>, FlumeSender<types::Response<T>>>;
crate::sync::channel::Sourcer<T, FlumeReceiver<types::Request>, FlumeSender<types::Response<T>>>;
/// Internal sender type for flume backend
pub struct FlumeSender<T>(flume::Sender<T>);
@ -68,8 +68,8 @@ impl<T> FlumeSuck<T> {
let state = Arc::new(crate::types::ValueSource::None);
let state = ArcSwap::new(state);
let sucker = crate::Sucker::new(request_tx, response_rx);
let sourcer = crate::Sourcer::new(request_rx, response_tx, state);
let sucker = crate::sync::channel::Sucker::new(request_tx, response_rx);
let sourcer = crate::sync::channel::Sourcer::new(request_rx, response_tx, state);
(sucker, sourcer)
}

View file

@ -1,4 +1,5 @@
pub mod traits;
pub mod channel;
#[cfg(feature = "sync-crossbeam")]
pub mod crossbeam;

View file

@ -6,8 +6,10 @@ use std::sync::Arc;
#[cfg(feature = "sync-std")]
use std::sync::mpsc;
type StdSucker<T> = crate::Sucker<T, StdSender<types::Request>, StdReceiver<types::Response<T>>>;
type StdSourcer<T> = crate::Sourcer<T, StdReceiver<types::Request>, StdSender<types::Response<T>>>;
type StdSucker<T> =
crate::sync::channel::Sucker<T, StdSender<types::Request>, StdReceiver<types::Response<T>>>;
type StdSourcer<T> =
crate::sync::channel::Sourcer<T, StdReceiver<types::Request>, StdSender<types::Response<T>>>;
/// Internal sender type for std backend
pub struct StdSender<T>(mpsc::Sender<T>);
@ -66,8 +68,8 @@ impl<T> StdSuck<T> {
let state = Arc::new(crate::types::ValueSource::None);
let state = ArcSwap::new(state);
let sucker = crate::Sucker::new(request_tx, response_rx);
let sourcer = crate::Sourcer::new(request_rx, response_tx, state);
let sucker = crate::sync::channel::Sucker::new(request_tx, response_rx);
let sourcer = crate::sync::channel::Sourcer::new(request_rx, response_tx, state);
(sucker, sourcer)
}
@ -236,7 +238,7 @@ mod tests {
let state = Arc::new(crate::types::ValueSource::None);
let state = ArcSwap::new(state);
let sourcer = crate::Sourcer::new(request_rx, response_tx, state);
let sourcer = crate::sync::channel::Sourcer::new(request_rx, response_tx, state);
sourcer.set_static(42).unwrap();
let producer_handle = thread::spawn(move || sourcer.run().unwrap());
@ -254,7 +256,7 @@ mod tests {
let state = Arc::new(crate::types::ValueSource::None);
let state = ArcSwap::new(state);
let sourcer = crate::Sourcer::new(request_rx, response_tx, state);
let sourcer = crate::sync::channel::Sourcer::new(request_rx, response_tx, state);
sourcer.run().unwrap();
}