Add bench runnable and code lens

1 files changed, 5 insertions, 0 deletions

diff --git a/crates/ra_ide_api/src/runnables.rs b/crates/ra_ide_api/src/runnables.rs
index 98b1d2d55..53e49da5b 100644
--- a/crates/ra_ide_api/src/runnables.rs
+++ b/crates/ra_ide_api/src/runnables.rs
@@ -17,6 +17,7 @@ pub struct Runnable {
 pub enum RunnableKind {
     Test { name: String },
     TestMod { path: String },
+    Bench { name: String },
     Bin,
 }
 
@@ -48,6 +49,10 @@ fn runnable_fn(fn_def: &ast::FnDef) -> Option<Runnable> {
         RunnableKind::Test {
             name: name.to_string(),
         }
+    } else if fn_def.has_atom_attr("bench") {
+        RunnableKind::Bench {
+            name: name.to_string(),
+        }
     } else {
         return None;
     };