Improve performances (get over findOne when possible)

inProgress-team · Mar 21, 2016 · 409c7a4 · 409c7a4
1 parent 369dde8
commit 409c7a4
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/Collection.js b/src/Collection.js
@@ -7,12 +7,25 @@ export default function(name) {
 
   return {
     find(selector, options) {
-      if(typeof selector == 'string') return [Data.db[name].findOne({_id: selector}, options)];
+      if(typeof selector == 'string') {
+        if(options) {
+          return [Data.db[name].findOne({_id: selector}, options)];
+        } else {
+          return [Data.db[name].get(selector)];
+        }
+      }
       return Data.db[name].find(selector, options);
 
     },
     findOne(selector, options) {
-      if(typeof selector == 'string') return Data.db[name].findOne({_id: selector}, options);
+
+      if(typeof selector == 'string') {
+        if(options) {
+          return Data.db[name].findOne({_id: selector}, options);
+        } else {
+          return Data.db[name].get(selector);
+        }
+      }
       return Data.db[name] && Data.db[name].findOne(selector, options)
 
     },