Add OLAP support with topics in BufferWriteActor (#12686)

Author: dahbka-lis

ydb/core/kqp/runtime/kqp_write_actor.cpp

@@ -1419,7 +1419,6 @@ class TKqpBufferWriteActor :public TActorBootstrapped<TKqpBufferWriteActor>, pub
         State = EState::WRITING;
         Alloc->Release();
         Counters->BufferActorsCount->Inc();
-        TxManager->AddTopicsToShards();
     }
 
     void Bootstrap() {

ydb/core/kqp/session_actor/kqp_session_actor.cpp

@@ -1378,6 +1378,7 @@ class TKqpSessionActor : public TActorBootstrapped<TKqpSessionActor> {
             && !txCtx->BufferActorId
             && (txCtx->HasTableWrite || request.TopicOperations.GetSize() != 0)) {
             txCtx->TxManager->SetTopicOperations(std::move(request.TopicOperations));
+            txCtx->TxManager->AddTopicsToShards();
 
             TKqpBufferWriterSettings settings {
                 .SessionActorId = SelfId(),
@@ -1388,6 +1389,9 @@ class TKqpSessionActor : public TActorBootstrapped<TKqpSessionActor> {
             };
             auto* actor = CreateKqpBufferWriterActor(std::move(settings));
             txCtx->BufferActorId = RegisterWithSameMailbox(actor);
+        } else if (Settings.TableService.GetEnableOltpSink() && txCtx->BufferActorId) {
+            txCtx->TxManager->SetTopicOperations(std::move(request.TopicOperations));
+            txCtx->TxManager->AddTopicsToShards();
         }
 
         auto executerActor = CreateKqpExecuter(std::move(request), Settings.Database,

ydb/core/testlib/test_client.h

@@ -231,6 +231,18 @@ namespace Tests {
             AppConfig->MutableTableServiceConfig()->SetEnableOltpSink(withOltpSink);
             return *this;
         }
+        TServerSettings& SetEnableOlapSink(bool withOlapSink) {
+            AppConfig->MutableTableServiceConfig()->SetEnableOlapSink(withOlapSink);
+            return *this;
+        }
+        TServerSettings& SetEnableHtapTx(bool withHtapTx) {
+            AppConfig->MutableTableServiceConfig()->SetEnableHtapTx(withHtapTx);
+            return *this;
+        }
+        TServerSettings& SetAllowOlapDataQuery(bool withAllowOlapDataQuery) {
+            AppConfig->MutableTableServiceConfig()->SetAllowOlapDataQuery(withAllowOlapDataQuery);
+            return *this;
+        }
 
 
         // Add additional grpc services