diff options
| -rw-r--r-- | src/mongo/db/s/SConscript | 22 | ||||
| -rw-r--r-- | src/mongo/db/s/chunk_splitter.cpp | 74 | ||||
| -rw-r--r-- | src/mongo/db/s/chunk_splitter.h | 60 | ||||
| -rw-r--r-- | src/mongo/db/s/chunk_splitter_test.cpp | 44 |
4 files changed, 200 insertions, 0 deletions
diff --git a/src/mongo/db/s/SConscript b/src/mongo/db/s/SConscript index 947488ae3e0..32814079adc 100644 --- a/src/mongo/db/s/SConscript +++ b/src/mongo/db/s/SConscript @@ -291,6 +291,17 @@ env.Library( ] ) +env.Library( + target='chunk_splitter', + source=[ + 'chunk_splitter.cpp', + ], + LIBDEPS=[ + '$BUILD_DIR/mongo/util/concurrency/thread_pool', + '$BUILD_DIR/mongo/db/service_context', + ] +) + env.CppUnitTest( target='chunk_size_tracker_test', source=[ @@ -300,3 +311,14 @@ env.CppUnitTest( 'chunk_size_tracker' ] ) + +env.CppUnitTest( + target='chunk_splitter_test', + source=[ + 'chunk_splitter_test.cpp' + ], + LIBDEPS=[ + 'chunk_splitter', + '$BUILD_DIR/mongo/s/shard_server_test_fixture' + ] +) diff --git a/src/mongo/db/s/chunk_splitter.cpp b/src/mongo/db/s/chunk_splitter.cpp new file mode 100644 index 00000000000..1f0575ce278 --- /dev/null +++ b/src/mongo/db/s/chunk_splitter.cpp @@ -0,0 +1,74 @@ +/** + * Copyright (C) 2017 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. + */ + +#include "mongo/platform/basic.h" + +#include "mongo/db/s/chunk_splitter.h" + +#include "mongo/db/client.h" +#include "mongo/db/namespace_string.h" +#include "mongo/db/operation_context.h" +#include "mongo/s/catalog/type_chunk.h" + +namespace mongo { +namespace { + +/** + * Constructs the default options for the thread pool used by the cache loader. + */ +ThreadPool::Options makeDefaultThreadPoolOptions() { + ThreadPool::Options options; + options.poolName = "ChunkSplitter"; + options.minThreads = 0; + options.maxThreads = 1; + + // Ensure all threads have a client + options.onCreateThread = [](const std::string& threadName) { + Client::initThread(threadName.c_str()); + }; + return options; +} + +} // namespace + +ChunkSplitter::ChunkSplitter() : _threadPool(makeDefaultThreadPoolOptions()) { + _threadPool.startup(); +} + +ChunkSplitter::~ChunkSplitter() { + _threadPool.shutdown(); + _threadPool.join(); +} + +bool ChunkSplitter::trySplitting(OperationContext* opCtx, + const NamespaceString& nss, + const ChunkRange& chunkRange) { + return false; +} + +} // namespace mongo diff --git a/src/mongo/db/s/chunk_splitter.h b/src/mongo/db/s/chunk_splitter.h new file mode 100644 index 00000000000..5f6b62affc3 --- /dev/null +++ b/src/mongo/db/s/chunk_splitter.h @@ -0,0 +1,60 @@ +/** + * Copyright (C) 2017 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/util/concurrency/thread_pool.h" + +namespace mongo { + +class ChunkRange; +class NamespaceString; +class OperationContext; +class ThreadPool; + +/** + * Handles asynchronous auto-splitting of chunks. + */ +class ChunkSplitter { +public: + ChunkSplitter(); + ~ChunkSplitter(); + + /** + * Schedules an autosplit task. Returns whether or not the task was successfully scheduled. + */ + bool trySplitting(OperationContext* opCtx, + const NamespaceString& nss, + const ChunkRange& chunkRange); + +private: + // Thread pool for parallelizing splits. + ThreadPool _threadPool; +}; + +} // namespace mongo diff --git a/src/mongo/db/s/chunk_splitter_test.cpp b/src/mongo/db/s/chunk_splitter_test.cpp new file mode 100644 index 00000000000..4668fbd9743 --- /dev/null +++ b/src/mongo/db/s/chunk_splitter_test.cpp @@ -0,0 +1,44 @@ +/** + * Copyright (C) 2017 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. + */ + +#include "mongo/platform/basic.h" + +#include "mongo/db/s/chunk_splitter.h" + +#include "mongo/s/shard_server_test_fixture.h" + +namespace mongo { +namespace { + +class ChunkSplitterTest : public ShardServerTestFixture {}; + +TEST_F(ChunkSplitterTest, SplitTest) {} +TEST_F(ChunkSplitterTest, DontSplitTest) {} + +} // namespace +} // namespace mongo |