44#include " counters.h"
55
66#include < ydb/core/base/appdata.h>
7+ #include < ydb/core/grpc_services/base/base.h>
78#include < ydb/core/grpc_services/grpc_request_proxy.h>
89#include < ydb/core/mind/address_classification/net_classifier.h>
910#include < ydb/core/mon/mon.h>
@@ -196,22 +197,23 @@ class TClusterDiscoveryServiceActor: public TActorBootstrapped<TClusterDiscovery
196197 switch (ev->GetTypeRewrite ()) {
197198 hFunc (NNetClassifier::TEvNetClassifier::TEvClassifierUpdate, HandleClassifierUpdateWhileIniting);
198199 hFunc (NClusterTracker::TEvClusterTracker::TEvClustersUpdate, HandleClustersUpdateWhileIniting);
199- hFunc (NGRpcService::TEvDiscoverPQClustersRequest , HandleDiscoverPQClustersRequestWhileIniting);
200+ fFunc (NGRpcService::TRpcServices::EvDiscoverPQClusters , HandleDiscoverPQClustersRequestWhileIniting);
200201 hFunc (NMon::TEvHttpInfo, HandleHttpRequest);
201202 hFunc (TEvents::TEvWakeup, UpdateTimedCounters);
202203 }
203204 }
204205
205- void RespondServiceUnavailable (NGRpcService::TEvDiscoverPQClustersRequest::TPtr& ev) {
206+ void RespondServiceUnavailable (NGRpcService::TEvDiscoverPQClustersRequest* ev) {
206207 Counters->DroppedRequestsCount ->Inc ();
207208
208- ev->Get ()-> ReplyWithYdbStatus (Ydb::StatusIds::UNAVAILABLE);
209+ ev->ReplyWithYdbStatus (Ydb::StatusIds::UNAVAILABLE);
209210 }
210211
211- void HandleDiscoverPQClustersRequestWhileIniting (NGRpcService::TEvDiscoverPQClustersRequest::TPtr& ev) {
212+ void HandleDiscoverPQClustersRequestWhileIniting (TAutoPtr<NActors::IEventHandle>& ev) {
213+ ev->DropRewrite ();
212214 Counters->TotalRequestsCount ->Inc ();
213215
214- RespondServiceUnavailable (ev);
216+ RespondServiceUnavailable (ev-> Get <NGRpcService::TEvDiscoverPQClustersRequest>() );
215217 }
216218
217219 void HandleClassifierUpdateWhileWorking (NNetClassifier::TEvNetClassifier::TEvClassifierUpdate::TPtr& ev) {
@@ -222,23 +224,24 @@ class TClusterDiscoveryServiceActor: public TActorBootstrapped<TClusterDiscovery
222224 UpdateClustersList (ev);
223225 }
224226
225- void HandleDiscoverPQClustersRequestWhileWorking (NGRpcService::TEvDiscoverPQClustersRequest::TPtr& ev) {
227+ void HandleDiscoverPQClustersRequestWhileWorking (TAutoPtr<NActors::IEventHandle>& ev) {
228+ ev->DropRewrite ();
226229 Counters->TotalRequestsCount ->Inc ();
227230
228231 if (!IsHealthy ()) {
229- RespondServiceUnavailable (ev);
232+ RespondServiceUnavailable (ev-> Get <NGRpcService::TEvDiscoverPQClustersRequest>() );
230233 return ;
231234 }
232235
233- IActor* actor = NWorker::CreateClusterDiscoveryWorker (ev , DatacenterClassifier, CloudNetworksClassifier, ClustersList, Counters);
236+ IActor* actor = NWorker::CreateClusterDiscoveryWorker (THolder (ev-> Release <NGRpcService::TEvDiscoverPQClustersRequest>(). Release ()) , DatacenterClassifier, CloudNetworksClassifier, ClustersList, Counters);
234237 Register (actor, TMailboxType::HTSwap, AppData (Ctx ())->UserPoolId );
235238 }
236239
237240 STATEFN (Working) {
238241 switch (ev->GetTypeRewrite ()) {
239242 hFunc (NNetClassifier::TEvNetClassifier::TEvClassifierUpdate, HandleClassifierUpdateWhileWorking);
240243 hFunc (NClusterTracker::TEvClusterTracker::TEvClustersUpdate, HandleClustersUpdateWhileWorking);
241- hFunc (NGRpcService::TEvDiscoverPQClustersRequest , HandleDiscoverPQClustersRequestWhileWorking);
244+ fFunc (NGRpcService::TRpcServices::EvDiscoverPQClusters , HandleDiscoverPQClustersRequestWhileWorking);
242245 hFunc (NMon::TEvHttpInfo, HandleHttpRequest);
243246 hFunc (TEvents::TEvWakeup, UpdateTimedCounters);
244247 }
0 commit comments