Moved updateResults function to a static method of Location

cmccormack · Jun 11, 2018 · b9ddf93 · b9ddf93
1 parent 9245f14
commit b9ddf93
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/models/Location.js b/models/Location.js
@@ -9,9 +9,11 @@ const locationSchema = new mongoose.Schema({
   },],
 })
 
-locationSchema.methods.goingCount = function() {
-  console.log("location.going.length: " + this.going.length)
-  return this.going.length
+
+
+locationSchema.statics.updateResults = function(results) {
+  return this.find({ id: { $in: results, }, })
+    .populate("going", "twitter.displayName")
 }
 
 const Location = mongoose.model("Location", locationSchema, "locations")

diff --git a/routes/app.js b/routes/app.js
@@ -44,10 +44,11 @@ module.exports = () => {
   // Return db entries based on received Location IDs
   ///////////////////////////////////////////////////////////
   router.post("/updatedresults", async (req, res, next) => {
+    if (!req.body) return next(Error("Must supply array of locations"))
+
     const { resultIds, } = req.body
+    const updates = await Location.updateResults(resultIds)
 
-    const updates = await Location.find({ id: { $in: resultIds,},})
-    console.log(updates)
     res.json({success: true, updates,})
   })