summaryrefslogtreecommitdiffstats
path: root/tests/ui/unboxed-closures/unboxed-closures-infer-recursive-fn.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-17 12:19:03 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-17 12:19:03 +0000
commit64d98f8ee037282c35007b64c2649055c56af1db (patch)
tree5492bcf97fce41ee1c0b1cc2add283f3e66cdab0 /tests/ui/unboxed-closures/unboxed-closures-infer-recursive-fn.rs
parentAdding debian version 1.67.1+dfsg1-1. (diff)
downloadrustc-64d98f8ee037282c35007b64c2649055c56af1db.tar.xz
rustc-64d98f8ee037282c35007b64c2649055c56af1db.zip
Merging upstream version 1.68.2+dfsg1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/ui/unboxed-closures/unboxed-closures-infer-recursive-fn.rs')
-rw-r--r--tests/ui/unboxed-closures/unboxed-closures-infer-recursive-fn.rs45
1 files changed, 45 insertions, 0 deletions
diff --git a/tests/ui/unboxed-closures/unboxed-closures-infer-recursive-fn.rs b/tests/ui/unboxed-closures/unboxed-closures-infer-recursive-fn.rs
new file mode 100644
index 000000000..a0fbbafe2
--- /dev/null
+++ b/tests/ui/unboxed-closures/unboxed-closures-infer-recursive-fn.rs
@@ -0,0 +1,45 @@
+// run-pass
+#![feature(fn_traits, unboxed_closures)]
+
+use std::marker::PhantomData;
+
+// Test that we are able to infer a suitable kind for a "recursive"
+// closure. As far as I can tell, coding up a recursive closure
+// requires the good ol' [Y Combinator].
+//
+// [Y Combinator]: https://en.wikipedia.org/wiki/Fixed-point_combinator#Y_combinator
+
+struct YCombinator<F,A,R> {
+ func: F,
+ marker: PhantomData<(A,R)>,
+}
+
+impl<F,A,R> YCombinator<F,A,R> {
+ fn new(f: F) -> YCombinator<F,A,R> {
+ YCombinator { func: f, marker: PhantomData }
+ }
+}
+
+impl<A,R,F : Fn(&dyn Fn(A) -> R, A) -> R> Fn<(A,)> for YCombinator<F,A,R> {
+ extern "rust-call" fn call(&self, (arg,): (A,)) -> R {
+ (self.func)(self, arg)
+ }
+}
+
+impl<A,R,F : Fn(&dyn Fn(A) -> R, A) -> R> FnMut<(A,)> for YCombinator<F,A,R> {
+ extern "rust-call" fn call_mut(&mut self, args: (A,)) -> R { self.call(args) }
+}
+
+impl<A,R,F : Fn(&dyn Fn(A) -> R, A) -> R> FnOnce<(A,)> for YCombinator<F,A,R> {
+ type Output = R;
+ extern "rust-call" fn call_once(self, args: (A,)) -> R { self.call(args) }
+}
+
+fn main() {
+ let factorial = |recur: &dyn Fn(u32) -> u32, arg: u32| -> u32 {
+ if arg == 0 {1} else {arg * recur(arg-1)}
+ };
+ let factorial: YCombinator<_,u32,u32> = YCombinator::new(factorial);
+ let r = factorial(10);
+ assert_eq!(3628800, r);
+}