/
RsocketsTransmitter.cpp
73 lines (58 loc) · 1.64 KB
/
RsocketsTransmitter.cpp
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
#include <string>
#include <memory>
#include <thread>
#include <queue>
#include <iostream>
#include <system_error>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <rdma/rdma_cma.h>
#include <rdma/rsocket.h>
#include <capnp/serialize.h>
#include "utils.h"
#include "RsocketsTransmitter.h"
#include "message.capnp.h"
using namespace std;
namespace transport {
RsocketsTransmitter::RsocketsTransmitter(const size_t i, const std::string &address, const size_t port, TransmitterPool &pool)
: Transmitter(i, address, port, pool) {
cout << "Connecting Rsockets Transmitter to " << serverIP << ", port " << serverPort << endl;
int rc;
servAddr.sin_family = AF_INET;
servAddr.sin_addr.s_addr = inet_addr(serverIP.c_str());
servAddr.sin_port = htons(serverPort);
cout << "Thread " << index << ": Connecting to " << endPoint.str() << "..." << endl;
sock = rsocket(AF_INET, SOCK_STREAM, 0);
assert(sock);
rc = rconnect(sock, (struct sockaddr *) &servAddr, sizeof(servAddr));
assert(rc == 0);
}
void
RsocketsTransmitter::operator()() {
try {
// send the messages
cout << "Thread " << index << ": Sending messages..." << endl;
int flags{0};
int sent;
struct pollfd fds;
fds.events = POLLOUT;
fds.fd = sock;
while(true) {
int rc = rpoll(&fds, 1, -1);
assert(rc != 1);
MessageItem rawMsg;
parentPool.msgQueue.wait_dequeue(rawMsg);
if (rawMsg.first != nullptr && rawMsg.second != 0) {
sent = rsend(sock, rawMsg.first, rawMsg.second, flags);
assert(sent != -1);
} else {
break;
}
}
cout << "Finished." << endl;
} catch (const std::system_error &e) {
std::cerr << e.what() << std::endl;
return;
}
}
}