mongoose plugin aggregate and search support

Author: 2muchcoffeecom

spec/mongoosePluginSpec.js

@@ -15,13 +15,18 @@ let Author =  mongoose.model('Author', AuthorSchema);
 let PostSchema = new mongoose.Schema({
   title: String,
   date: Date,
-  body : String,
+  body : {
+    type: String,
+    text: true,
+  },
   author: {
     type: mongoose.Schema.ObjectId,
     ref: 'Author'
   }
 });
 
+PostSchema.index({body: 'text'});
+
 PostSchema.plugin(mongooseCursorPaginate);
 
 let Post = mongoose.model('Post', PostSchema);
@@ -67,4 +72,37 @@ describe('mongoose plugin', (it) => {
     t.is(data.hasOwnProperty('next'), true);
     t.is(data.hasOwnProperty('hasNext'), true);
   });
+
+  it('return find results', async function(t) {
+    let data = await Post.paginate({
+      query: {title: 'Post #99'},
+    });
+    t.is(data.results.length, 1);
+  });
+
+  it('return aggregate results', async function(t) {
+    let data = await Author.paginateFN({
+      aggregation: [
+        {
+          $lookup: {
+            from: 'posts',
+            localField: '_id',
+            foreignField: 'author',
+            as: 'posts',
+          }
+        },
+      ],
+    });
+    t.is(data.results.length, 1);
+    t.is(Array.isArray(data.results[0].posts), true);
+    t.is(data.results[0].posts.length, 100);
+  });
+
+  it('return search results', async function(t) {
+    let data = await Post.paginate({
+      search: '99',
+    });
+    t.is(data.results.length, 1);
+  });
+
 });

src/mongoose.plugin.js

@@ -1,5 +1,7 @@
 
 const find = require('./find');
+const search = require('./search');
+const aggregate = require('./aggregate');
 const _ = require('underscore');
 
 /**
@@ -22,7 +24,9 @@ module.exports = function (schema, options) {
     }
 
     param = _.extend({}, param);
-        
+
+    if (param.aggregation) return aggregate(this.collection, param);
+    if (param.search) return search(this.collection, param.search, param);
     return find(this.collection, param);
   };