only check for specific mongo methods

officert · Jan 27, 2016 · ca7a9da · ca7a9da
1 parent 34e6021
commit ca7a9da
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/ui/directives/query/queryCtrl.js b/src/ui/directives/query/queryCtrl.js
@@ -109,7 +109,7 @@ angular.module('app').controller('queryCtrl', [
                 name: expressionResult.mongoCollectionName
               });
 
-              if (expressionResult.mongoMethodName !== 'find' && expressionResult.mongoMethodName !== 'aggregate' && expressionResult.mongoMethodName !== 'count') {
+              if (_isModifyingMongoMethod(expressionResult.mongoMethodName)) {
                 notificationService.success(`${expressionResult.mongoMethodName} was successful`);
 
                 return _evalExpression(`db.${expressionResult.mongoCollectionName}.find()`, $scope.database.collections);
@@ -132,6 +132,10 @@ angular.module('app').controller('queryCtrl', [
         });
     }
 
+    function _isModifyingMongoMethod(methodName) {
+      return methodName && (methodName === 'updateMany'|| methodName === 'updateById' || methodName === 'updateOne' || methodName === 'deleteMany' || methodName === 'deleteById' || methodName === 'deleteOne');
+    }
+
     function _deleteDocument(doc) {
       if (!doc) return;