mirror of
https://github.com/Syping/dtranslatebot.git
synced 2024-11-22 13:50:22 +01:00
rename queue to message_queue, add exception handling to settings
This commit is contained in:
parent
b710fa3050
commit
acbab3c97b
6 changed files with 84 additions and 77 deletions
|
@ -19,12 +19,14 @@
|
||||||
cmake_minimum_required(VERSION 3.16)
|
cmake_minimum_required(VERSION 3.16)
|
||||||
project(dtranslatebot VERSION 0.1 DESCRIPTION "Discord Translation Bot")
|
project(dtranslatebot VERSION 0.1 DESCRIPTION "Discord Translation Bot")
|
||||||
|
|
||||||
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
|
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake")
|
||||||
|
|
||||||
add_executable(${PROJECT_NAME}
|
add_executable(${PROJECT_NAME}
|
||||||
src/main.cpp
|
src/main.cpp
|
||||||
src/queue.cpp
|
src/message_queue.cpp
|
||||||
|
src/message_queue.h
|
||||||
src/settings.cpp
|
src/settings.cpp
|
||||||
|
src/settings.h
|
||||||
)
|
)
|
||||||
|
|
||||||
set(THREADS_PREFER_PTHREAD_FLAG ON)
|
set(THREADS_PREFER_PTHREAD_FLAG ON)
|
||||||
|
|
10
src/main.cpp
10
src/main.cpp
|
@ -19,7 +19,7 @@
|
||||||
#include <dpp/dpp.h>
|
#include <dpp/dpp.h>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
#include "queue.h"
|
#include "message_queue.h"
|
||||||
#include "settings.h"
|
#include "settings.h"
|
||||||
|
|
||||||
int main(int argc, char* argv[]) {
|
int main(int argc, char* argv[]) {
|
||||||
|
@ -36,10 +36,10 @@ int main(int argc, char* argv[]) {
|
||||||
|
|
||||||
bot.on_log(dpp::utility::cout_logger());
|
bot.on_log(dpp::utility::cout_logger());
|
||||||
|
|
||||||
bot::queue queue;
|
bot::message_queue message_queue;
|
||||||
std::thread queue_loop(&bot::queue::run, &queue, &bot, &settings);
|
std::thread message_queue_loop(&bot::message_queue::run, &message_queue, &bot, &settings);
|
||||||
|
|
||||||
bot.on_message_create([&bot, &queue, &settings](const dpp::message_create_t &event) {
|
bot.on_message_create([&bot, &message_queue, &settings](const dpp::message_create_t &event) {
|
||||||
if (event.msg.author.is_bot())
|
if (event.msg.author.is_bot())
|
||||||
return;
|
return;
|
||||||
|
|
||||||
|
@ -55,7 +55,7 @@ int main(int argc, char* argv[]) {
|
||||||
message.webhook = channel->webhook;
|
message.webhook = channel->webhook;
|
||||||
message.source = channel->source;
|
message.source = channel->source;
|
||||||
message.target = channel->target;
|
message.target = channel->target;
|
||||||
queue.add(message);
|
message_queue.add(message);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
settings.unlock();
|
settings.unlock();
|
||||||
|
|
|
@ -18,18 +18,18 @@
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
#include "queue.h"
|
#include "message_queue.h"
|
||||||
#include "settings.h"
|
#include "settings.h"
|
||||||
using namespace std::chrono_literals;
|
using namespace std::chrono_literals;
|
||||||
|
|
||||||
void bot::queue::add(const bot::message &message)
|
void bot::message_queue::add(const bot::message &message)
|
||||||
{
|
{
|
||||||
m_mutex.lock();
|
m_mutex.lock();
|
||||||
m_queue.push_back(message);
|
m_queue.push_back(message);
|
||||||
m_mutex.unlock();
|
m_mutex.unlock();
|
||||||
}
|
}
|
||||||
|
|
||||||
void bot::queue::run(dpp::cluster *bot, bot::settings::settings *settings)
|
void bot::message_queue::run(dpp::cluster *bot, bot::settings::settings *settings)
|
||||||
{
|
{
|
||||||
while (true) {
|
while (true) {
|
||||||
m_mutex.lock();
|
m_mutex.lock();
|
|
@ -16,8 +16,8 @@
|
||||||
* responsible for anything with use of the software, you are self responsible.
|
* responsible for anything with use of the software, you are self responsible.
|
||||||
*****************************************************************************/
|
*****************************************************************************/
|
||||||
|
|
||||||
#ifndef QUEUE_H
|
#ifndef MESSAGE_QUEUE_H
|
||||||
#define QUEUE_H
|
#define MESSAGE_QUEUE_H
|
||||||
#include <dpp/dpp.h>
|
#include <dpp/dpp.h>
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
@ -36,7 +36,7 @@ namespace bot {
|
||||||
std::string target;
|
std::string target;
|
||||||
};
|
};
|
||||||
|
|
||||||
class queue {
|
class message_queue {
|
||||||
public:
|
public:
|
||||||
void add(const bot::message &message);
|
void add(const bot::message &message);
|
||||||
void run(dpp::cluster *bot, bot::settings::settings *settings);
|
void run(dpp::cluster *bot, bot::settings::settings *settings);
|
||||||
|
@ -47,4 +47,4 @@ namespace bot {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif //QUEUE_H
|
#endif //MESSAGE_QUEUE_H
|
|
@ -17,6 +17,7 @@
|
||||||
*****************************************************************************/
|
*****************************************************************************/
|
||||||
|
|
||||||
#include <dpp/dpp.h>
|
#include <dpp/dpp.h>
|
||||||
|
#include <mutex>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include "settings.h"
|
#include "settings.h"
|
||||||
|
|
||||||
|
@ -76,20 +77,19 @@ bool bot::settings::settings::parse(const std::string &filename)
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
m_mutex.lock();
|
const std::lock_guard<std::recursive_mutex> guard(m_mutex);
|
||||||
|
try {
|
||||||
m_token = json_token.value();
|
m_token = json_token.value();
|
||||||
|
|
||||||
auto json_translate = json.find("translate");
|
auto json_translate = json.find("translate");
|
||||||
if (json_translate == json.end()) {
|
if (json_translate == json.end()) {
|
||||||
std::cerr << "Translate settings can not be found" << std::endl;
|
std::cerr << "Translate settings can not be found" << std::endl;
|
||||||
m_mutex.unlock();
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto json_translate_hostname = json_translate.value().find("hostname");
|
auto json_translate_hostname = json_translate.value().find("hostname");
|
||||||
if (json_translate_hostname == json_translate.value().end()) {
|
if (json_translate_hostname == json_translate.value().end()) {
|
||||||
std::cerr << "\"hostname\" can not be found in Translate settings" << std::endl;
|
std::cerr << "\"hostname\" can not be found in Translate settings" << std::endl;
|
||||||
m_mutex.unlock();
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
m_translate_settings.hostname = json_translate_hostname.value();
|
m_translate_settings.hostname = json_translate_hostname.value();
|
||||||
|
@ -97,7 +97,6 @@ bool bot::settings::settings::parse(const std::string &filename)
|
||||||
auto json_translate_port = json_translate.value().find("port");
|
auto json_translate_port = json_translate.value().find("port");
|
||||||
if (json_translate_port == json_translate.value().end()) {
|
if (json_translate_port == json_translate.value().end()) {
|
||||||
std::cerr << "\"port\" can not be found in Translate settings" << std::endl;
|
std::cerr << "\"port\" can not be found in Translate settings" << std::endl;
|
||||||
m_mutex.unlock();
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
m_translate_settings.port = json_translate_port.value();
|
m_translate_settings.port = json_translate_port.value();
|
||||||
|
@ -105,7 +104,6 @@ bool bot::settings::settings::parse(const std::string &filename)
|
||||||
auto json_translate_url = json_translate.value().find("url");
|
auto json_translate_url = json_translate.value().find("url");
|
||||||
if (json_translate_url == json_translate.value().end()) {
|
if (json_translate_url == json_translate.value().end()) {
|
||||||
std::cerr << "\"url\" can not be found in Translate settings" << std::endl;
|
std::cerr << "\"url\" can not be found in Translate settings" << std::endl;
|
||||||
m_mutex.unlock();
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
m_translate_settings.url = json_translate_url.value();
|
m_translate_settings.url = json_translate_url.value();
|
||||||
|
@ -151,9 +149,15 @@ bool bot::settings::settings::parse(const std::string &filename)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
m_mutex.unlock();
|
|
||||||
return true;
|
return true;
|
||||||
|
}
|
||||||
|
catch (const std::exception &exception) {
|
||||||
|
std::cerr << "Exception thrown while parsing configuration: " << exception.what() << std::endl;
|
||||||
|
}
|
||||||
|
catch (...) {
|
||||||
|
std::cerr << "Exception thrown while parsing configuration: unknown" << std::endl;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
void bot::settings::settings::unlock()
|
void bot::settings::settings::unlock()
|
||||||
|
|
|
@ -20,6 +20,7 @@
|
||||||
#define SETTINGS_H
|
#define SETTINGS_H
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
|
#include <string>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
namespace bot {
|
namespace bot {
|
||||||
|
|
Loading…
Reference in a new issue