diff options
author | Hugh Tong <hugh.tong@mongodb.com> | 2022-05-09 16:00:06 +0000 |
---|---|---|
committer | Evergreen Agent <no-reply@evergreen.mongodb.com> | 2022-05-09 21:31:49 +0000 |
commit | 6989f87e400acff7e668ba4472d4f91adf938683 (patch) | |
tree | 276ba283e6ef6f0506b940f2dab2b6f1f71371a6 /src/mongo/db/database_name.h | |
parent | 32e2e0d1385f6e47f829ff70b9fce95b3873a54f (diff) | |
download | mongo-6989f87e400acff7e668ba4472d4f91adf938683.tar.gz |
SERVER-64609 Rename instances of TenantDatabaseName to DatabaseName
Diffstat (limited to 'src/mongo/db/database_name.h')
-rw-r--r-- | src/mongo/db/database_name.h | 151 |
1 files changed, 151 insertions, 0 deletions
diff --git a/src/mongo/db/database_name.h b/src/mongo/db/database_name.h new file mode 100644 index 00000000000..c2099fc654e --- /dev/null +++ b/src/mongo/db/database_name.h @@ -0,0 +1,151 @@ +/** + * Copyright (C) 2022-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 <algorithm> +#include <boost/algorithm/string/predicate.hpp> +#include <boost/optional.hpp> +#include <string> + +#include "mongo/base/string_data.h" +#include "mongo/db/tenant_id.h" +#include "mongo/logv2/log_attr.h" + +namespace mongo { + +/** + * A DatabaseName is a unique name for database. + * It holds a database name and tenant id, if one exists. In a serverless environment, a tenant id + * is expected to exist so that a database can be uniquely identified. + */ +class DatabaseName { +public: + /** + * Constructs an empty DatabaseName. + */ + DatabaseName() : _tenantId(boost::none), _dbString(""), _tenantDbString(boost::none){}; + + /** + * Constructs a DatabaseName from the given tenantId and database name. + * "dbName" is expected only consist of a db name. It is the caller's responsibility to ensure + * the dbName is a valid db name. + */ + DatabaseName(boost::optional<TenantId> tenantId, StringData dbString) { + _tenantId = tenantId; + _dbString = dbString.toString(); + + _tenantDbString = + _tenantId ? boost::make_optional(_tenantId->toString() + "_" + _dbString) : boost::none; + } + + /** + * Prefer to use the constructor above. + * TODO SERVER-65456 Remove this constructor. + */ + DatabaseName(StringData dbName, boost::optional<TenantId> tenantId = boost::none) + : DatabaseName(tenantId, dbName) {} + + static DatabaseName createSystemTenantDbName(StringData dbString); + + boost::optional<TenantId> tenantId() const { + return _tenantId; + } + + const std::string& db() const { + return _dbString; + } + + const std::string& toString() const { + if (_tenantDbString) + return *_tenantDbString; + + invariant(!_tenantId); + return _dbString; + } + + bool equalCaseInsensitive(const DatabaseName& other) const { + return boost::iequals(toString(), other.toString()); + } + + /** + * Returns -1, 0, or 1 if 'this' is less, equal, or greater than 'other' in + * lexicographical order. + */ + int compare(const DatabaseName& other) const { + return toString().compare(other.toString()); + } + + template <typename H> + friend H AbslHashValue(H h, const DatabaseName& obj) { + return H::combine(std::move(h), obj.toString()); + } + + friend auto logAttrs(const DatabaseName& obj) { + return "databaseName"_attr = obj; + } + +private: + boost::optional<TenantId> _tenantId; + std::string _dbString; + boost::optional<std::string> _tenantDbString; +}; + +inline std::ostream& operator<<(std::ostream& stream, const DatabaseName& tdb) { + return stream << tdb.toString(); +} + +inline StringBuilder& operator<<(StringBuilder& builder, const DatabaseName& tdb) { + return builder << tdb.toString(); +} + +inline bool operator==(const DatabaseName& lhs, const DatabaseName& rhs) { + return lhs.compare(rhs) == 0; +} + +inline bool operator!=(const DatabaseName& lhs, const DatabaseName& rhs) { + return !(lhs == rhs); +} + +inline bool operator<(const DatabaseName& lhs, const DatabaseName& rhs) { + return lhs.compare(rhs) < 0; +} + +inline bool operator>(const DatabaseName& lhs, const DatabaseName& rhs) { + return rhs < lhs; +} + +inline bool operator<=(const DatabaseName& lhs, const DatabaseName& rhs) { + return !(lhs > rhs); +} + +inline bool operator>=(const DatabaseName& lhs, const DatabaseName& rhs) { + return !(lhs < rhs); +} + +} // namespace mongo |