diff options
author | Qingyang Chen <qingyang.chen@10gen.com> | 2015-07-29 16:06:50 -0400 |
---|---|---|
committer | Qingyang Chen <qingyang.chen@10gen.com> | 2015-08-12 15:48:52 -0400 |
commit | ca402fdcd2b416fb91ad08b886c53551ea9ebee0 (patch) | |
tree | 79ed6faed405962a7b5ef4a4ef809f241c465df6 /src/mongo/scripting/mozjs/cursor_handle.h | |
parent | d2155f39efb73cdf3e229957f1eb639855e35be3 (diff) | |
download | mongo-ca402fdcd2b416fb91ad08b886c53551ea9ebee0.tar.gz |
SERVER-19570 Issue kill against find command cursors in shell when cursor object is destroyed
Diffstat (limited to 'src/mongo/scripting/mozjs/cursor_handle.h')
-rw-r--r-- | src/mongo/scripting/mozjs/cursor_handle.h | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/src/mongo/scripting/mozjs/cursor_handle.h b/src/mongo/scripting/mozjs/cursor_handle.h new file mode 100644 index 00000000000..95f00a3d964 --- /dev/null +++ b/src/mongo/scripting/mozjs/cursor_handle.h @@ -0,0 +1,72 @@ +/** + * Copyright (C) 2015 MongoDB 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/>. + * + * As a special exception, the copyright holders give permission to link the + * code of portions of this program with the OpenSSL library under certain + * conditions as described in each individual source file and distribute + * linked combinations including the program with the OpenSSL library. You + * must comply with the GNU Affero General Public License in all respects + * for all of the code used other than as permitted herein. If you modify + * file(s) with this exception, you may extend this exception to your + * version of the file(s), but you are not obligated to do so. If you do not + * wish to do so, delete this exception statement from your version. If you + * delete this exception statement from all source files in the program, + * then also delete it in the license file. + */ + +#pragma once + +#include "mongo/client/dbclientcursor.h" +#include "mongo/client/dbclientinterface.h" +#include "mongo/scripting/mozjs/wraptype.h" + +namespace mongo { +namespace mozjs { + +/** + * Wraps a DBClientCursor in javascript. + * + * Note that the install is private, so this class should only be constructible from C++. Current + * callers are all via the Mongo object. + */ +struct CursorHandleInfo : public BaseInfo { + static void construct(JSContext* cx, JS::CallArgs args); + static void finalize(JSFreeOp* fop, JSObject* obj); + + struct Functions { + MONGO_DEFINE_JS_FUNCTION(zeroCursorId); + }; + + static const JSFunctionSpec methods[2]; + + static const char* const className; + static const unsigned classFlags = JSCLASS_HAS_PRIVATE; + static const InstallType installType = InstallType::Private; + + /** + * We need this because the DBClientBase can go out of scope before all of its children (as + * in global shutdown). So we have to manage object lifetimes in C++ land. + */ + struct CursorTracker { + CursorTracker(long long curId, std::shared_ptr<DBClientBase> client) + : client(std::move(client)), cursorId(curId) {} + + std::shared_ptr<DBClientBase> client; + long long cursorId; + }; +}; + +} // namespace mozjs +} // namespace mongo |