void DataServer::onClientConnection(const muduo::net::TcpConnectionPtr& conn) {
LOG_INFO << conn->peerAddress().toIpPort() << (conn->connected() ? " is UP" : " is DOWN");
}
void onConnection(const muduo::net::TcpConnectionPtr& conn) {
if(conn->connected()) {
LOG_INFO << "connect to " << conn->peerAddress().toIpPort();
conn->setTcpNoDelay(true);
conn->send("ready\r\n");
}
else {
LOG_INFO << conn->peerAddress().toIpPort() << " is down";
client.disconnect();
}
}
void onMessage(const muduo::net::TcpConnectionPtr& conn,
muduo::net::Buffer* buffer, muduo::Timestamp time) {
//[solutions] <clientId> <id> <subid> <size> <pos>\r\n
while(buffer->findCRLF()) {
const char* crlf = buffer->findCRLF();
std::string request(buffer->peek(), crlf);
buffer->retrieveUntil(crlf+2);
LOG_INFO << "worker receive " << request;
bool computeSolutions = (request.find("solutions") == 0);
if(computeSolutions)
request = request.substr(request.find("solutions"), request.size());
std::vector<std::string> tokens;
boost::split(tokens, request, boost::is_any_of(" "));
if(tokens.size() >= 4) {
int size = std::stoi(tokens[3]);
std::vector<int> pos;
for(size_t i = 4; i < tokens.size(); ++i)
pos.push_back(std::stoi(tokens[i]));
QueenTask task(tokens[0], tokens[1], tokens[2], size, pos, computeSolutions);
threadPool.run(boost::bind(&QueenWorker::processRequest, this, conn, task));
}
else {
LOG_ERROR << "receive bad request " << request << " from " << conn->peerAddress().toIpPort();
}
}
}
void MedianExecutor::onMessage(const muduo::net::TcpConnectionPtr& conn,
muduo::net::Buffer* buffer, muduo::Timestamp time) {
while(buffer->findCRLF()) {
const char* crlf = buffer->findCRLF();
std::string response(buffer->peek(), crlf);
buffer->retrieveUntil(crlf+2);
std::string id(conn->peerAddress().toIpPort().c_str());
LOG_INFO << "MedianExecutor receive: [" << response << "] from " << id;
std::vector<std::string> tokens;
boost::split(tokens, response, boost::is_any_of(" "));
if(tokens[0] == "random" || tokens[0] == "split") {
for(size_t i = 1; i < tokens.size(); ++i) {
int64_t n = std::stol(tokens[i]);
workerBuffers[id].push_back(n);
}
}
else {
LOG_ERROR << "MedianExecutor receive unknown response: [" << response << "] from " << id;
conn->shutdown();
}
{
std::unique_lock<std::mutex> lock(mt);
++workingSize;
cond.notify_all();
}
}
}
void onConnection(const muduo::net::TcpConnectionPtr& conn) {
if(conn->connected()) {
conn_ = conn;
sendRequest(14, false);
}
else {
LOG_INFO << conn->peerAddress().toIpPort() << " is down";
conn->shutdown();
}
}
void DataServer::onWorkerConnection(const muduo::net::TcpConnectionPtr& conn) {
std::string peer(conn->peerAddress().toIpPort().c_str());
if(conn->connected()) {
LOG_INFO << "connected to worker: " << peer;
connections.insert(std::make_pair(peer, conn));
{
std::unique_lock<std::mutex> lock(mt);
size -= 1;
cond.notify_all();
}
}
else {
LOG_INFO << "worker: " << peer << " is down";
workers.erase(peer);
connections.erase(peer);
}
}
void SudoKuServer::onConnection(const muduo::net::TcpConnectionPtr& conn)
{
LOG_INFO<< "SudoKuServer - " << conn->peerAddress().toIpPort() << " -> "
<< conn->localAddress().toIpPort() << " is "
<< (conn->connected() ? "UP" : "DOWN");
}
// command:
// 1. genNumber n response: ok
// 2. average response: number<double>
// 3. median response: number<int64_t>
// 4. sort response: ok
// 5. freq n response: <n1, freq1> <n2, freq2> ... <n, freq>
void DataServer::onClientMessage(const muduo::net::TcpConnectionPtr& conn,
muduo::net::Buffer* buf, muduo::Timestamp time) {
while(buf->findCRLF()) {
const char* crlf = buf->findCRLF();
std::string command(buf->peek(), crlf);
buf->retrieveUntil(crlf+2);
std::vector<std::string> tokens;
boost::split(tokens, command, boost::is_any_of(" "));
if(command.find("genNumber") == 0) {
if(tokens.size() == 3) {
GenNumberExecutor executor(connections);
dataExecutor = &executor;
int64_t number = std::stol(tokens[1]);
char mode = tokens[2][0];
executor.execute(number, mode);
conn->send("OK\r\n");
}
else {
conn->send("usage: genNumber [n] [n/u/z]\r\n");
}
}
else if(command == "average") {
AverageExecutor executor(connections);
dataExecutor = &executor;
double average = executor.execute();
conn->send("average: " + std::to_string(average) + "\r\n");
}
else if(command == "median") {
MedianExecutor executor(connections);
dataExecutor = &executor;
int64_t median = executor.execute();
conn->send("median: " + std::to_string(median) + "\r\n");
}
else if(command == "sort") {
SortExecutor executor(connections);
dataExecutor = &executor;
executor.execute();
conn->send("OK\r\n");
}
else if(command.find("freq") == 0) {
size_t freqNumber = std::stoi(tokens[1]);
FreqExecutor executor(connections);
dataExecutor = &executor;
std::vector<std::pair<int64_t, int64_t>> freqs = executor.execute(freqNumber);
std::string response = "freqs:";
for(auto& freq : freqs) {
response += " " + std::to_string(freq.first) + " " + std::to_string(freq.second);
}
response += "\r\n";
conn->send(response);
}
else {
LOG_ERROR << "receive unknown command [" << command << "] from "
<< conn->peerAddress().toIpPort();
conn->shutdown();
}
}
}