From 43745d2baeabfa11d5b24f33e2a1993196fcb46e Mon Sep 17 00:00:00 2001
From: Harish Krishna <harish.krsn@gmail.com>
Date: Mon, 8 Dec 2014 15:08:20 -0500
Subject: [PATCH] moving mixing logic into mix_results helper/suggest

---
 controller/suggest.js |  9 +------
 helper/suggest.js     | 57 ++++++++++++++++++++++++++++++-------------
 2 files changed, 41 insertions(+), 25 deletions(-)

diff --git a/controller/suggest.js b/controller/suggest.js
index 8ff68cb1..34df46f6 100644
--- a/controller/suggest.js
+++ b/controller/suggest.js
@@ -46,15 +46,8 @@ function setup( backend, query, query_mixer ){
     
     async.parallel(async_query, function(err, results) {
       // results is equal to: {a: docs, b: docs, c: docs}
-      var splice_length = parseInt((SIZE / Object.keys(results).length), 10);
       var results_keys = Object.keys(async_query);
-      
-      var combined = [];
-      results_keys.forEach(function(key){
-        combined = combined.concat(suggest.sort_by_score(results[key]).splice(0,splice_length));
-      });
-      
-      combined = suggest.dedup(combined);
+      var combined = suggest.mix_results(results, results_keys, SIZE);
       suggest.respond(combined);
     });
   
diff --git a/helper/suggest.js b/helper/suggest.js
index 2af16dd3..30e0e4f1 100644
--- a/helper/suggest.js
+++ b/helper/suggest.js
@@ -7,7 +7,7 @@ var geojsonify = require('../helper/geojsonify').search;
 function setup(backend, res, next) {
   var query_backend = function(cmd, callback) {
     // query backend
-    
+
     service.suggest( backend, cmd, function( err, docs ){
       
       // error handler
@@ -17,23 +17,48 @@ function setup(backend, res, next) {
     });
   };
 
-  var dedup = function(combined) {
-    var unique_ids = [];
-    return combined.filter(function(item, pos) {
-      if (unique_ids.indexOf(item.text) === -1) {
-        unique_ids.push(item.text);
-        return true;
-      }
-      return false;
-    });
-  };
-
   var sort_by_score = function(combined) {
-    return combined.sort(function(a,b) {
-      return b.score - a.score;
+    return combined.map(function(doc) {
+      return doc.sort(function(a,b) {
+        return b.score - a.score;
+      });
+    }).reduce(function(a,b) { //flatten
+      return a.concat(b);
     });
   };
 
+  var mix_results = function(results, results_keys, size) {
+    var i = 0;
+    var j = 0;
+    var l = results_keys.length;
+    var combined = [];
+    var unique_ids = [];
+
+    while (i<size && l > 0) {
+      if (results[results_keys[j]].length) {
+        combined[j] = combined[j] || [];
+
+        var res = results[results_keys[j]][0];
+        if (unique_ids.indexOf(res.text) === -1) {
+          combined[j].push(res);
+          unique_ids.push(res.text);
+          i++;
+        } 
+        results[results_keys[j]].splice(0,1);
+      } else {
+        results_keys.splice(j,1);
+        l = results_keys.length;
+        j--;
+      }
+      j++;
+      if (j%l === 0) {
+        j=0;
+      }
+    }
+    
+    return sort_by_score(combined);
+  };
+  
   var reply = function(res, docs) {
     
     // convert docs to geojson
@@ -77,9 +102,7 @@ function setup(backend, res, next) {
 
   return {
     query_backend: query_backend,
-    dedup: dedup,
-    sort_by_score: sort_by_score,
-    reply: reply,
+    mix_results: mix_results,
     respond: respond
   };