061b8933a7
Reviewers: zuza, lion, buda, mferencevic Reviewed By: zuza Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D669
103 lines
3.0 KiB
C++
103 lines
3.0 KiB
C++
// command to run:
|
|
// gnome-terminal --tab -e './network_chat --port 10000 --minloglevel 2' --tab -e './network_chat --port 10001 --minloglevel 2'
|
|
|
|
#include "communication.hpp"
|
|
|
|
class ChatMessage : public SenderMessage {
|
|
public:
|
|
ChatMessage() : SenderMessage(), message_("") {}
|
|
|
|
ChatMessage(std::string reactor, std::string channel, std::string message)
|
|
: SenderMessage(reactor, channel), message_(message) {}
|
|
|
|
std::string Message() const { return message_; }
|
|
|
|
template <class Archive>
|
|
void serialize(Archive &ar) {
|
|
ar(cereal::base_class<SenderMessage>(this), message_);
|
|
}
|
|
|
|
private:
|
|
std::string message_;
|
|
};
|
|
CEREAL_REGISTER_TYPE(ChatMessage);
|
|
|
|
class ChatACK : public ChatMessage {
|
|
public:
|
|
ChatACK() : ChatMessage() {}
|
|
|
|
ChatACK(std::string reactor, std::string channel, std::string message)
|
|
: ChatMessage(reactor, channel, message) {}
|
|
|
|
template <class Archive>
|
|
void serialize(Archive &ar) {
|
|
ar(cereal::base_class<ChatMessage>(this));
|
|
}
|
|
};
|
|
CEREAL_REGISTER_TYPE(ChatACK);
|
|
|
|
class ChatServer : public DistributedReactor {
|
|
public:
|
|
ChatServer(System *system, std::string name, Distributed &distributed)
|
|
: DistributedReactor(system, name, distributed) {}
|
|
|
|
virtual void Run() {
|
|
std::cout << "ChatServer is active" << std::endl;
|
|
|
|
auto chat = Open("chat").first;
|
|
|
|
chat->OnEvent<ChatACK>([](const ChatACK& ack, const EventStream::Subscription&) {
|
|
std::cout << "Received ACK from " << ack.Address() << ":"
|
|
<< ack.Port() << " -> '" << ack.Message() << "'"
|
|
<< std::endl;
|
|
});
|
|
|
|
chat->OnEvent<ChatMessage>([this](const ChatMessage& msg, const EventStream::Subscription&) {
|
|
std::cout << "Received message from " << msg.Address() << ":"
|
|
<< msg.Port() << " -> '" << msg.Message() << "'"
|
|
<< std::endl;
|
|
auto channel = msg.GetChannelToSender(system_);
|
|
if (channel != nullptr) {
|
|
channel->Send<ChatACK>("server", "chat", msg.Message());
|
|
}
|
|
});
|
|
}
|
|
};
|
|
|
|
class ChatClient : public DistributedReactor {
|
|
public:
|
|
ChatClient(System *system, std::string name, Distributed &distributed)
|
|
: DistributedReactor(system, name, distributed) {}
|
|
|
|
virtual void Run() {
|
|
std::cout << "ChatClient is active" << std::endl;
|
|
|
|
std::string address, message;
|
|
uint16_t port;
|
|
while (true) {
|
|
std::cout << "Enter IP, port and message to send." << std::endl;
|
|
std::cin >> address >> port >> message;
|
|
|
|
auto channel =
|
|
distributed_.network().Resolve(address, port, "server", "chat");
|
|
if (channel != nullptr) {
|
|
channel->Send<ChatMessage>("server", "chat", message);
|
|
} else {
|
|
std::cerr << "Couldn't resolve that server!" << std::endl;
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
int main(int argc, char *argv[]) {
|
|
gflags::ParseCommandLineFlags(&argc, &argv, true);
|
|
System system;
|
|
Distributed distributed(system);
|
|
distributed.StartServices();
|
|
system.Spawn<ChatServer>("server", distributed);
|
|
system.Spawn<ChatClient>("client", distributed);
|
|
system.AwaitShutdown();
|
|
distributed.StopServices();
|
|
return 0;
|
|
}
|