Skip to content

Commit 0475ae0

Browse files
authored
TCredentialsKey -> TConnectionParams (#7062)
1 parent 6f10c09 commit 0475ae0

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

ydb/core/tx/replication/service/service.cpp

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,8 @@ class TSessionInfo {
125125

126126
}; // TSessionInfo
127127

128-
struct TCredentialsKey: std::tuple<TString, TString, bool, TString> {
129-
explicit TCredentialsKey(const TString& endpoint, const TString& database, bool ssl, const TString& user)
128+
struct TConnectionParams: std::tuple<TString, TString, bool, TString> {
129+
explicit TConnectionParams(const TString& endpoint, const TString& database, bool ssl, const TString& user)
130130
: std::tuple<TString, TString, bool, TString>(endpoint, database, ssl, user)
131131
{
132132
}
@@ -143,27 +143,27 @@ struct TCredentialsKey: std::tuple<TString, TString, bool, TString> {
143143
return std::get<2>(*this);
144144
}
145145

146-
static TCredentialsKey FromParams(const NKikimrReplication::TConnectionParams& params) {
146+
static TConnectionParams FromProto(const NKikimrReplication::TConnectionParams& params) {
147147
const auto& endpoint = params.GetEndpoint();
148148
const auto& database = params.GetDatabase();
149149
const bool ssl = params.GetEnableSsl();
150150

151151
switch (params.GetCredentialsCase()) {
152152
case NKikimrReplication::TConnectionParams::kStaticCredentials:
153-
return TCredentialsKey(endpoint, database, ssl, params.GetStaticCredentials().GetUser());
153+
return TConnectionParams(endpoint, database, ssl, params.GetStaticCredentials().GetUser());
154154
case NKikimrReplication::TConnectionParams::kOAuthToken:
155-
return TCredentialsKey(endpoint, database, ssl, params.GetOAuthToken().GetToken());
155+
return TConnectionParams(endpoint, database, ssl, params.GetOAuthToken().GetToken());
156156
default:
157157
Y_ABORT("Unexpected credentials");
158158
}
159159
}
160160

161-
}; // TCredentialsKey
161+
}; // TConnectionParams
162162

163163
} // NKikimr::NReplication::NService
164164

165165
template <>
166-
struct THash<NKikimr::NReplication::NService::TCredentialsKey> : THash<std::tuple<TString, TString, bool, TString>> {};
166+
struct THash<NKikimr::NReplication::NService::TConnectionParams> : THash<std::tuple<TString, TString, bool, TString>> {};
167167

168168
namespace NKikimr::NReplication {
169169

@@ -217,11 +217,11 @@ class TReplicationService: public TActorBootstrapped<TReplicationService> {
217217
}
218218

219219
template <typename... Args>
220-
const TActorId& GetOrCreateYdbProxy(TCredentialsKey&& key, Args&&... args) {
221-
auto it = YdbProxies.find(key);
220+
const TActorId& GetOrCreateYdbProxy(TConnectionParams&& params, Args&&... args) {
221+
auto it = YdbProxies.find(params);
222222
if (it == YdbProxies.end()) {
223-
auto ydbProxy = Register(CreateYdbProxy(key.Endpoint(), key.Database(), key.EnableSsl(), std::forward<Args>(args)...));
224-
auto res = YdbProxies.emplace(std::move(key), std::move(ydbProxy));
223+
auto ydbProxy = Register(CreateYdbProxy(params.Endpoint(), params.Database(), params.EnableSsl(), std::forward<Args>(args)...));
224+
auto res = YdbProxies.emplace(std::move(params), std::move(ydbProxy));
225225
Y_ABORT_UNLESS(res.second);
226226
it = res.first;
227227
}
@@ -234,10 +234,10 @@ class TReplicationService: public TActorBootstrapped<TReplicationService> {
234234
const auto& params = settings.GetConnectionParams();
235235
switch (params.GetCredentialsCase()) {
236236
case NKikimrReplication::TConnectionParams::kStaticCredentials:
237-
ydbProxy = GetOrCreateYdbProxy(TCredentialsKey::FromParams(params), params.GetStaticCredentials());
237+
ydbProxy = GetOrCreateYdbProxy(TConnectionParams::FromProto(params), params.GetStaticCredentials());
238238
break;
239239
case NKikimrReplication::TConnectionParams::kOAuthToken:
240-
ydbProxy = GetOrCreateYdbProxy(TCredentialsKey::FromParams(params), params.GetOAuthToken().GetToken());
240+
ydbProxy = GetOrCreateYdbProxy(TConnectionParams::FromProto(params), params.GetOAuthToken().GetToken());
241241
break;
242242
default:
243243
Y_ABORT("Unexpected credentials");
@@ -438,7 +438,7 @@ class TReplicationService: public TActorBootstrapped<TReplicationService> {
438438
mutable TMaybe<TString> LogPrefix;
439439
TActorId BoardPublisher;
440440
THashMap<ui64, TSessionInfo> Sessions;
441-
THashMap<TCredentialsKey, TActorId> YdbProxies;
441+
THashMap<TConnectionParams, TActorId> YdbProxies;
442442
THashMap<TActorId, ui64> WorkerActorIdToSession;
443443

444444
}; // TReplicationService

0 commit comments

Comments
 (0)