From fccd0452299c8261e0298688d29e934b0fef1605 Mon Sep 17 00:00:00 2001
From: Edwin Cheng <edwin0cheng@gmail.com>
Date: Mon, 22 Apr 2019 22:46:39 +0800
Subject: Add cached for SubtreeSource

---
 crates/ra_mbe/src/subtree_source.rs | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

(limited to 'crates/ra_mbe/src')

diff --git a/crates/ra_mbe/src/subtree_source.rs b/crates/ra_mbe/src/subtree_source.rs
index 20da1e9f7..3229cfa8f 100644
--- a/crates/ra_mbe/src/subtree_source.rs
+++ b/crates/ra_mbe/src/subtree_source.rs
@@ -219,17 +219,33 @@ pub(crate) trait Querier {
 #[derive(Debug)]
 pub(crate) struct WalkerOwner<'a> {
     walker: RefCell<SubTreeWalker<'a>>,
+    cached: RefCell<Vec<Option<TtToken>>>,
 }
 
 impl<'a> WalkerOwner<'a> {
     fn new<I: Into<TokenSeq<'a>>>(ts: I) -> Self {
-        WalkerOwner { walker: RefCell::new(SubTreeWalker::new(ts.into())) }
+        WalkerOwner {
+            walker: RefCell::new(SubTreeWalker::new(ts.into())),
+            cached: RefCell::new(Vec::with_capacity(10)),
+        }
     }
 
     fn get<'b>(&self, pos: usize) -> Option<TtToken> {
-        self.set_pos(pos);
-        let walker = self.walker.borrow();
-        walker.current().cloned()
+        let mut cached = self.cached.borrow_mut();
+        if pos < cached.len() {
+            return cached[pos].clone();
+        }
+
+        while pos >= cached.len() {
+            let len = cached.len();
+            cached.push({
+                self.set_pos(len);
+                let walker = self.walker.borrow();
+                walker.current().cloned()
+            });
+        }
+
+        return cached[pos].clone();
     }
 
     fn set_pos(&self, pos: usize) {
-- 
cgit v1.2.3