Use UserPool for pqv0 actors (ydb-platform#8316)

GrigoriyPA · Aug 28, 2024 · d9565af · d9565af
1 parent 01e094b
commit d9565af
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/ydb/services/deprecated/persqueue_v0/grpc_pq_read.cpp b/ydb/services/deprecated/persqueue_v0/grpc_pq_read.cpp
@@ -154,10 +154,10 @@ void TPQReadService::TSession::SendEvent(IEventBase* ev) {
 void TPQReadService::TSession::CreateActor(std::unique_ptr<NPersQueue::TTopicsListController>&& topicsHandler) {
     auto classifier = Proxy->GetClassifier();
 
-    ActorId = Proxy->ActorSystem->Register(
-        new TReadSessionActor(this, *topicsHandler, Cookie, SchemeCache, NewSchemeCache, Counters,
-                                    classifier ? classifier->ClassifyAddress(GetPeerName())
-                                                         : "unknown"));
+    auto* actor = new TReadSessionActor(this, *topicsHandler, Cookie, SchemeCache, NewSchemeCache, Counters,
+                                    classifier ? classifier->ClassifyAddress(GetPeerName()) : "unknown");
+    ui32 poolId = Proxy->ActorSystem->AppData<::NKikimr::TAppData>()->UserPoolId;
+    ActorId = Proxy->ActorSystem->Register(actor, TMailboxType::HTSwap, poolId);
 }
 
 

diff --git a/ydb/services/deprecated/persqueue_v0/grpc_pq_write.cpp b/ydb/services/deprecated/persqueue_v0/grpc_pq_write.cpp
@@ -113,11 +113,10 @@ bool TPQWriteServiceImpl::TSession::CreateActor(const TString &localCluster) {
     }
 
     auto classifier = Proxy->GetClassifier();
-    ActorId = Proxy->ActorSystem->Register(
-        new TWriteSessionActor(this, Cookie, SchemeCache, Counters, localCluster,
-                                        classifier ? classifier->ClassifyAddress(GetPeerName())
-                                                   : "unknown"), TMailboxType::Simple, 0
-    );
+    auto* actor = new TWriteSessionActor(this, Cookie, SchemeCache, Counters, localCluster,
+                    classifier ? classifier->ClassifyAddress(GetPeerName()) : "unknown");
+    ui32 poolId = Proxy->ActorSystem->AppData<::NKikimr::TAppData>()->UserPoolId;
+    ActorId = Proxy->ActorSystem->Register(actor, TMailboxType::HTSwap, poolId);
     return true;
 }