zrx_stream/stream/function/traits/

select.rs

// Copyright (c) Zensical LLC <https://zensical.org>

// SPDX-License-Identifier: MIT
// Third-party contributions licensed under CLA

// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:

// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.

// ----------------------------------------------------------------------------

//! Select function.

use std::fmt::Display;
use zrx_scheduler::action::report::IntoReport;
use zrx_scheduler::action::Result;
use zrx_scheduler::Value;

use crate::stream::function::adapter::{WithId, WithSplat};
use crate::stream::function::{catch, Splat};

// ----------------------------------------------------------------------------
// Traits
// ----------------------------------------------------------------------------

/// Select function.
///
/// This trait defines a function that can be used to select data from an item,
/// e.g., to extract a key to group items or to compute a duration for a timer.
///
/// There's a range of different implementations of this trait, allowing you to
/// use a variety of function shapes, including support for [`Splat`], as well
/// as support for the [`WithId`] and [`WithSplat`] adapters. Furthermore, the
/// trait can be implemented for custom types to add new behaviors. Note that
/// all implementations also allow to return a [`Report`][], which makes it
/// possible to return diagnostics from the function execution.
///
/// The `'static` lifetimes is mandatory as closures must be moved into actions,
/// so requiring it here allows us to reduce the verbosity of trait bounds.
///
/// [`Report`]: zrx_scheduler::action::report::Report
///
/// # Examples
///
/// Group by odd/even:
///
/// ```
/// # use std::error::Error;
/// # fn main() -> Result<(), Box<dyn Error>> {
/// use zrx_stream::function::SelectFn;
///
/// // Define and execute function
/// let f = |&n: &i32| n & 1;
/// f.execute(&"id", &42)?;
/// # Ok(())
/// # }
/// ```
pub trait SelectFn<I, T, S>: Send + 'static
where
    T: ?Sized,
{
    /// Executes the select function.
    ///
    /// # Errors
    ///
    /// This method returns an error if the function fails to execute.
    fn execute(&self, id: &I, data: &T) -> Result<S>;
}

// ----------------------------------------------------------------------------
// Blanket implementations
// ----------------------------------------------------------------------------

impl<F, R, I, T, S> SelectFn<I, T, S> for F
where
    F: Fn(&T) -> R + Send + 'static,
    R: IntoReport<S>,
    I: Display,
    T: Value + ?Sized,
{
    #[cfg_attr(
        feature = "tracing",
        tracing::instrument(level = "debug", skip_all, fields(id = %id))
    )]
    #[inline]
    fn execute(&self, id: &I, data: &T) -> Result<S> {
        catch(|| self(data).into_report())
    }
}

impl<F, R, I, T, S> SelectFn<I, T, S> for WithId<F>
where
    F: Fn(&I, &T) -> R + Send + 'static,
    R: IntoReport<S>,
    I: Display,
    T: Value + ?Sized,
{
    #[cfg_attr(
        feature = "tracing",
        tracing::instrument(level = "debug", skip_all, fields(id = %id))
    )]
    #[inline]
    fn execute(&self, id: &I, data: &T) -> Result<S> {
        catch(|| self(id, data).into_report())
    }
}

impl<F, I, T, S> SelectFn<I, T, S> for WithSplat<F>
where
    F: SelectFn<I, Splat<T>, S>,
{
    #[inline]
    fn execute(&self, id: &I, data: &T) -> Result<S> {
        F::execute(self, id, Splat::from_ref(data))
    }
}

// ----------------------------------------------------------------------------
// Macros
// ----------------------------------------------------------------------------

/// Implements select function trait for splat arguments.
macro_rules! impl_select_fn_for_splat {
    ($($T:ident),+) => {
        impl<F, R, I, $($T,)+ S> SelectFn<I, Splat<($($T,)+)>, S> for F
        where
            F: Fn($(&$T),+) -> R + Send + 'static,
            R: IntoReport<S>,
            I: Display,
        {
            #[cfg_attr(
                feature = "tracing",
                tracing::instrument(level = "debug", skip_all, fields(id = %id))
            )]
            #[inline]
            fn execute(
                &self, id: &I, data: &Splat<($($T,)+)>
            ) -> Result<S> {
                #[allow(non_snake_case)]
                let ($($T,)+) = data.inner();
                catch(|| self($($T),+).into_report())
            }
        }
    };
}

// ----------------------------------------------------------------------------

impl_select_fn_for_splat!(T1);
impl_select_fn_for_splat!(T1, T2);
impl_select_fn_for_splat!(T1, T2, T3);
impl_select_fn_for_splat!(T1, T2, T3, T4);
impl_select_fn_for_splat!(T1, T2, T3, T4, T5);
impl_select_fn_for_splat!(T1, T2, T3, T4, T5, T6);
impl_select_fn_for_splat!(T1, T2, T3, T4, T5, T6, T7);
impl_select_fn_for_splat!(T1, T2, T3, T4, T5, T6, T7, T8);