issue/50 Moved segment local collection to the DocSet object.

1 files changed, 9 insertions, 1 deletions

diff --git a/src/query/scorer.rs b/src/query/scorer.rs
index 7e9f174..21d57ee 100644
--- a/src/query/scorer.rs
+++ b/src/query/scorer.rs
@@ -1,5 +1,6 @@
 use DocSet;
-
+use ScoredDoc;
+use collector::Collector;
 
 /// Scored `DocSet`
 pub trait Scorer: DocSet {
@@ -8,6 +9,13 @@ pub trait Scorer: DocSet {
     /// 
     /// This method will perform a bit of computation and is not cached.
     fn score(&self,) -> f32;
+    
+    fn collect(&mut self, collector: &mut Collector) {
+        while self.advance() {
+            let scored_doc = ScoredDoc(self.score(), self.doc());
+            collector.collect(scored_doc);
+        }
+    }
 }