diff options
author | Anton Samokhvalov <pg83@yandex.ru> | 2022-02-10 16:45:17 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:45:17 +0300 |
commit | d3a398281c6fd1d3672036cb2d63f842d2cb28c5 (patch) | |
tree | dd4bd3ca0f36b817e96812825ffaf10d645803f2 /library/cpp/actors/interconnect/ut/lib/test_actors.h | |
parent | 72cb13b4aff9bc9cf22e49251bc8fd143f82538f (diff) | |
download | ydb-d3a398281c6fd1d3672036cb2d63f842d2cb28c5.tar.gz |
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/actors/interconnect/ut/lib/test_actors.h')
-rw-r--r-- | library/cpp/actors/interconnect/ut/lib/test_actors.h | 92 |
1 files changed, 46 insertions, 46 deletions
diff --git a/library/cpp/actors/interconnect/ut/lib/test_actors.h b/library/cpp/actors/interconnect/ut/lib/test_actors.h index bcb3504ebe..7591200471 100644 --- a/library/cpp/actors/interconnect/ut/lib/test_actors.h +++ b/library/cpp/actors/interconnect/ut/lib/test_actors.h @@ -1,56 +1,56 @@ #pragma once namespace NActors { - class TSenderBaseActor: public TActorBootstrapped<TSenderBaseActor> { - protected: + class TSenderBaseActor: public TActorBootstrapped<TSenderBaseActor> { + protected: const TActorId RecipientActorId; - const ui32 Preload; - ui64 SequenceNumber = 0; - ui32 InFlySize = 0; + const ui32 Preload; + ui64 SequenceNumber = 0; + ui32 InFlySize = 0; - public: + public: TSenderBaseActor(const TActorId& recipientActorId, ui32 preload = 1) - : RecipientActorId(recipientActorId) - , Preload(preload) - { - } + : RecipientActorId(recipientActorId) + , Preload(preload) + { + } - virtual ~TSenderBaseActor() { - } + virtual ~TSenderBaseActor() { + } - virtual void Bootstrap(const TActorContext& ctx) { - Become(&TSenderBaseActor::StateFunc); + virtual void Bootstrap(const TActorContext& ctx) { + Become(&TSenderBaseActor::StateFunc); ctx.Send(ctx.ExecutorThread.ActorSystem->InterconnectProxy(RecipientActorId.NodeId()), new TEvInterconnect::TEvConnectNode); - } + } - virtual void SendMessagesIfPossible(const TActorContext& ctx) { - while (InFlySize < Preload) { - SendMessage(ctx); - } + virtual void SendMessagesIfPossible(const TActorContext& ctx) { + while (InFlySize < Preload) { + SendMessage(ctx); + } } - virtual void SendMessage(const TActorContext& /*ctx*/) { - ++SequenceNumber; - } + virtual void SendMessage(const TActorContext& /*ctx*/) { + ++SequenceNumber; + } - virtual void Handle(TEvents::TEvUndelivered::TPtr& /*ev*/, const TActorContext& ctx) { - SendMessage(ctx); - } + virtual void Handle(TEvents::TEvUndelivered::TPtr& /*ev*/, const TActorContext& ctx) { + SendMessage(ctx); + } - virtual void Handle(TEvTestResponse::TPtr& /*ev*/, const TActorContext& ctx) { - SendMessagesIfPossible(ctx); - } + virtual void Handle(TEvTestResponse::TPtr& /*ev*/, const TActorContext& ctx) { + SendMessagesIfPossible(ctx); + } void Handle(TEvInterconnect::TEvNodeConnected::TPtr& /*ev*/, const TActorContext& ctx) { SendMessagesIfPossible(ctx); - } + } - void Handle(TEvInterconnect::TEvNodeDisconnected::TPtr& /*ev*/, const TActorContext& /*ctx*/) { - } + void Handle(TEvInterconnect::TEvNodeDisconnected::TPtr& /*ev*/, const TActorContext& /*ctx*/) { + } - virtual void Handle(TEvents::TEvPoisonPill::TPtr& /*ev*/, const TActorContext& ctx) { - Die(ctx); - } + virtual void Handle(TEvents::TEvPoisonPill::TPtr& /*ev*/, const TActorContext& ctx) { + Die(ctx); + } virtual STRICT_STFUNC(StateFunc, HFunc(TEvTestResponse, Handle) @@ -59,25 +59,25 @@ namespace NActors { HFunc(TEvInterconnect::TEvNodeConnected, Handle) HFunc(TEvInterconnect::TEvNodeDisconnected, Handle) ) - }; + }; - class TReceiverBaseActor: public TActor<TReceiverBaseActor> { - protected: - ui64 ReceivedCount = 0; + class TReceiverBaseActor: public TActor<TReceiverBaseActor> { + protected: + ui64 ReceivedCount = 0; - public: - TReceiverBaseActor() - : TActor(&TReceiverBaseActor::StateFunc) - { - } + public: + TReceiverBaseActor() + : TActor(&TReceiverBaseActor::StateFunc) + { + } - virtual ~TReceiverBaseActor() { - } + virtual ~TReceiverBaseActor() { + } virtual STRICT_STFUNC(StateFunc, HFunc(TEvTest, Handle) ) virtual void Handle(TEvTest::TPtr& /*ev*/, const TActorContext& /*ctx*/) {} - }; + }; } |