diff options
author | jannaerin <golden.janna@gmail.com> | 2018-04-03 13:30:30 -0400 |
---|---|---|
committer | jannaerin <golden.janna@gmail.com> | 2018-04-03 17:38:42 -0400 |
commit | d053db6ac35f106823eb225635139286d0cff6a9 (patch) | |
tree | b037a76e0bc68cc759560a9a6b19ae91d2e2b491 /src/mongo/db/s/flush_database_cache_updates_command.cpp | |
parent | 5fa649cf8e0b77020d680c554aef1d380605611b (diff) | |
download | mongo-d053db6ac35f106823eb225635139286d0cff6a9.tar.gz |
SERVER-34145 Create flushDatabaseCacheUpdates internal command
Diffstat (limited to 'src/mongo/db/s/flush_database_cache_updates_command.cpp')
-rw-r--r-- | src/mongo/db/s/flush_database_cache_updates_command.cpp | 154 |
1 files changed, 154 insertions, 0 deletions
diff --git a/src/mongo/db/s/flush_database_cache_updates_command.cpp b/src/mongo/db/s/flush_database_cache_updates_command.cpp new file mode 100644 index 00000000000..deae485cca9 --- /dev/null +++ b/src/mongo/db/s/flush_database_cache_updates_command.cpp @@ -0,0 +1,154 @@ +/** + * Copyright (C) 2018 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/auth/action_set.h" +#include "mongo/db/auth/action_type.h" +#include "mongo/db/auth/authorization_manager.h" +#include "mongo/db/auth/authorization_session.h" +#include "mongo/db/auth/privilege.h" +#include "mongo/db/catalog_raii.h" +#include "mongo/db/client.h" +#include "mongo/db/commands.h" +#include "mongo/db/operation_context.h" +#include "mongo/db/repl/repl_client_info.h" +#include "mongo/db/s/database_sharding_state.h" +#include "mongo/db/s/migration_source_manager.h" +#include "mongo/db/s/operation_sharding_state.h" +#include "mongo/db/s/shard_filtering_metadata_refresh.h" +#include "mongo/db/s/sharding_state.h" +#include "mongo/s/catalog_cache_loader.h" +#include "mongo/s/grid.h" +#include "mongo/s/request_types/flush_database_cache_updates_gen.h" + + +#include "mongo/util/log.h" + +namespace mongo { +namespace { + +class FlushDatabaseCacheUpdates : public BasicCommand { +public: + FlushDatabaseCacheUpdates() : BasicCommand("_flushDatabaseCacheUpdates") {} + + std::string help() const override { + return "Internal command which waits for any pending routing table cache updates for a " + "particular database to be written locally. The operationTime returned in the " + "response metadata is guaranteed to be at least as late as the last routing table " + "cache update to the local disk. Takes a 'forceRemoteRefresh' option to make this " + "node refresh its cache from the config server before waiting for the last refresh " + "to be persisted."; + } + + bool adminOnly() const override { + return true; + } + + AllowedOnSecondary secondaryAllowed(ServiceContext*) const override { + return AllowedOnSecondary::kNever; + } + + bool supportsWriteConcern(const BSONObj& cmd) const override { + return false; + } + + Status checkAuthForCommand(Client* client, + const std::string& dbname, + const BSONObj& cmdObj) const override { + if (!AuthorizationSession::get(client)->isAuthorizedForActionsOnResource( + ResourcePattern::forClusterResource(), ActionType::internal)) { + return Status(ErrorCodes::Unauthorized, "Unauthorized"); + } + return Status::OK(); + } + + void addRequiredPrivileges(const std::string& dbname, + const BSONObj& cmdObj, + std::vector<Privilege>* out) const override { + ActionSet actions; + actions.addAction(ActionType::internal); + out->push_back(Privilege(ResourcePattern::forClusterResource(), actions)); + } + + bool run(OperationContext* opCtx, + const std::string& dbname, + const BSONObj& cmdObj, + BSONObjBuilder& result) override { + auto const shardingState = ShardingState::get(opCtx); + uassertStatusOK(shardingState->canAcceptShardedCommands()); + + uassert(ErrorCodes::IllegalOperation, + "Can't issue _flushDatabaseCacheUpdates from 'eval'", + !opCtx->getClient()->isInDirectClient()); + + uassert(ErrorCodes::IllegalOperation, + "Can't call _flushDatabaseCacheUpdates if in read-only mode", + !storageGlobalParams.readOnly); + + auto& oss = OperationShardingState::get(opCtx); + + const auto request = _flushDatabaseCacheUpdatesRequest::parse( + IDLParserErrorContext("_FlushDatabaseCacheUpdatesRequest"), cmdObj); + auto name = request.getCommandParameter().toString(); + + { + AutoGetDb autoDb(opCtx, name, MODE_IS); + + // If the primary is in the critical section, secondaries must wait for the commit to + // finish on the primary in case a secondary's caller has an afterClusterTime inclusive + // of the commit (and new writes to the committed chunk) that hasn't yet propagated back + // to this shard. This ensures the read your own writes causal consistency guarantee. + auto criticalSectionSignal = + DatabaseShardingState::get(autoDb.getDb()) + .getCriticalSectionSignal(ShardingMigrationCriticalSection::kRead); + if (criticalSectionSignal) { + oss.setMigrationCriticalSectionSignal(criticalSectionSignal); + } + } + + oss.waitForMigrationCriticalSectionSignal(opCtx); + + if (request.getSyncFromConfig()) { + LOG(1) << "Forcing remote routing table refresh for " << name; + forceDatabaseRefresh(opCtx, name); + } + + CatalogCacheLoader::get(opCtx).waitForDatabaseFlush(opCtx, name); + + repl::ReplClientInfo::forClient(opCtx->getClient()).setLastOpToSystemLastOpTime(opCtx); + + return true; + } + +} _flushDatabaseCacheUpdatesCmd; + +} // namespace +} // namespace mongo |