continuwuity/src/database/stream/items_rev.rs
Jason Volk 98e6c81e49 offload remaining db iterator initial seeks on cache miss
consume task budget on cache hit

Signed-off-by: Jason Volk <jason@zemos.net>
2025-01-01 23:28:01 -05:00

41 lines
922 B
Rust

use std::pin::Pin;
use conduwuit::Result;
use futures::{
stream::FusedStream,
task::{Context, Poll},
Stream,
};
use super::{keyval_longevity, Cursor, State};
use crate::keyval::KeyVal;
pub(crate) struct ItemsRev<'a> {
state: State<'a>,
}
impl<'a> From<State<'a>> for ItemsRev<'a> {
fn from(state: State<'a>) -> Self { Self { state } }
}
impl<'a> Cursor<'a, KeyVal<'a>> for ItemsRev<'a> {
fn state(&self) -> &State<'a> { &self.state }
fn fetch(&self) -> Option<KeyVal<'a>> { self.state.fetch().map(keyval_longevity) }
#[inline]
fn seek(&mut self) { self.state.seek_rev(); }
}
impl<'a> Stream for ItemsRev<'a> {
type Item = Result<KeyVal<'a>>;
fn poll_next(mut self: Pin<&mut Self>, _ctx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
Poll::Ready(self.seek_and_get())
}
}
impl FusedStream for ItemsRev<'_> {
#[inline]
fn is_terminated(&self) -> bool { !self.state.init && !self.state.valid() }
}