zacheryasc
f4f26c9f41
Improve the runtime, increase test coverage, add new actor types for monitoring and supervision to an `std` crate.
63 lines
1.2 KiB
Rust
63 lines
1.2 KiB
Rust
use std::sync::Arc;
|
|
|
|
use crossbeam_queue::{ArrayQueue, SegQueue};
|
|
|
|
pub struct HybridChannel<T> {
|
|
ring: ArrayQueue<T>,
|
|
overflow: SegQueue<T>,
|
|
}
|
|
|
|
impl<T> HybridChannel<T> {
|
|
pub fn new(capacity: usize) -> Self {
|
|
Self {
|
|
ring: ArrayQueue::new(capacity),
|
|
overflow: SegQueue::new(),
|
|
}
|
|
}
|
|
|
|
pub fn push(&self, value: T) {
|
|
if !self.overflow.is_empty() {
|
|
self.overflow.push(value);
|
|
return;
|
|
}
|
|
if let Err(v) = self.ring.push(value) {
|
|
self.overflow.push(v);
|
|
}
|
|
}
|
|
|
|
pub fn pop(&self) -> Option<T> {
|
|
self.ring.pop().or_else(|| self.overflow.pop())
|
|
}
|
|
}
|
|
|
|
pub(crate) struct Receiver<T> {
|
|
queue: Arc<HybridChannel<T>>,
|
|
}
|
|
|
|
impl<T> Receiver<T> {
|
|
pub fn new(capacity: usize) -> Self {
|
|
let queue = Arc::new(HybridChannel::new(capacity));
|
|
|
|
Self { queue }
|
|
}
|
|
pub fn try_recv(&self) -> Option<T> {
|
|
self.queue.pop()
|
|
}
|
|
|
|
pub fn new_sender(&self) -> Sender<T> {
|
|
Sender {
|
|
queue: self.queue.clone(),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub(crate) struct Sender<T> {
|
|
queue: Arc<HybridChannel<T>>,
|
|
}
|
|
|
|
impl<T> Sender<T> {
|
|
pub fn send(&self, value: T) {
|
|
self.queue.push(value)
|
|
}
|
|
}
|
|
|