Remap impl-trait lifetimes on HIR instead of AST lowering.
This commit is contained in:
parent
297f28c308
commit
d9a0fec48a
@ -420,15 +420,6 @@ fn visit_poly_trait_ref(&mut self, poly_tref: &'tcx PolyTraitRef<'tcx>) {
|
|||||||
|
|
||||||
fn visit_ty(&mut self, ty: &'tcx Ty<'_>) {
|
fn visit_ty(&mut self, ty: &'tcx Ty<'_>) {
|
||||||
match ty.kind {
|
match ty.kind {
|
||||||
TyKind::OpaqueDef(opaque, bounds) => {
|
|
||||||
let len = self.lts.len();
|
|
||||||
self.visit_opaque_ty(opaque);
|
|
||||||
self.lts.truncate(len);
|
|
||||||
self.lts.extend(bounds.iter().filter_map(|bound| match bound {
|
|
||||||
GenericArg::Lifetime(&l) => Some(l),
|
|
||||||
_ => None,
|
|
||||||
}));
|
|
||||||
},
|
|
||||||
TyKind::BareFn(&BareFnTy { decl, .. }) => {
|
TyKind::BareFn(&BareFnTy { decl, .. }) => {
|
||||||
let mut sub_visitor = RefVisitor::new(self.cx);
|
let mut sub_visitor = RefVisitor::new(self.cx);
|
||||||
sub_visitor.visit_fn_decl(decl);
|
sub_visitor.visit_fn_decl(decl);
|
||||||
|
@ -4,9 +4,11 @@
|
|||||||
use rustc_hir::intravisit::FnKind;
|
use rustc_hir::intravisit::FnKind;
|
||||||
use rustc_hir::{
|
use rustc_hir::{
|
||||||
Block, Body, Closure, ClosureKind, CoroutineDesugaring, CoroutineKind, CoroutineSource, Expr, ExprKind, FnDecl,
|
Block, Body, Closure, ClosureKind, CoroutineDesugaring, CoroutineKind, CoroutineSource, Expr, ExprKind, FnDecl,
|
||||||
FnRetTy, GenericArg, GenericBound, ImplItem, Item, LifetimeName, Node, TraitRef, Ty, TyKind,
|
FnRetTy, GenericBound, ImplItem, Item, Node, OpaqueTy, TraitRef, Ty, TyKind,
|
||||||
};
|
};
|
||||||
use rustc_lint::{LateContext, LateLintPass};
|
use rustc_lint::{LateContext, LateLintPass};
|
||||||
|
use rustc_middle::middle::resolve_bound_vars::ResolvedArg;
|
||||||
|
use rustc_middle::ty;
|
||||||
use rustc_session::declare_lint_pass;
|
use rustc_session::declare_lint_pass;
|
||||||
use rustc_span::def_id::LocalDefId;
|
use rustc_span::def_id::LocalDefId;
|
||||||
use rustc_span::{Span, sym};
|
use rustc_span::{Span, sym};
|
||||||
@ -44,21 +46,22 @@ fn check_fn(
|
|||||||
decl: &'tcx FnDecl<'_>,
|
decl: &'tcx FnDecl<'_>,
|
||||||
body: &'tcx Body<'_>,
|
body: &'tcx Body<'_>,
|
||||||
span: Span,
|
span: Span,
|
||||||
def_id: LocalDefId,
|
fn_def_id: LocalDefId,
|
||||||
) {
|
) {
|
||||||
if let Some(header) = kind.header()
|
if let Some(header) = kind.header()
|
||||||
&& !header.asyncness.is_async()
|
&& !header.asyncness.is_async()
|
||||||
// Check that this function returns `impl Future`
|
// Check that this function returns `impl Future`
|
||||||
&& let FnRetTy::Return(ret_ty) = decl.output
|
&& let FnRetTy::Return(ret_ty) = decl.output
|
||||||
&& let Some((trait_ref, output_lifetimes)) = future_trait_ref(cx, ret_ty)
|
&& let TyKind::OpaqueDef(opaque) = ret_ty.kind
|
||||||
|
&& let Some(trait_ref) = future_trait_ref(cx, opaque)
|
||||||
&& let Some(output) = future_output_ty(trait_ref)
|
&& let Some(output) = future_output_ty(trait_ref)
|
||||||
&& captures_all_lifetimes(decl.inputs, &output_lifetimes)
|
&& captures_all_lifetimes(cx, fn_def_id, opaque.def_id)
|
||||||
// Check that the body of the function consists of one async block
|
// Check that the body of the function consists of one async block
|
||||||
&& let ExprKind::Block(block, _) = body.value.kind
|
&& let ExprKind::Block(block, _) = body.value.kind
|
||||||
&& block.stmts.is_empty()
|
&& block.stmts.is_empty()
|
||||||
&& let Some(closure_body) = desugared_async_block(cx, block)
|
&& let Some(closure_body) = desugared_async_block(cx, block)
|
||||||
&& let Node::Item(Item {vis_span, ..}) | Node::ImplItem(ImplItem {vis_span, ..}) =
|
&& let Node::Item(Item {vis_span, ..}) | Node::ImplItem(ImplItem {vis_span, ..}) =
|
||||||
cx.tcx.hir_node_by_def_id(def_id)
|
cx.tcx.hir_node_by_def_id(fn_def_id)
|
||||||
{
|
{
|
||||||
let header_span = span.with_hi(ret_ty.span.hi());
|
let header_span = span.with_hi(ret_ty.span.hi());
|
||||||
|
|
||||||
@ -101,12 +104,8 @@ fn check_fn(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn future_trait_ref<'tcx>(
|
fn future_trait_ref<'tcx>(cx: &LateContext<'tcx>, opaque: &'tcx OpaqueTy<'tcx>) -> Option<&'tcx TraitRef<'tcx>> {
|
||||||
cx: &LateContext<'tcx>,
|
if let Some(trait_ref) = opaque.bounds.iter().find_map(|bound| {
|
||||||
ty: &'tcx Ty<'tcx>,
|
|
||||||
) -> Option<(&'tcx TraitRef<'tcx>, Vec<LifetimeName>)> {
|
|
||||||
if let TyKind::OpaqueDef(opaque, bounds) = ty.kind
|
|
||||||
&& let Some(trait_ref) = opaque.bounds.iter().find_map(|bound| {
|
|
||||||
if let GenericBound::Trait(poly) = bound {
|
if let GenericBound::Trait(poly) = bound {
|
||||||
Some(&poly.trait_ref)
|
Some(&poly.trait_ref)
|
||||||
} else {
|
} else {
|
||||||
@ -115,18 +114,7 @@ fn future_trait_ref<'tcx>(
|
|||||||
})
|
})
|
||||||
&& trait_ref.trait_def_id() == cx.tcx.lang_items().future_trait()
|
&& trait_ref.trait_def_id() == cx.tcx.lang_items().future_trait()
|
||||||
{
|
{
|
||||||
let output_lifetimes = bounds
|
return Some(trait_ref);
|
||||||
.iter()
|
|
||||||
.filter_map(|bound| {
|
|
||||||
if let GenericArg::Lifetime(lt) = bound {
|
|
||||||
Some(lt.res)
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
return Some((trait_ref, output_lifetimes));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
None
|
None
|
||||||
@ -145,27 +133,35 @@ fn future_output_ty<'tcx>(trait_ref: &'tcx TraitRef<'tcx>) -> Option<&'tcx Ty<'t
|
|||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
fn captures_all_lifetimes(inputs: &[Ty<'_>], output_lifetimes: &[LifetimeName]) -> bool {
|
fn captures_all_lifetimes(cx: &LateContext<'_>, fn_def_id: LocalDefId, opaque_def_id: LocalDefId) -> bool {
|
||||||
let input_lifetimes: Vec<LifetimeName> = inputs
|
let early_input_params = ty::GenericArgs::identity_for_item(cx.tcx, fn_def_id);
|
||||||
.iter()
|
let late_input_params = cx.tcx.late_bound_vars(cx.tcx.local_def_id_to_hir_id(fn_def_id));
|
||||||
.filter_map(|ty| {
|
|
||||||
if let TyKind::Ref(lt, _) = ty.kind {
|
|
||||||
Some(lt.res)
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
// The lint should trigger in one of these cases:
|
let num_early_lifetimes = early_input_params
|
||||||
// - There are no input lifetimes
|
.iter()
|
||||||
// - There's only one output lifetime bound using `+ '_`
|
.filter(|param| param.as_region().is_some())
|
||||||
// - All input lifetimes are explicitly bound to the output
|
.count();
|
||||||
input_lifetimes.is_empty()
|
let num_late_lifetimes = late_input_params
|
||||||
|| (output_lifetimes.len() == 1 && matches!(output_lifetimes[0], LifetimeName::Infer))
|
.iter()
|
||||||
|| input_lifetimes
|
.filter(|param_kind| matches!(param_kind, ty::BoundVariableKind::Region(_)))
|
||||||
.iter()
|
.count();
|
||||||
.all(|in_lt| output_lifetimes.iter().any(|out_lt| in_lt == out_lt))
|
|
||||||
|
// There is no lifetime, so they are all captured.
|
||||||
|
if num_early_lifetimes == 0 && num_late_lifetimes == 0 {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// By construction, each captured lifetime only appears once in `opaque_captured_lifetimes`.
|
||||||
|
let num_captured_lifetimes = cx
|
||||||
|
.tcx
|
||||||
|
.opaque_captured_lifetimes(opaque_def_id)
|
||||||
|
.iter()
|
||||||
|
.filter(|&(lifetime, _)| match *lifetime {
|
||||||
|
ResolvedArg::EarlyBound(_) | ResolvedArg::LateBound(ty::INNERMOST, _, _) => true,
|
||||||
|
_ => false,
|
||||||
|
})
|
||||||
|
.count();
|
||||||
|
num_captured_lifetimes == num_early_lifetimes + num_late_lifetimes
|
||||||
}
|
}
|
||||||
|
|
||||||
fn desugared_async_block<'tcx>(cx: &LateContext<'tcx>, block: &'tcx Block<'tcx>) -> Option<&'tcx Body<'tcx>> {
|
fn desugared_async_block<'tcx>(cx: &LateContext<'tcx>, block: &'tcx Block<'tcx>) -> Option<&'tcx Body<'tcx>> {
|
||||||
|
@ -1231,16 +1231,13 @@ pub fn hash_tykind(&mut self, ty: &TyKind<'_>) {
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
TyKind::Path(ref qpath) => self.hash_qpath(qpath),
|
TyKind::Path(ref qpath) => self.hash_qpath(qpath),
|
||||||
TyKind::OpaqueDef(_, arg_list) => {
|
|
||||||
self.hash_generic_args(arg_list);
|
|
||||||
},
|
|
||||||
TyKind::TraitObject(_, lifetime, _) => {
|
TyKind::TraitObject(_, lifetime, _) => {
|
||||||
self.hash_lifetime(lifetime);
|
self.hash_lifetime(lifetime);
|
||||||
},
|
},
|
||||||
TyKind::Typeof(anon_const) => {
|
TyKind::Typeof(anon_const) => {
|
||||||
self.hash_body(anon_const.body);
|
self.hash_body(anon_const.body);
|
||||||
},
|
},
|
||||||
TyKind::Err(_) | TyKind::Infer | TyKind::Never | TyKind::InferDelegation(..) | TyKind::AnonAdt(_) => {},
|
TyKind::Err(_) | TyKind::Infer | TyKind::Never | TyKind::InferDelegation(..) | TyKind::OpaqueDef(_) | TyKind::AnonAdt(_) => {},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -11,7 +11,7 @@ error: the following explicit lifetimes could be elided: 'a
|
|||||||
--> tests/ui/issue_4266.rs:10:21
|
--> tests/ui/issue_4266.rs:10:21
|
||||||
|
|
|
|
||||||
LL | async fn one_to_one<'a>(s: &'a str) -> &'a str {
|
LL | async fn one_to_one<'a>(s: &'a str) -> &'a str {
|
||||||
| ^^ ^^
|
| ^^ ^^ ^^
|
||||||
|
|
||||||
error: methods called `new` usually take no `self`
|
error: methods called `new` usually take no `self`
|
||||||
--> tests/ui/issue_4266.rs:31:22
|
--> tests/ui/issue_4266.rs:31:22
|
||||||
|
Loading…
Reference in New Issue
Block a user