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
|
#include "interconnect.h"
#include "interconnect_impl.h"
#include "interconnect_address.h"
#include "events_local.h"
#include <library/cpp/actors/core/hfunc.h>
#include <library/cpp/actors/memory_log/memlog.h>
namespace NActors {
template<typename TDerived>
class TInterconnectNameserverBase : public TActor<TDerived> {
protected:
const TMap<ui32, TTableNameserverSetup::TNodeInfo>& NodeTable;
TInterconnectNameserverBase(void (TDerived::*func)(TAutoPtr<IEventHandle>& ev, const TActorContext& ctx)
, const TMap<ui32, TTableNameserverSetup::TNodeInfo>& nodeTable)
: TActor<TDerived>(func)
, NodeTable(nodeTable)
{
}
public:
void HandleMissedNodeId(TEvInterconnect::TEvResolveNode::TPtr& ev,
const TActorContext& ctx,
const TInstant&) {
auto reply = new TEvLocalNodeInfo;
reply->NodeId = ev->Get()->Record.GetNodeId();
ctx.Send(ev->Sender, reply);
}
void Handle(TEvInterconnect::TEvResolveNode::TPtr& ev,
const TActorContext& ctx) {
const TEvInterconnect::TEvResolveNode* request = ev->Get();
auto& record = request->Record;
const ui32 nodeId = record.GetNodeId();
const TInstant deadline = record.HasDeadline() ? TInstant::FromValue(record.GetDeadline()) : TInstant::Max();
auto it = NodeTable.find(nodeId);
if (it == NodeTable.end()) {
static_cast<TDerived*>(this)->HandleMissedNodeId(ev, ctx, deadline);
} else {
IActor::RegisterWithSameMailbox(
CreateResolveActor(nodeId, it->second, ev->Sender, this->SelfId(), deadline));
}
}
void Handle(TEvResolveAddress::TPtr& ev,
const TActorContext&) {
const TEvResolveAddress* request = ev->Get();
IActor::RegisterWithSameMailbox(
CreateResolveActor(request->Address, request->Port, ev->Sender, this->SelfId(), TInstant::Max()));
}
void Handle(TEvInterconnect::TEvListNodes::TPtr& ev,
const TActorContext& ctx) {
THolder<TEvInterconnect::TEvNodesInfo>
reply(new TEvInterconnect::TEvNodesInfo());
reply->Nodes.reserve(NodeTable.size());
for (const auto& pr : NodeTable) {
reply->Nodes.emplace_back(pr.first,
pr.second.Address, pr.second.Host, pr.second.ResolveHost,
pr.second.Port, pr.second.Location);
}
ctx.Send(ev->Sender, reply.Release());
}
void Handle(TEvInterconnect::TEvGetNode::TPtr& ev,
const TActorContext& ctx) {
ui32 nodeId = ev->Get()->NodeId;
THolder<TEvInterconnect::TEvNodeInfo>
reply(new TEvInterconnect::TEvNodeInfo(nodeId));
auto it = NodeTable.find(nodeId);
if (it != NodeTable.end()) {
reply->Node = MakeHolder<TEvInterconnect::TNodeInfo>(it->first, it->second.Address,
it->second.Host, it->second.ResolveHost,
it->second.Port, it->second.Location);
}
ctx.Send(ev->Sender, reply.Release());
}
};
}
|
