aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorhor911 <hor911@yandex-team.ru>2022-03-11 16:26:33 +0300
committerhor911 <hor911@yandex-team.ru>2022-03-11 16:26:33 +0300
commitc803d2fd68814e58789e1bb0a74f21db36cdc664 (patch)
tree1072dccafaa596f9bbe68ee593963ac960f899c5
parent5afb64373cc06988e6ff0b8a7df45c547c51970a (diff)
downloadydb-c803d2fd68814e58789e1bb0a74f21db36cdc664.tar.gz
Replace proto map
ref:ad31bb9cf4b54a7b7aa7daddafeedbdf49741c3d
-rw-r--r--ydb/core/yq/libs/config/protos/control_plane_storage.proto11
-rw-r--r--ydb/core/yq/libs/control_plane_storage/ydb_control_plane_storage_queries.cpp14
2 files changed, 16 insertions, 9 deletions
diff --git a/ydb/core/yq/libs/config/protos/control_plane_storage.proto b/ydb/core/yq/libs/config/protos/control_plane_storage.proto
index 086d980ccf1..6ed1169fae2 100644
--- a/ydb/core/yq/libs/config/protos/control_plane_storage.proto
+++ b/ydb/core/yq/libs/config/protos/control_plane_storage.proto
@@ -8,10 +8,15 @@ import "ydb/core/yq/libs/config/protos/storage.proto";
////////////////////////////////////////////////////////////
+message TMapItem {
+ string Key = 1;
+ string Value = 2;
+}
+
message TQueryMapping {
- map<string, string> CloudIdToTenantName = 1;
- map<string, string> ScopeToTenantName = 2;
- repeated string CommonTenantName = 3;
+ repeated TMapItem CloudIdToTenantName = 1;
+ repeated TMapItem ScopeToTenantName = 2;
+ repeated string CommonTenantName = 3;
}
message TControlPlaneStorageConfig {
diff --git a/ydb/core/yq/libs/control_plane_storage/ydb_control_plane_storage_queries.cpp b/ydb/core/yq/libs/control_plane_storage/ydb_control_plane_storage_queries.cpp
index 1c6957443aa..cec2f1b7fb6 100644
--- a/ydb/core/yq/libs/control_plane_storage/ydb_control_plane_storage_queries.cpp
+++ b/ydb/core/yq/libs/control_plane_storage/ydb_control_plane_storage_queries.cpp
@@ -44,16 +44,18 @@ TString TYdbControlPlaneStorageActor::AssignTenantName(const TString& cloudId, c
const auto& mapping = Config.Proto.GetMapping();
if (scope) {
- const auto it = mapping.GetScopeToTenantName().find(scope);
- if (it != mapping.GetScopeToTenantName().end()) {
- return it->second;
+ for (const auto& scopeToTenant: mapping.GetScopeToTenantName()) {
+ if (scopeToTenant.GetKey() == scope) {
+ return scopeToTenant.GetValue();
+ }
}
}
if (cloudId) {
- const auto it = mapping.GetCloudIdToTenantName().find(cloudId);
- if (it != mapping.GetCloudIdToTenantName().end()) {
- return it->second;
+ for (const auto& cloudToTenant: mapping.GetCloudIdToTenantName()) {
+ if (cloudToTenant.GetKey() == cloudId) {
+ return cloudToTenant.GetValue();
+ }
}
}