Skip to content

Commit 272d2eb

Browse files
Optimize the writing session creation time (#19807)
1 parent e65821c commit 272d2eb

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

ydb/core/persqueue/writer/writer.cpp

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,10 @@ class TPartitionWriter : public TActorBootstrapped<TPartitionWriter>, private TR
254254
/// GetWriteId
255255

256256
void GetWriteId(const TActorContext& ctx) {
257+
INFO("Start of a request to KQP for a WriteId. " <<
258+
"SessionId: " << Opts.SessionId <<
259+
" TxId: " << Opts.TxId);
260+
257261
auto ev = MakeWriteIdRequest();
258262
ctx.Send(NKqp::MakeKqpProxyID(ctx.SelfID.NodeId()), ev.Release());
259263
Become(&TThis::StateGetWriteId);
@@ -270,6 +274,10 @@ class TPartitionWriter : public TActorBootstrapped<TPartitionWriter>, private TR
270274
}
271275

272276
void HandleWriteId(NKqp::TEvKqp::TEvQueryResponse::TPtr& ev, const TActorContext& ctx) {
277+
INFO("End of the request to KQP for the WriteId. " <<
278+
"SessionId: " << Opts.SessionId <<
279+
" TxId: " << Opts.TxId);
280+
273281
auto& record = ev->Get()->Record;
274282
switch (record.GetYdbStatus()) {
275283
case Ydb::StatusIds::SUCCESS:
@@ -404,11 +412,19 @@ class TPartitionWriter : public TActorBootstrapped<TPartitionWriter>, private TR
404412
Y_ABORT_UNLESS(HasWriteId());
405413
Y_ABORT_UNLESS(HasSupportivePartitionId());
406414

415+
INFO("Start of a request to KQP to save PartitionId. " <<
416+
"SessionId: " << Opts.SessionId <<
417+
" TxId: " << Opts.TxId);
418+
407419
auto ev = MakeWriteIdRequest();
408420
ctx.Send(NKqp::MakeKqpProxyID(ctx.SelfID.NodeId()), ev.Release());
409421
}
410422

411423
void HandlePartitionIdSaved(NKqp::TEvKqp::TEvQueryResponse::TPtr& ev, const TActorContext&) {
424+
INFO("End of a request to KQP to save PartitionId. " <<
425+
"SessionId: " << Opts.SessionId <<
426+
" TxId: " << Opts.TxId);
427+
412428
auto& record = ev->Get()->Record;
413429
switch (record.GetYdbStatus()) {
414430
case Ydb::StatusIds::SUCCESS:
@@ -968,7 +984,10 @@ class TPartitionWriter : public TActorBootstrapped<TPartitionWriter>, private TR
968984
using IRetryState = IRetryPolicy::IRetryState;
969985

970986
static IRetryPolicy::TPtr GetRetryPolicy() {
971-
return IRetryPolicy::GetExponentialBackoffPolicy(Retryable);
987+
return IRetryPolicy::GetExponentialBackoffPolicy(Retryable,
988+
TDuration::MilliSeconds(10),
989+
TDuration::MilliSeconds(10),
990+
TDuration::MilliSeconds(100));
972991
};
973992

974993
static ERetryErrorClass Retryable(Ydb::StatusIds::StatusCode code) {

0 commit comments

Comments
 (0)