Simplify call_fn API, no need to pass &mut references.
This commit is contained in:
39
src/api.rs
39
src/api.rs
@@ -2,7 +2,7 @@
|
||||
|
||||
use crate::any::{Any, AnyExt, Dynamic};
|
||||
use crate::call::FuncArgs;
|
||||
use crate::engine::{Engine, FnAny, FnIntExt, FnSpec};
|
||||
use crate::engine::{Engine, FnAny, FnCallArgs, FnIntExt, FnSpec};
|
||||
use crate::error::ParseError;
|
||||
use crate::fn_register::RegisterFn;
|
||||
use crate::parser::{lex, parse, Position, AST};
|
||||
@@ -273,17 +273,32 @@ impl<'e> Engine<'e> {
|
||||
///
|
||||
/// let ast = engine.compile("fn add(x, y) { x.len() + y }")?;
|
||||
///
|
||||
/// let result: i64 = engine.call_fn("add", &ast, (&mut String::from("abc"), &mut 123_i64))?;
|
||||
/// let result: i64 = engine.call_fn("add", &ast, (String::from("abc"), 123_i64))?;
|
||||
///
|
||||
/// assert_eq!(result, 126);
|
||||
/// # Ok(())
|
||||
/// # }
|
||||
/// ```
|
||||
pub fn call_fn<'f, A: FuncArgs<'f>, T: Any + Clone>(
|
||||
pub fn call_fn<A: FuncArgs, T: Any + Clone>(
|
||||
&mut self,
|
||||
name: &str,
|
||||
ast: &AST,
|
||||
args: A,
|
||||
) -> Result<T, EvalAltResult> {
|
||||
let mut arg_values = args.into_vec();
|
||||
|
||||
self.call_fn_internal(
|
||||
name,
|
||||
ast,
|
||||
arg_values.iter_mut().map(|v| v.as_mut()).collect(),
|
||||
)
|
||||
}
|
||||
|
||||
pub(crate) fn call_fn_internal<T: Any + Clone>(
|
||||
&mut self,
|
||||
name: &str,
|
||||
ast: &AST,
|
||||
args: FnCallArgs,
|
||||
) -> Result<T, EvalAltResult> {
|
||||
let pos = Default::default();
|
||||
|
||||
@@ -297,16 +312,14 @@ impl<'e> Engine<'e> {
|
||||
);
|
||||
});
|
||||
|
||||
let result = self
|
||||
.call_fn_raw(name, args.into_vec(), None, pos)
|
||||
.and_then(|b| {
|
||||
b.downcast().map(|b| *b).map_err(|a| {
|
||||
EvalAltResult::ErrorMismatchOutputType(
|
||||
self.map_type_name((*a).type_name()).into(),
|
||||
pos,
|
||||
)
|
||||
})
|
||||
});
|
||||
let result = self.call_fn_raw(name, args, None, pos).and_then(|b| {
|
||||
b.downcast().map(|b| *b).map_err(|a| {
|
||||
EvalAltResult::ErrorMismatchOutputType(
|
||||
self.map_type_name((*a).type_name()).into(),
|
||||
pos,
|
||||
)
|
||||
})
|
||||
});
|
||||
|
||||
self.clear_functions();
|
||||
|
||||
|
||||
28
src/call.rs
28
src/call.rs
@@ -1,35 +1,29 @@
|
||||
//! Helper module which defines `FnArgs` to make function calling easier.
|
||||
|
||||
use crate::any::{Any, Variant};
|
||||
use crate::any::{Any, Dynamic};
|
||||
|
||||
/// Trait that represent arguments to a function call.
|
||||
pub trait FuncArgs<'a> {
|
||||
/// Convert to a `Vec` of `Variant` arguments.
|
||||
fn into_vec(self) -> Vec<&'a mut Variant>;
|
||||
pub trait FuncArgs {
|
||||
/// Convert to a `Vec` of `Dynamic` arguments.
|
||||
fn into_vec(self) -> Vec<Dynamic>;
|
||||
}
|
||||
|
||||
impl<'a> FuncArgs<'a> for Vec<&'a mut Variant> {
|
||||
fn into_vec(self) -> Self {
|
||||
self
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a, T: Any> FuncArgs<'a> for &'a mut Vec<T> {
|
||||
fn into_vec(self) -> Vec<&'a mut Variant> {
|
||||
self.iter_mut().map(|x| x as &mut Variant).collect()
|
||||
impl<T: Any> FuncArgs for &mut Vec<T> {
|
||||
fn into_vec(self) -> Vec<Dynamic> {
|
||||
self.iter_mut().map(|x| x.into_dynamic()).collect()
|
||||
}
|
||||
}
|
||||
|
||||
macro_rules! impl_args {
|
||||
($($p:ident),*) => {
|
||||
impl<'a, $($p: Any + Clone),*> FuncArgs<'a> for ($(&'a mut $p,)*)
|
||||
impl<$($p: Any + Clone),*> FuncArgs for ($($p,)*)
|
||||
{
|
||||
fn into_vec(self) -> Vec<&'a mut Variant> {
|
||||
fn into_vec(self) -> Vec<Dynamic> {
|
||||
let ($($p,)*) = self;
|
||||
|
||||
#[allow(unused_variables, unused_mut)]
|
||||
let mut v = Vec::new();
|
||||
$(v.push($p as &mut Variant);)*
|
||||
$(v.push($p.into_dynamic());)*
|
||||
|
||||
v
|
||||
}
|
||||
@@ -45,4 +39,4 @@ macro_rules! impl_args {
|
||||
}
|
||||
|
||||
#[cfg_attr(rustfmt, rustfmt_skip)]
|
||||
impl_args!(A, B, C, D, E, F, G, H, I, J, K, L, M, N, O, P, Q, R, S);
|
||||
impl_args!(A, B, C, D, E, F, G, H, I, J, K, L, M, N, O, P, Q, R, S, T);
|
||||
|
||||
Reference in New Issue
Block a user