profile memory in transactions (#3816)

Author: ivanmorozov333

ydb/core/tx/columnshard/blobs_action/transaction/tx_draft.cpp

@@ -3,6 +3,7 @@
 namespace NKikimr::NColumnShard {
 
 bool TTxWriteDraft::Execute(TTransactionContext& txc, const TActorContext& /*ctx*/) {
+    TMemoryProfileGuard mpg("TTxWriteDraft::Execute");
     NOlap::TBlobManagerDb blobManagerDb(txc.DB);
     for (auto&& action : WriteController->GetBlobActions()) {
         action.second->OnExecuteTxBeforeWrite(*Self, blobManagerDb);
@@ -11,6 +12,7 @@ bool TTxWriteDraft::Execute(TTransactionContext& txc, const TActorContext& /*ctx
 }
 
 void TTxWriteDraft::Complete(const TActorContext& ctx) {
+    TMemoryProfileGuard mpg("TTxWriteDraft::Complete");
     Completed = true;
     for (auto&& action : WriteController->GetBlobActions()) {
         action.second->OnCompleteTxBeforeWrite(*Self);

ydb/core/tx/columnshard/blobs_action/transaction/tx_gc_indexed.cpp

@@ -2,12 +2,14 @@
 
 namespace NKikimr::NColumnShard {
 bool TTxGarbageCollectionFinished::Execute(TTransactionContext& txc, const TActorContext& /*ctx*/) {
+    TMemoryProfileGuard mpg("TTxGarbageCollectionFinished::Execute");
     AFL_DEBUG(NKikimrServices::TX_COLUMNSHARD)("tx", "TxGarbageCollectionFinished")("event", "execute");
     NOlap::TBlobManagerDb blobManagerDb(txc.DB);
     Action->OnExecuteTxAfterCleaning(*Self, blobManagerDb);
     return true;
 }
 void TTxGarbageCollectionFinished::Complete(const TActorContext& /*ctx*/) {
+    TMemoryProfileGuard mpg("TTxGarbageCollectionFinished::Complete");
     AFL_DEBUG(NKikimrServices::TX_COLUMNSHARD)("tx", "TxGarbageCollectionFinished")("event", "complete");
     Action->OnCompleteTxAfterCleaning(*Self, Action);
 }

ydb/core/tx/columnshard/blobs_action/transaction/tx_gc_insert_table.cpp

@@ -4,6 +4,7 @@
 namespace NKikimr::NColumnShard {
 
 bool TTxInsertTableCleanup::Execute(TTransactionContext& txc, const TActorContext& /*ctx*/) {
+    TMemoryProfileGuard mpg("TTxInsertTableCleanup::Execute");
     TBlobGroupSelector dsGroupSelector(Self->Info());
     NOlap::TDbWrapper dbTable(txc.DB, &dsGroupSelector);
     NIceDb::TNiceDb db(txc.DB);
@@ -21,6 +22,7 @@ bool TTxInsertTableCleanup::Execute(TTransactionContext& txc, const TActorContex
     return true;
 }
 void TTxInsertTableCleanup::Complete(const TActorContext& /*ctx*/) {
+    TMemoryProfileGuard mpg("TTxInsertTableCleanup::Complete");
     auto allAborted = Self->InsertTable->GetAborted();
     for (auto& [abortedWriteId, abortedData] : allAborted) {
         Self->InsertTable->EraseAbortedOnComplete(abortedData);

ydb/core/tx/columnshard/blobs_action/transaction/tx_remove_blobs.cpp

@@ -4,13 +4,15 @@
 namespace NKikimr::NColumnShard {
 
 bool TTxRemoveSharedBlobs::Execute(TTransactionContext& txc, const TActorContext&) {
+    TMemoryProfileGuard mpg("TTxRemoveSharedBlobs::Execute");
     NActors::TLogContextGuard logGuard = NActors::TLogContextBuilder::Build(NKikimrServices::TX_COLUMNSHARD)("tablet_id", Self->TabletID())("tx_state", "execute");
     NOlap::TBlobManagerDb blobManagerDb(txc.DB);
     RemoveAction->OnExecuteTxAfterRemoving(*Self, blobManagerDb, true);
     return true;
 }
 
 void TTxRemoveSharedBlobs::Complete(const TActorContext& ctx) {
+    TMemoryProfileGuard mpg("TTxRemoveSharedBlobs::Complete");
     NActors::TLogContextGuard logGuard = NActors::TLogContextBuilder::Build(NKikimrServices::TX_COLUMNSHARD)("tablet_id", Self->TabletID())("tx_state", "complete");
     RemoveAction->OnCompleteTxAfterRemoving(*Self, true);
 

ydb/core/tx/columnshard/engines/reader/transaction/tx_scan.cpp

@@ -117,6 +117,7 @@ static bool FillPredicatesFromRange(TReadDescription& read, const ::NKikimrTx::T
 }
 
 bool TTxScan::Execute(TTransactionContext& /*txc*/, const TActorContext& /*ctx*/) {
+    TMemoryProfileGuard mpg("TTxScan::Execute");
     auto& record = Ev->Get()->Record;
     TSnapshot snapshot(record.GetSnapshot().GetStep(), record.GetSnapshot().GetTxId());
     const auto scanId = record.GetScanId();
@@ -204,7 +205,7 @@ struct TContainerPrinter {
 };
 
 void TTxScan::Complete(const TActorContext& ctx) {
-
+    TMemoryProfileGuard mpg("TTxScan::Complete");
     auto& request = Ev->Get()->Record;
     auto scanComputeActor = Ev->Sender;
     const auto& snapshot = request.GetSnapshot();