1 #include "hmbdc/Copyright.hpp" 3 #include "hmbdc/tips/rmcast/Transport.hpp" 4 #include "hmbdc/tips/rmcast/McRecvTransport.hpp" 5 #include "hmbdc/tips/rmcast/BackupRecvSession.hpp" 6 #include "hmbdc/tips/rmcast/DefaultUserConfig.hpp" 7 #include "hmbdc/tips/reliable/AttBufferAdaptor.hpp" 8 #include "hmbdc/app/Logger.hpp" 9 #include "hmbdc/comm/inet/Hash.hpp" 11 #include <boost/bind.hpp> 12 #include <boost/lexical_cast.hpp> 13 #include <boost/unordered_map.hpp> 17 #include <type_traits> 20 namespace hmbdc {
namespace tips {
namespace rmcast {
34 namespace recvtransportengine_detail {
48 template <
typename OutputBuffer,
typename AttachmentAllocator>
66 , config_.getExt<uint16_t>(
"cmdBufferSizePower2"))
67 , outputBuffer_(outputBuffer)
68 , myBackupIp_(inet_addr(hmbdc::comm::inet::getLocalIpMatchMask(
69 config_.getExt<
string>(
"tcpIfaceAddr") ==
string(
"ifaceAddr")
70 ?config_.getExt<
string>(
"ifaceAddr"):config_.getExt<
string>(
"tcpIfaceAddr")
72 , mcRecvTransport_(config_
76 , loopback_(config_.getExt<
bool>(
"loopback")) {
77 subscriptions_.addAll<std::tuple<app::StartMemorySegTrain, app::MemorySeg>>();
78 if (!config_.getExt<
bool>(
"allowRecvWithinProcess")) {
88 mcRecvTransport_.start();
91 void runOnce() HMBDC_RESTRICT {
92 utils::EpollTask::instance().poll();
94 auto n = cmdBuffer_.peek(begin, end);
99 cmdBuffer_.wasteAfterPeek(begin, n);
100 mcRecvTransport_.runOnce();
101 for (
auto it = recvSessions_.begin(); it != recvSessions_.end();) {
102 if (hmbdc_unlikely(!it->second->runOnce())) {
105 ep2SessionDict_.erase(it->second->sendFrom);
106 recvSessions_.erase(it++);
117 auto ip = inet_addr(t.ip);
119 if (ip == myBackupIp_
123 if (t.srcPid == myPid_ && ip == myBackupIp_)
return;
124 auto key = make_pair(ip, t.port);
125 if (recvSessions_.find(key) == recvSessions_.end()) {
126 for (
auto i = 0u; i < t.typeTagCountContained; ++i) {
127 if (subscriptions_.check(t.typeTags[i])) {
129 typename Session::ptr sess {
138 sess->start(ip, t.port);
139 recvSessions_[key] = sess;
140 ep2SessionDict_[t.sendFrom] = sess;
141 schedule(SysTime::now(), *sess);
142 }
catch (std::exception
const& e) {
143 HMBDC_LOG_C(e.what());
156 return recvSessions_.size();
159 template <MessageTupleC Messages,
typename CcNode>
160 void subscribeFor(CcNode
const& node, uint16_t mod, uint16_t res) {
161 subscriptions_.addSubsFor<Messages>(node, mod, res);
168 OutputBuffer outputBuffer_;
169 in_addr_t myBackupIp_;
173 = boost::unordered_map<sockaddr_in,
typename Session::ptr
174 , endpointhash, endpointequal>;
175 Ep2SessionDict ep2SessionDict_;
176 boost::unordered_map<pair<uint64_t, uint16_t>
177 ,
typename Session::ptr> recvSessions_;
183 template <
typename OutputBuffer,
typename AttachmentAllocator>
186 ,
Client<RecvTransportEngine<OutputBuffer, AttachmentAllocator>> {
192 , OutputBuffer& outputBuffer)
193 :
Impl(cfg, outputBuffer) {
198 if (runLock_.try_lock()) {
199 this->checkTimers(time::SysTime::now());
222 HMBDC_LOG_C(e.what());
225 using EngineTransport::hmbdcName;
227 std::tuple<char const*, int> schedSpec()
const {
228 return std::make_tuple(this->schedPolicy_.c_str(), this->schedPriority_);
232 this->runLock_.unlock();
240 template <
typename OutputBuffer,
typename AttachmentAllocator>
241 using RecvTransportEngine = recvtransportengine_detail::RecvTransportEngine<OutputBuffer
242 , AttachmentAllocator>;
Definition: MonoLockFreeBuffer.hpp:16
class to hold an hmbdc configuration
Definition: Config.hpp:45
void setAdditionalFallbackConfig(Config const &c)
set additional defaults
Definition: Config.hpp:154
Definition: Messages.hpp:90
impl class
Definition: McRecvTransport.hpp:29
Definition: TypedString.hpp:84
RecvTransportImpl(Config const &cfg, OutputBuffer &outputBuffer)
ctor
Definition: RecvTransportEngine.hpp:63
impl class
Definition: RecvTransportEngine.hpp:49
Definition: Timers.hpp:70
Config & resetSection(char const *section, bool sectionExists=true)
change section name
Definition: Config.hpp:177
void stoppedCb(std::exception const &e) override
should not happen ever unless an exception thrown
Definition: RecvTransportEngine.hpp:221
Definition: Transport.hpp:42
Definition: RecvTransportEngine.hpp:184
Definition: MessageDispacher.hpp:184
Definition: Message.hpp:212
interface to power a multicast transport receiving functions
Definition: RecvTransportEngine.hpp:25
void invokedCb(size_t) HMBDC_RESTRICT override
this callback is called all the time (frequently) - the exact timing is after a batch of messages are...
Definition: RecvTransportEngine.hpp:211
Definition: Message.hpp:263
size_t sessionsRemainingActive() const
check how many other parties are sending to this engine
Definition: RecvTransportEngine.hpp:155
A Client represents a thread of execution/a task. The execution is managed by a Context. a Client object could participate in message dispatching as the receiver of specifed message types.
Definition: Client.hpp:122
void handleMessageCb(TypeTagBackupSource const &t)
only used by MH
Definition: RecvTransportEngine.hpp:116
void start()
start the show by schedule the mesage recv
Definition: RecvTransportEngine.hpp:87
Definition: BackupRecvSessionT.hpp:38
Definition: LockFreeBufferMisc.hpp:89
void messageDispatchingStartedCb(size_t const *) override
called before any messages got dispatched - only once
Definition: RecvTransportEngine.hpp:206