mirror of
https://github.com/callumio/suck.git
synced 2026-03-21 14:08:11 +00:00
refactor: reorganize channel modules and implement async/sync structures
This commit is contained in:
parent
8366421ede
commit
a8d73eba4b
9 changed files with 46 additions and 30 deletions
|
|
@ -1,3 +1,4 @@
|
|||
pub mod channel;
|
||||
pub mod traits;
|
||||
|
||||
#[cfg(feature = "async-tokio")]
|
||||
|
|
|
|||
|
|
@ -9,10 +9,16 @@ use crate::asynchronous::traits::{
|
|||
};
|
||||
use crate::types;
|
||||
|
||||
type TokioSucker<T> =
|
||||
crate::AsyncSucker<T, TokioSender<types::Request>, TokioReceiver<types::Response<T>>>;
|
||||
type TokioSourcer<T> =
|
||||
crate::AsyncSourcer<T, TokioReceiver<types::Request>, TokioSender<types::Response<T>>>;
|
||||
type TokioSucker<T> = crate::asynchronous::channel::AsyncSucker<
|
||||
T,
|
||||
TokioSender<types::Request>,
|
||||
TokioReceiver<types::Response<T>>,
|
||||
>;
|
||||
type TokioSourcer<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 +76,9 @@ 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)
|
||||
}
|
||||
|
|
|
|||
10
src/lib.rs
10
src/lib.rs
|
|
@ -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};
|
||||
#[cfg(feature = "sync")]
|
||||
pub use channel::{Sourcer, Sucker};
|
||||
pub use asynchronous::channel::{AsyncSourcer, AsyncSucker};
|
||||
pub use error::Error;
|
||||
#[cfg(feature = "sync")]
|
||||
pub use sync::channel::{Sourcer, Sucker};
|
||||
|
|
|
|||
|
|
@ -6,10 +6,16 @@ use crate::types;
|
|||
use arc_swap::ArcSwap;
|
||||
use crossbeam_channel;
|
||||
|
||||
type CrossbeamSucker<T> =
|
||||
crate::Sucker<T, CrossbeamSender<types::Request>, CrossbeamReceiver<types::Response<T>>>;
|
||||
type CrossbeamSourcer<T> =
|
||||
crate::Sourcer<T, CrossbeamReceiver<types::Request>, CrossbeamSender<types::Response<T>>>;
|
||||
type CrossbeamSucker<T> = crate::sync::channel::Sucker<
|
||||
T,
|
||||
CrossbeamSender<types::Request>,
|
||||
CrossbeamReceiver<types::Response<T>>,
|
||||
>;
|
||||
type CrossbeamSourcer<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 +73,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)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -7,9 +7,12 @@ use arc_swap::ArcSwap;
|
|||
use flume;
|
||||
|
||||
type FlumeSucker<T> =
|
||||
crate::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::Sucker<T, FlumeSender<types::Request>, FlumeReceiver<types::Response<T>>>;
|
||||
type FlumeSourcer<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 +71,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)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,3 +1,4 @@
|
|||
pub mod channel;
|
||||
pub mod traits;
|
||||
|
||||
#[cfg(feature = "sync-crossbeam")]
|
||||
|
|
|
|||
|
|
@ -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();
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue