1 files changed, 88 insertions, 0 deletions
diff --git a/src/mongo/db/kill_current_op.cpp b/src/mongo/db/kill_current_op.cpp
new file mode 100644
index 00000000000..17d0a6248d7
--- /dev/null
+++ b/src/mongo/db/kill_current_op.cpp
@@ -0,0 +1,88 @@
+/**
+*    Copyright (C) 2009 10gen Inc.
+*
+*    This program is free software: you can redistribute it and/or  modify
+*    it under the terms of the GNU Affero General Public License, version 3,
+*    as published by the Free Software Foundation.
+*
+*    This program is distributed in the hope that it will be useful,
+*    but WITHOUT ANY WARRANTY; without even the implied warranty of
+*    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+*    GNU Affero General Public License for more details.
+*
+*    You should have received a copy of the GNU Affero General Public License
+*    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "mongo/db/kill_current_op.h"
+
+#include <set>
+
+#include "mongo/bson/util/atomic_int.h"
+#include "mongo/db/client.h"
+#include "mongo/db/curop.h"
+#include "mongo/scripting/engine.h"
+
+namespace mongo {
+
+    void KillCurrentOp::interruptJs( AtomicUInt *op ) {
+        if ( !globalScriptEngine )
+            return;
+        if ( !op ) {
+            globalScriptEngine->interruptAll();
+        }
+        else {
+            globalScriptEngine->interrupt( *op );
+        }
+    }
+
+    void KillCurrentOp::killAll() {
+        _globalKill = true;
+        interruptJs( 0 );
+    }
+
+    void KillCurrentOp::kill(AtomicUInt i) {
+        bool found = false;
+        {
+            scoped_lock l( Client::clientsMutex );
+            for( set< Client* >::const_iterator j = Client::clients.begin(); !found && j != Client::clients.end(); ++j ) {
+                for( CurOp *k = ( *j )->curop(); !found && k; k = k->parent() ) {
+                    if ( k->opNum() == i ) {
+                        k->kill();
+                        for( CurOp *l = ( *j )->curop(); l != k; l = l->parent() ) {
+                            l->kill();
+                        }
+                        found = true;
+                    }
+                }
+            }
+        }
+        if ( found ) {
+            interruptJs( &i );
+        }
+    }
+
+    void KillCurrentOp::checkForInterrupt( bool heedMutex ) {
+        Client& c = cc();
+        if ( heedMutex && Lock::somethingWriteLocked() && c.hasWrittenThisPass() )
+            return;
+        if( _globalKill )
+            uasserted(11600,"interrupted at shutdown");
+        if( c.curop()->killed() ) {
+            uasserted(11601,"operation was interrupted");
+        }
+    }
+    
+    const char * KillCurrentOp::checkForInterruptNoAssert() {
+        Client& c = cc();
+        if( _globalKill )
+            return "interrupted at shutdown";
+        if( c.curop()->killed() )
+            return "interrupted";
+        return "";
+    }
+
+
+
+
+}