1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#include "events.h"
#include "actorsystem.h"
namespace NActors {
TString TEvents::TEvUndelivered::ToStringHeader() const {
return "TSystem::Undelivered";
}
bool TEvents::TEvUndelivered::SerializeToArcadiaStream(TChunkSerializer *serializer) const {
Y_VERIFY(!Unsure); // these are local-only events generated by Interconnect
return serializer->WriteString(&Data);
}
void TEvents::TEvUndelivered::Out(IOutputStream& o, EReason x) {
switch (x) {
case ReasonActorUnknown:
o << "ActorUnknown";
break;
case Disconnected:
o << "Disconnected";
break;
default:
o << "Undefined";
break;
}
}
bool TEvents::TEvUndelivered::IsSerializable() const {
return true;
}
IEventBase* TEvents::TEvUndelivered::Load(TEventSerializedData* bufs) {
TString str = bufs->GetString();
Y_VERIFY(str.size() == (sizeof(ui32) + sizeof(ui32)));
const char* p = str.data();
const ui64 sourceType = ReadUnaligned<ui32>(p + 0);
const ui64 reason = ReadUnaligned<ui32>(p + 4);
return new TEvUndelivered(sourceType, reason);
}
TAutoPtr<IEventHandle> IEventHandle::ForwardOnNondelivery(TAutoPtr<IEventHandle>& ev, ui32 reason, bool unsure) {
std::unique_ptr<IEventHandle> tev(ev.Release());
TAutoPtr<IEventHandle> fw = ForwardOnNondelivery(tev, reason, unsure);
if (tev) {
// we don't want to delete original event handle here
ev = tev.release();
}
return fw;
}
TAutoPtr<IEventHandle> IEventHandle::ForwardOnNondelivery(std::unique_ptr<IEventHandle>& ev, ui32 reason, bool unsure) {
if (ev->IsEventFat()) {
std::unique_ptr<IEventHandleFat> evf(IEventHandleFat::GetFat(ev.release()));
std::unique_ptr<IEventHandle> fw = IEventHandleFat::ForwardOnNondelivery(evf, reason, unsure);
if (evf) {
ev = std::unique_ptr<IEventHandle>(evf.release());
}
return fw.release();
}
if (ev->IsEventLight()) {
std::unique_ptr<IEventHandleLight> evl(IEventHandleLight::GetLight(ev.release()));
std::unique_ptr<IEventHandle> fw = IEventHandleLight::ForwardOnNondelivery(evl, reason, unsure);
if (evl) {
ev = std::unique_ptr<IEventHandle>(evl.release());
}
return fw.release();
}
return {};
}
std::unique_ptr<IEventHandle> IEventHandleFat::ForwardOnNondelivery(std::unique_ptr<IEventHandleFat>& ev, ui32 reason, bool unsure) {
if (ev->ForwardOnNondeliveryFlag) {
const ui32 updatedFlags = ev->Flags & ~(FlagForwardOnNondelivery | FlagSubscribeOnSession);
const TActorId recp = ev->OnNondeliveryHolder ? ev->OnNondeliveryHolder->Recipient : TActorId();
if (ev->Event)
return std::unique_ptr<IEventHandle>(new IEventHandleFat(recp, ev->Sender, ev->Event.Release(), updatedFlags, ev->Cookie, &ev->Recipient, std::move(ev->TraceId)));
else
return std::unique_ptr<IEventHandle>(new IEventHandleFat(ev->Type, updatedFlags, recp, ev->Sender, ev->Buffer, ev->Cookie, &ev->Recipient, std::move(ev->TraceId)));
}
if (ev->TrackDelivery) {
const ui32 updatedFlags = ev->Flags & ~(FlagTrackDelivery | FlagSubscribeOnSession | FlagGenerateUnsureUndelivered);
return std::unique_ptr<IEventHandle>(new IEventHandleFat(ev->Sender, ev->Recipient, new TEvents::TEvUndelivered(ev->Type, reason, unsure), updatedFlags,
ev->Cookie, nullptr, std::move(ev->TraceId)));
}
return {};
}
std::unique_ptr<IEventHandle> IEventHandleLight::ForwardOnNondelivery(std::unique_ptr<IEventHandleLight>& ev, ui32 reason, bool unsure) {
if (ev->ForwardOnNondeliveryFlag) {
ev->ForwardOnNondeliveryFlag = false;
ev->SubscribeOnSession = false;
auto recpt = ev->Recipient;
ev->Recipient = ev->OnNondeliveryHolder ? ev->OnNondeliveryHolder->Recipient : TActorId();
ev->OnNondeliveryHolder = MakeHolder<TOnNondelivery>(recpt);
return std::unique_ptr<IEventHandle>(ev.release());
}
if (ev->TrackDelivery) {
const ui32 updatedFlags = ev->Flags & ~(FlagTrackDelivery | FlagSubscribeOnSession | FlagGenerateUnsureUndelivered);
return std::unique_ptr<IEventHandle>(new IEventHandleFat(ev->Sender, ev->Recipient, new TEvents::TEvUndelivered(ev->Type, reason, unsure), updatedFlags,
ev->Cookie, nullptr, std::move(ev->TraceId)));
}
return {};
}
}
|
