summaryrefslogtreecommitdiff
path: root/src/mongo/db/s/shard_local.h
diff options
context:
space:
mode:
authorKaloian Manassiev <kaloian.manassiev@mongodb.com>2020-06-16 07:53:22 -0400
committerEvergreen Agent <no-reply@evergreen.mongodb.com>2020-06-21 12:32:58 +0000
commit33201c068f9ec37196190be083d61f46dc83d470 (patch)
tree59a30062c8fd5ece6cecc4702c28d2d5417b5b8c /src/mongo/db/s/shard_local.h
parent85707dafe1d2894505fdcf74336f93ed7a5f6636 (diff)
downloadmongo-33201c068f9ec37196190be083d61f46dc83d470.tar.gz
SERVER-48775 Move all Sharding unit-tests that assume a MongoD to the db/s directory
With this change there are no more references from mongo/s (which is code common between MongoS and MongoD) and mongo/db/s (which is code specific for MongoD only).
Diffstat (limited to 'src/mongo/db/s/shard_local.h')
-rw-r--r--src/mongo/db/s/shard_local.h100
1 files changed, 100 insertions, 0 deletions
diff --git a/src/mongo/db/s/shard_local.h b/src/mongo/db/s/shard_local.h
new file mode 100644
index 00000000000..2bce2dad081
--- /dev/null
+++ b/src/mongo/db/s/shard_local.h
@@ -0,0 +1,100 @@
+/**
+ * Copyright (C) 2018-present MongoDB, Inc.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the Server Side Public License, version 1,
+ * as published by MongoDB, Inc.
+ *
+ * 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
+ * Server Side Public License for more details.
+ *
+ * You should have received a copy of the Server Side Public License
+ * along with this program. If not, see
+ * <http://www.mongodb.com/licensing/server-side-public-license>.
+ *
+ * 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 Server Side 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/db/rs_local_client.h"
+#include "mongo/s/client/shard.h"
+
+namespace mongo {
+
+class ShardLocal : public Shard {
+ ShardLocal(const ShardLocal&) = delete;
+ ShardLocal& operator=(const ShardLocal&) = delete;
+
+public:
+ explicit ShardLocal(const ShardId& id);
+
+ ~ShardLocal() = default;
+
+ /**
+ * These functions are implemented for the Shard interface's sake. They should not be called on
+ * ShardLocal because doing so triggers invariants.
+ */
+ const ConnectionString getConnString() const override;
+ const ConnectionString originalConnString() const override;
+ std::shared_ptr<RemoteCommandTargeter> getTargeter() const override;
+ void updateReplSetMonitor(const HostAndPort& remoteHost,
+ const Status& remoteCommandStatus) override;
+
+ std::string toString() const override;
+
+ bool isRetriableError(ErrorCodes::Error code, RetryPolicy options) final;
+
+ Status createIndexOnConfig(OperationContext* opCtx,
+ const NamespaceString& ns,
+ const BSONObj& keys,
+ bool unique) override;
+
+ void updateLastCommittedOpTime(LogicalTime lastCommittedOpTime) final;
+
+ LogicalTime getLastCommittedOpTime() const final;
+
+ void runFireAndForgetCommand(OperationContext* opCtx,
+ const ReadPreferenceSetting& readPref,
+ const std::string& dbName,
+ const BSONObj& cmdObj) override;
+
+private:
+ StatusWith<Shard::CommandResponse> _runCommand(OperationContext* opCtx,
+ const ReadPreferenceSetting& unused,
+ StringData dbName,
+ Milliseconds maxTimeMSOverrideUnused,
+ const BSONObj& cmdObj) final;
+
+ StatusWith<Shard::QueryResponse> _runExhaustiveCursorCommand(
+ OperationContext* opCtx,
+ const ReadPreferenceSetting& readPref,
+ StringData dbName,
+ Milliseconds maxTimeMSOverride,
+ const BSONObj& cmdObj) final;
+
+ StatusWith<Shard::QueryResponse> _exhaustiveFindOnConfig(
+ OperationContext* opCtx,
+ const ReadPreferenceSetting& readPref,
+ const repl::ReadConcernLevel& readConcernLevel,
+ const NamespaceString& nss,
+ const BSONObj& query,
+ const BSONObj& sort,
+ boost::optional<long long> limit) final;
+
+ RSLocalClient _rsLocalClient;
+};
+
+} // namespace mongo