prevent app crash when document doesn't exist

correctly return 404 instead
oori · Oct 12, 2012 · 91c9686 · 91c9686
1 parent c34f179
commit 91c9686
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/routes.js b/lib/routes.js
@@ -107,7 +107,7 @@ module.exports = function MongooseRestRouter(base, router, options) {
     router.put(base + '/:type/:id', function (req, res, next) {
         var id = req.param('id');
         m(req).findOne(idObj(req), function (err, obj) {
-            if (err)return next(err);
+            if (err || doc==null) return next(err);
             _u.extend(obj, clean(req.body));
             obj.save(function (err, ret) {
                 if (err) return next(err);
@@ -133,7 +133,7 @@ module.exports = function MongooseRestRouter(base, router, options) {
         }
 
         M.findOne(idObj(req), function(err, doc){
-            if (err) return next(err);
+            if (err || doc==null) return next(err);
             doc.remove(send);
         });