fix test

ydb-platform · May 15, 2024 · 11aa167 · 11aa167
1 parent bbb9214
commit 11aa167
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/ydb/core/tx/columnshard/data_locks/manager/manager.cpp b/ydb/core/tx/columnshard/data_locks/manager/manager.cpp
@@ -6,10 +6,12 @@ namespace NKikimr::NOlap::NDataLocks {
 std::shared_ptr<TManager::TGuard> TManager::RegisterLock(const std::shared_ptr<ILock>& lock) {
     AFL_VERIFY(lock);
     AFL_VERIFY(ProcessLocks.emplace(lock->GetLockName(), lock).second)("process_id", lock->GetLockName());
+    AFL_DEBUG(NKikimrServices::TX_COLUMNSHARD)("event", "lock")("process_id", lock->GetLockName());
     return std::make_shared<TGuard>(lock->GetLockName(), StopFlag);
 }
 
 void TManager::UnregisterLock(const TString& processId) {
+    AFL_DEBUG(NKikimrServices::TX_COLUMNSHARD)("event", "unlock")("process_id", processId);
     AFL_VERIFY(ProcessLocks.erase(processId))("process_id", processId);
 }
 

diff --git a/ydb/core/tx/columnshard/data_sharing/manager/sessions.cpp b/ydb/core/tx/columnshard/data_sharing/manager/sessions.cpp
@@ -17,6 +17,9 @@ void TSessionsManager::Start(const NColumnShard::TColumnShard& shard) const {
     for (auto&& i : DestSessions) {
         if (!i.second->IsStarted() && i.second->IsConfirmed()) {
             i.second->Start(shard);
+            if (!i.second->GetSourcesInProgressCount()) {
+                i.second->Finish(shard.GetDataLocksManager());
+            }
         }
     }
     NYDBTest::TControllers::GetColumnShardController()->OnAfterSharingSessionsManagerStart(shard);