Skip to content

Commit

Permalink
Only consider impliciy unboxed closure impl if the obligation is
Browse files Browse the repository at this point in the history
actually for `Fn`, `FnMut`, or `FnOnce`.

Fixes #18019
  • Loading branch information
nikomatsakis committed Oct 13, 2014
1 parent 4a382d7 commit 590a61f
Show file tree
Hide file tree
Showing 2 changed files with 52 additions and 29 deletions.
49 changes: 20 additions & 29 deletions src/librustc/middle/traits/select.rs
Original file line number Diff line number Diff line change
Expand Up @@ -608,6 +608,17 @@ impl<'cx, 'tcx> SelectionContext<'cx, 'tcx> {
* unified during the confirmation step.
*/

let tcx = self.tcx();
let kind = if Some(obligation.trait_ref.def_id) == tcx.lang_items.fn_trait() {
ty::FnUnboxedClosureKind
} else if Some(obligation.trait_ref.def_id) == tcx.lang_items.fn_mut_trait() {
ty::FnMutUnboxedClosureKind
} else if Some(obligation.trait_ref.def_id) == tcx.lang_items.fn_once_trait() {
ty::FnOnceUnboxedClosureKind
} else {
return Ok(()); // not a fn trait, ignore
};

let self_ty = self.infcx.shallow_resolve(obligation.self_ty());
let closure_def_id = match ty::get(self_ty).sty {
ty::ty_unboxed_closure(id, _) => id,
Expand All @@ -622,37 +633,17 @@ impl<'cx, 'tcx> SelectionContext<'cx, 'tcx> {
self_ty.repr(self.tcx()),
obligation.repr(self.tcx()));

let tcx = self.tcx();
let fn_traits = [
(ty::FnUnboxedClosureKind, tcx.lang_items.fn_trait()),
(ty::FnMutUnboxedClosureKind, tcx.lang_items.fn_mut_trait()),
(ty::FnOnceUnboxedClosureKind, tcx.lang_items.fn_once_trait()),
];
for tuple in fn_traits.iter() {
let kind = match tuple {
&(kind, Some(ref fn_trait))
if *fn_trait == obligation.trait_ref.def_id =>
{
kind
}
_ => continue,
};

// Check to see whether the argument and return types match.
let closure_kind = match self.typer.unboxed_closures().borrow().find(&closure_def_id) {
Some(closure) => closure.kind,
None => {
self.tcx().sess.span_bug(
obligation.cause.span,
format!("No entry for unboxed closure: {}",
closure_def_id.repr(self.tcx())).as_slice());
}
};

if closure_kind != kind {
continue;
let closure_kind = match self.typer.unboxed_closures().borrow().find(&closure_def_id) {
Some(closure) => closure.kind,
None => {
self.tcx().sess.span_bug(
obligation.cause.span,
format!("No entry for unboxed closure: {}",
closure_def_id.repr(self.tcx())).as_slice());
}
};

if closure_kind == kind {
candidates.vec.push(UnboxedClosureCandidate(closure_def_id));
}

Expand Down
32 changes: 32 additions & 0 deletions src/test/run-pass/multidispatch-infer-from-single-impl.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

// Test that we correctly infer that `E` must be `()` here. This is
// known because there is just one impl that could apply where
// `Self=()`.

pub trait FromError<E> {
fn from_error(err: E) -> Self;
}

impl<E> FromError<E> for E {
fn from_error(err: E) -> E {
err
}
}

fn test() -> Result<(), ()> {
Err(FromError::from_error(()))
}

fn main() {
let result = (|| Err(FromError::from_error(())))();
let foo: () = result.unwrap_or(());
}

4 comments on commit 590a61f

@bors
Copy link
Contributor

@bors bors commented on 590a61f Oct 15, 2014

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

saw approval from pcwalton
at nikomatsakis@590a61f

@bors
Copy link
Contributor

@bors bors commented on 590a61f Oct 15, 2014

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

merging nikomatsakis/rust/issue-18019 = 590a61f into auto

@bors
Copy link
Contributor

@bors bors commented on 590a61f Oct 15, 2014

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nikomatsakis/rust/issue-18019 = 590a61f merged ok, testing candidate = 88f35be5

@bors
Copy link
Contributor

@bors bors commented on 590a61f Oct 15, 2014

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please sign in to comment.