diff options
author | Kaloian Manassiev <kaloian.manassiev@mongodb.com> | 2015-07-09 16:50:30 -0400 |
---|---|---|
committer | Kaloian Manassiev <kaloian.manassiev@mongodb.com> | 2015-07-11 05:49:49 -0400 |
commit | ac42068ddeaae27d2cd5cfc4808915491e5097c7 (patch) | |
tree | eda7ced79e545b3fce7f6ca0d892d330e932f0a1 /src/mongo/db/s/sharded_connection_info.cpp | |
parent | 15c72c8570c63e2e6ba0a3d339a8286d0be604db (diff) | |
download | mongo-ac42068ddeaae27d2cd5cfc4808915491e5097c7.tar.gz |
SERVER-18084 Move code out of d_state.h/.cpp
Move the ShardingState/ShardedConnectionInfo classes out of d_state and
put them in separate sources under mongo/db.
No functional changes.
Diffstat (limited to 'src/mongo/db/s/sharded_connection_info.cpp')
-rw-r--r-- | src/mongo/db/s/sharded_connection_info.cpp | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/src/mongo/db/s/sharded_connection_info.cpp b/src/mongo/db/s/sharded_connection_info.cpp new file mode 100644 index 00000000000..e3029bd950a --- /dev/null +++ b/src/mongo/db/s/sharded_connection_info.cpp @@ -0,0 +1,103 @@ +/** + * 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. + */ + +#define MONGO_LOG_DEFAULT_COMPONENT ::mongo::logger::LogComponent::kSharding + +#include "mongo/platform/basic.h" + +#include "mongo/db/s/sharded_connection_info.h" + +#include <boost/optional.hpp> + +#include "mongo/client/global_conn_pool.h" +#include "mongo/db/client.h" +#include "mongo/db/operation_context.h" +#include "mongo/s/chunk_version.h" +#include "mongo/s/client/shard_connection.h" +#include "mongo/s/client/sharding_connection_hook.h" +#include "mongo/util/log.h" + +namespace mongo { + +namespace { + +const auto clientSCI = Client::declareDecoration<boost::optional<ShardedConnectionInfo>>(); + +} // namespace + +ShardedConnectionInfo::ShardedConnectionInfo() { + _forceVersionOk = false; +} + +ShardedConnectionInfo::~ShardedConnectionInfo() = default; + +ShardedConnectionInfo* ShardedConnectionInfo::get(Client* client, bool create) { + auto& current = clientSCI(client); + + if (!current && create) { + LOG(1) << "entering shard mode for connection"; + current = boost::in_place(); + } + + return current ? ¤t.value() : nullptr; +} + +void ShardedConnectionInfo::reset(Client* client) { + clientSCI(client) = boost::none; +} + +const ChunkVersion ShardedConnectionInfo::getVersion(const std::string& ns) const { + NSVersionMap::const_iterator it = _versions.find(ns); + if (it != _versions.end()) { + return it->second; + } else { + return ChunkVersion(0, 0, OID()); + } +} + +void ShardedConnectionInfo::setVersion(const std::string& ns, const ChunkVersion& version) { + _versions[ns] = version; +} + +void ShardedConnectionInfo::addHook() { + static stdx::mutex lock; + static bool done = false; + + stdx::lock_guard<stdx::mutex> lk(lock); + if (!done) { + log() << "first cluster operation detected, adding sharding hook to enable versioning " + "and authentication to remote servers"; + + globalConnPool.addHook(new ShardingConnectionHook(false)); + shardConnectionPool.addHook(new ShardingConnectionHook(true)); + + done = true; + } +} + +} // namespace mongo |