peek-again

Crates.iopeek-again
lib.rspeek-again
version0.4.0
sourcesrc
created_at2024-11-09 02:27:55.802859
updated_at2024-11-23 21:24:09.699115
descriptionA performant iterator providing double peek functionality.
homepagehttps://github.com/CleveGreen/peek-again
repositoryhttps://github.com/CleveGreen/peek-again
max_upload_size
id1441774
size101,192
Cleve Green (CleveGreen)

documentation

https://docs.rs/peek_again/latest/peek_again

README

A performant iterator providing double peek functionality.

This crate provides a Peekable iterator adapter that allows looking ahead by up to two elements without advancing the iterator. It maintains performance parity with core::iter::Peekable for most single peek operations as well as next invocations.

Key Features

  • Double Peek: Look ahead by up to two elements while maintaining iterator position.
  • Performance: Most single peek operations are as fast as core::iter::Peekable.
  • Greater Control: Enhanced functionality for conditional iteration and element consumption.

Basic Usage

use peek_again::Peekable;

let mut iter = Peekable::new([1, 2, 3].into_iter());

assert_eq!(iter.peek(), Some(&1));   // Look at next element
assert_eq!(iter.peek_2(), Some(&2)); // Look two elements ahead
assert_eq!(iter.next(), Some(1));    // Iterator position unchanged

Enhanced Peek Control

The Peek type returned by peek provides additional control over iteration:

use peek_again::Peekable;
let mut iter = Peekable::new([1, 2, 3].into_iter());
let mut peek = iter.peek();

// Examine current and next elements
assert_eq!(peek, Some(&1));
assert_eq!(peek.peek(), Some(&2));

// Conditionally consume elements (moving the iterator forward)
if peek == Some(&1) {
    assert_eq!(peek.consume(), Some(1));
}

Conditional Draining

The drain_if method allows consuming multiple elements based on lookahead:

use peek_again::Peekable;
let mut iter = Peekable::new([1, 2, 3].into_iter());
let peek = iter.peek();

peek.drain_if(|&next| next == 2)
    .map_or_else(
        |_peek| unreachable!("The second element is two."),
        |(curr, next)| {
            assert_eq!(curr, Some(1));
            assert_eq!(next, 2);
        }
    );

Safety

This crate is marked with #![forbid(unsafe_code)] and employs design by contract principles, making the property tests far more meaningful. These contracts are also checked via kani.

This crate does have an allow-unsafe feature flag, this is not enabled by default, when enabled this crate will use core::hint::unreachable_unchecked for states that are guaranteed to be unreachable, offering a 15% performance improvement for certain operations.

Commit count: 10

cargo fmt