diff options
author | hor911 <hor911@yandex-team.ru> | 2022-03-11 16:26:33 +0300 |
---|---|---|
committer | hor911 <hor911@yandex-team.ru> | 2022-03-11 16:26:33 +0300 |
commit | c803d2fd68814e58789e1bb0a74f21db36cdc664 (patch) | |
tree | 1072dccafaa596f9bbe68ee593963ac960f899c5 | |
parent | 5afb64373cc06988e6ff0b8a7df45c547c51970a (diff) | |
download | ydb-c803d2fd68814e58789e1bb0a74f21db36cdc664.tar.gz |
Replace proto map
ref:ad31bb9cf4b54a7b7aa7daddafeedbdf49741c3d
-rw-r--r-- | ydb/core/yq/libs/config/protos/control_plane_storage.proto | 11 | ||||
-rw-r--r-- | ydb/core/yq/libs/control_plane_storage/ydb_control_plane_storage_queries.cpp | 14 |
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(); + } } } |