将通用函数作为参数传递

时间:2016-06-03 04:37:41

标签: rust

我希望能够将泛型函数传递给另一个函数(在本例中为闭包),而不会失去"泛型"传递的函数。由于这是一个非常复杂的陈述,以下是一个例子:

use std::fmt::Debug;

fn test<F, I: Debug>(gen: F) where F: Fn(fn(I) -> I) -> I {
    fn input<I: Debug>(x: I) -> I {
        x
    }

    println!("{:?}", gen(input));
}

fn main() {
    test(|input| {
        input(10);
        input(10.0)
    });
}

这不会编译,因为input的值是类型推断的,不再是通用的。

完整错误:

<anon>:14:15: 14:19 error: mismatched types:
 expected `_`,
    found `_`
(expected integral variable,
    found floating-point variable) [E0308]
<anon>:14         input(10.0)
                        ^~~~

生锈有可能吗?

编辑:

根据给出的解决方案,我使用以下方法解决了类似的问题:

#![feature(unboxed_closures)]
#![feature(fn_traits)]

use std::ops::Fn;
use std::ops::Add;
use std::ops::FnMut;

use std::fmt::Debug;

struct Builder;

impl Builder {
    pub fn build<A: Add<B>, B: Add<A>>(&self) -> fn(A, B) -> <A as std::ops::Add<B>>::Output {
        fn c<A: Add<B>, B: Add<A>>(a: A, b: B) -> <A as std::ops::Add<B>>::Output {
            a + b
        }

        return c;
    }
}

impl<A: Add<B>, B: Add<A>> Fn<(A, B)> for Builder {
    extern "rust-call" fn call(&self, args: (A, B)) -> <A as std::ops::Add<B>>::Output {
        let (a1, a2) = args;
        self.build()(a1, a2)
    }
}

impl<A: Add<B>, B: Add<A>> FnMut<(A, B)> for Builder {
    extern "rust-call" fn call_mut(&mut self, args: (A, B)) -> <A as std::ops::Add<B>>::Output {
        let (a1, a2) = args;
        self.build()(a1, a2)
    }
}

impl<A: Add<B>, B: Add<A>> FnOnce<(A, B)> for Builder {
    type Output = <A as std::ops::Add<B>>::Output;
    extern "rust-call" fn call_once(self, args: (A, B)) -> <A as std::ops::Add<B>>::Output {
        let (a1, a2) = args;
        self.build()(a1, a2)
    }
}

fn test<F, I: Debug>(gen: F) where F: Fn(Builder) -> I {
    let b = Builder;
    println!("{:?}", gen(b));
}

fn main() {
    test(|builder| {
        builder(10, 10);
        builder(10.1, 10.0)
    });
}

2 个答案:

答案 0 :(得分:7)

如前所述,遗憾的是,调用在调用站点被单态化,因此您无法传递泛型函数,只能传递泛型函数的单态版本。

可以传递的内容是构建器的函数:

use std::fmt::Debug;

struct Builder;

impl Builder {
    fn build<I: Debug>(&self) -> fn(I) -> I {
        fn input<I: Debug>(x: I) -> I { x }
        input
    }
}

fn test<F, T: Debug>(gen: F)
    where F: Fn(Builder) -> T
{
    let builder = Builder;
    println!("{:?}", gen(builder));
}

fn main() {
    test(|builder| {
        builder.build()(10);
        builder.build()(10.0)
    });
}

Builder能够根据需要生成input个实例。

答案 1 :(得分:4)

非常有趣的问题!我很确定这样的不是

Rust泛型通过单态化函数来工作。这意味着Rust编译器将为调用该函数的每个具体类型生成函数的机器代码。在函数的一次调用中,通用参数是固定的。因此,由于您在test中将main完全一次调用,因此该调用的通用参数是固定的。

这意味着闭包类型是固定的,闭包的input参数也具有具体类型。编译器为我们推断了所有类型,但是如果我们尝试注释这些类型,我们很快就会注意到我们遇到了与编译器相同的问题:

test::<_, usize>   // we can't ever spell out a closure type, therefore '_'
    (|input: fn(usize) -> usize|   // we can't have a generic closure right now
{
    input(10);   // works
    input(10.0)  // doesn't work
});

这看起来很像高级类型和通用闭包的用例。这两个功能在Rust还没有,AFAIK。

但是,您仍然可以通过动态调度实现您想要的效果:

fn test<F, I: Debug>(gen: F) where F: Fn(fn(Box<Debug>) -> Box<Debug>) -> I {
    fn input(x: Box<Debug>) -> Box<Debug> {
        x
    }

    println!("{:?}", gen(input));
}

fn main() {
    test(|input| {
        input(Box::new(10));
        input(Box::new(10.0))
    });
}

当然,这不如通用版本好,但至少它有效。另外:如果您实际上不需要input中的所有权,则可以将Box<Debug>更改为&Debug