mirror of
https://github.com/Relintai/rcpp_framework.git
synced 2024-11-14 04:57:21 +01:00
86 lines
2.0 KiB
C++
86 lines
2.0 KiB
C++
#include "mqtt_server.h"
|
|
|
|
void MQTTServer::add_local_session(const std::string &filter, void (*func)(const std::string &client_id, const std::vector<uint8_t> &data)) {
|
|
session_manager->add_local_session(filter, func);
|
|
}
|
|
|
|
void MQTTServer::initialize() {
|
|
evloop = event_base_new();
|
|
|
|
if (!evloop) {
|
|
printf("Could not initialize libevent\n");
|
|
return;
|
|
}
|
|
|
|
std::memset(&sin, 0, sizeof(sin));
|
|
sin.sin_family = AF_INET;
|
|
evutil_inet_pton(sin.sin_family, bind_address.c_str(), &sin.sin_addr);
|
|
sin.sin_port = htons(port);
|
|
|
|
listener = evconnlistener_new_bind(evloop, MQTTServer::listener_cb,
|
|
(void *)this,
|
|
LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE, -1,
|
|
(struct sockaddr *)&sin, sizeof(sin));
|
|
|
|
if (!listener) {
|
|
std::cerr << "Could not create listener!\n";
|
|
return;
|
|
}
|
|
}
|
|
|
|
void MQTTServer::listener_cb(struct evconnlistener *listener, evutil_socket_t fd, struct sockaddr *sa, int socklen, void *user_data) {
|
|
|
|
MQTTServer *server = static_cast<MQTTServer *>(user_data);
|
|
|
|
struct bufferevent *bev;
|
|
|
|
bev = bufferevent_socket_new(server->evloop, fd, BEV_OPT_CLOSE_ON_FREE);
|
|
if (!bev) {
|
|
std::cerr << "Error constructing bufferevent!\n";
|
|
event_base_loopbreak(server->evloop);
|
|
return;
|
|
}
|
|
|
|
server->session_manager->accept_connection(bev);
|
|
}
|
|
|
|
void MQTTServer::run_async() {
|
|
if (_thread) {
|
|
printf("MQTTServer::run_async Error! A thread is already runnig!\n");
|
|
return;
|
|
}
|
|
|
|
_thread = new std::thread([this]() { event_base_dispatch(this->evloop); });
|
|
}
|
|
|
|
MQTTServer::MQTTServer() {
|
|
bind_address = "0";
|
|
port = 1883;
|
|
_thread = nullptr;
|
|
|
|
session_manager = new SessionManager();
|
|
|
|
evloop = nullptr;
|
|
listener = nullptr;
|
|
}
|
|
|
|
MQTTServer::~MQTTServer() {
|
|
//this first, as evloop runs in _thread
|
|
if (evloop && event_base_loopexit(evloop, NULL)) {
|
|
std::cout << "failed to exit event loop\n";
|
|
}
|
|
|
|
if (_thread) {
|
|
_thread->join();
|
|
delete _thread;
|
|
}
|
|
|
|
if (listener)
|
|
evconnlistener_free(listener);
|
|
|
|
if (evloop)
|
|
event_base_free(evloop);
|
|
|
|
delete session_manager;
|
|
}
|