Question

#![feature(rustc_private)]
#![feature(box_syntax)]
extern crate rustc;
extern crate rustc_driver;

use rustc::hir::intravisit as hir_visit;
use rustc::hir;
use rustc_driver::driver::{CompileController, CompileState};

pub struct SomeVisitor<'a, 'tcx: 'a> {
    pub map: &'a hir::map::Map<'tcx>,
}

impl<'v, 'tcx: 'v> rustc::hir::intravisit::Visitor<'tcx> for SomeVisitor<'v, 'tcx> {
    fn nested_visit_map<'this>(&'this mut self) -> hir_visit::NestedVisitorMap<'this, 'tcx> {
        hir_visit::NestedVisitorMap::All(self.map)
    }
}

fn hir(s: &mut CompileState) {
    let krate = s.hir_crate.unwrap();
    let map = s.hir_map.unwrap();
    let mut visitor = SomeVisitor { map };
    hir_visit::walk_crate(&mut visitor, krate);
}

fn main() {
    {
        let mut controller = CompileController::basic();
        controller.after_hir_lowering.callback = box hir;
    }
}

playground

我明白为什么我会收到生命周期错误，通过为函数hir添加显式生存期来解决它很容易。

pub fn walk_crate<'v, V: hir_visit::Visitor<'v>>(visitor: &mut V, krate: &'v Crate) {}

由于此定义，参考的生命周期需要为'tcx生效。

fn hir<'v, 'tcx>(s: &'tcx mut CompileState<'v, 'tcx>) {
    let krate = s.hir_crate.unwrap();
    let map = s.hir_map.unwrap();
    let mut visitor = SomeVisitor { map };
    hir_visit::walk_crate(&mut visitor, krate);
}

但是函数hir变得与回调不兼容。 playground

我认为我可能需要在这里使用HRTB？

更新

我目前的解决方法是使用transmute。（playground）。当然必须有更好的方法吗？

Answer 1

hir_visit::walk_crate(&mut visitor, visitor.map.krate());

解决方案是要意识到地图还包含一个krate作为参考，但具有正确的生命周期。这意味着我不必引入明确的生命周期。

playground

显式生命周期会更改函数签名，并使函数与所需的类型签名不兼容

1 个答案: