diff options
Diffstat (limited to 'src/mongo/s/request_types/add_shard_request_type.cpp')
-rw-r--r-- | src/mongo/s/request_types/add_shard_request_type.cpp | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/src/mongo/s/request_types/add_shard_request_type.cpp b/src/mongo/s/request_types/add_shard_request_type.cpp new file mode 100644 index 00000000000..c54e7af5b3f --- /dev/null +++ b/src/mongo/s/request_types/add_shard_request_type.cpp @@ -0,0 +1,155 @@ +/** + * Copyright (C) 2016 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/s/request_types/add_shard_request_type.h" + +#include "mongo/bson/util/bson_extract.h" +#include "mongo/db/server_options.h" +#include "mongo/util/mongoutils/str.h" + +namespace mongo { + +using std::string; +using str::stream; + +class BSONObj; +template <typename T> +class StatusWith; + +const BSONField<std::string> AddShardRequest::mongosAddShard("addShard"); +const BSONField<std::string> AddShardRequest::mongosAddShardDeprecated("addshard"); +const BSONField<std::string> AddShardRequest::configsvrAddShard("_configsvrAddShard"); +const BSONField<std::string> AddShardRequest::shardName("name"); +const BSONField<long long> AddShardRequest::maxSizeMB("maxSize"); + +AddShardRequest::AddShardRequest(ConnectionString connString) + : _connString(std::move(connString)) {} + +StatusWith<AddShardRequest> AddShardRequest::parseFromMongosCommand(const BSONObj& obj) { + invariant(obj.nFields() >= 1); + + auto firstElement = obj.firstElement(); + invariant(firstElement.type() == BSONType::String); + invariant(mongosAddShard.name() == firstElement.fieldNameStringData() || + mongosAddShardDeprecated.name() == firstElement.fieldNameStringData()); + + return parseInternalFields(obj); +} + +StatusWith<AddShardRequest> AddShardRequest::parseFromConfigCommand(const BSONObj& obj) { + invariant(obj.nFields() >= 1); + + auto firstElement = obj.firstElement(); + invariant(firstElement.type() == BSONType::String); + invariant(configsvrAddShard.name() == firstElement.fieldNameStringData()); + + return parseInternalFields(obj); +} + +StatusWith<AddShardRequest> AddShardRequest::parseInternalFields(const BSONObj& obj) { + // required fields + + auto swConnString = ConnectionString::parse(obj.firstElement().valuestrsafe()); + if (!swConnString.isOK()) { + return swConnString.getStatus(); + } + ConnectionString connString = std::move(swConnString.getValue()); + + AddShardRequest request(std::move(connString)); + + // optional fields + + { + string requestShardName; + Status status = bsonExtractStringField(obj, shardName.name(), &requestShardName); + if (status.isOK()) { + request._name = std::move(requestShardName); + } else if (status != ErrorCodes::NoSuchKey) { + return status; + } + } + { + long long requestMaxSizeMB; + Status status = bsonExtractIntegerField(obj, maxSizeMB.name(), &requestMaxSizeMB); + if (status.isOK()) { + request._maxSizeMB = std::move(requestMaxSizeMB); + } else if (status != ErrorCodes::NoSuchKey) { + return status; + } + } + + return request; +} + +BSONObj AddShardRequest::toCommandForConfig() { + BSONObjBuilder cmdBuilder; + cmdBuilder.append(configsvrAddShard.name(), _connString.toString()); + if (hasMaxSize()) { + cmdBuilder.append(maxSizeMB.name(), *_maxSizeMB); + } + if (hasName()) { + cmdBuilder.append(shardName.name(), *_name); + } + return cmdBuilder.obj(); +} + +Status AddShardRequest::validate(bool allowLocalHost) { + // Check that if one of the new shard's hosts is localhost, we are allowed to use localhost + // as a hostname. (Using localhost requires that every server in the cluster uses + // localhost). + std::vector<HostAndPort> serverAddrs = _connString.getServers(); + for (size_t i = 0; i < serverAddrs.size(); i++) { + if (serverAddrs[i].isLocalHost() != allowLocalHost) { + string errmsg = str::stream() + << "Can't use localhost as a shard since all shards need to" + << " communicate. Either use all shards and configdbs in localhost" + << " or all in actual IPs. host: " << serverAddrs[i].toString() + << " isLocalHost:" << serverAddrs[i].isLocalHost(); + return Status(ErrorCodes::InvalidOptions, errmsg); + } + + // If the server has no port, assign it the default port. + if (!serverAddrs[i].hasPort()) { + serverAddrs[i] = + HostAndPort(serverAddrs[i].host(), ServerGlobalParams::ShardServerPort); + } + } + return Status::OK(); +} + +string AddShardRequest::toString() const { + stream ss; + ss << "AddShardRequest shard: " << _connString.toString(); + if (hasName()) + ss << ", name: " << *_name; + if (hasMaxSize()) + ss << ", maxSize: " << *_maxSizeMB; + return ss; +} + +} // namespace mongo |