125 lines
3.5 KiB
C++
125 lines
3.5 KiB
C++
#include "server.hpp"
|
|
#include "config.hpp"
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <csignal>
|
|
#include <atomic>
|
|
#include <memory>
|
|
#include <thread>
|
|
|
|
namespace simple_object_storage {
|
|
|
|
std::atomic<bool> g_running{true};
|
|
std::unique_ptr<Server> g_server;
|
|
std::thread g_server_thread;
|
|
|
|
std::filesystem::path get_config_path() {
|
|
std::filesystem::path system_config_path = "/data/sos_config.json";
|
|
std::filesystem::path user_config_path = std::filesystem::path(std::getenv("HOME")) / ".config/simple_object_storage/config.json";
|
|
|
|
if (std::filesystem::exists(system_config_path)) return system_config_path;
|
|
if (std::filesystem::exists(user_config_path)) return user_config_path;
|
|
|
|
std::cout << "No config file found. Checked " << system_config_path << " and " << user_config_path << std::endl;
|
|
return std::filesystem::path();
|
|
}
|
|
|
|
bool initialize_server() {
|
|
std::filesystem::path config_path = get_config_path();
|
|
if (config_path.empty()) {
|
|
return false;
|
|
}
|
|
std::cout << "Config file: " << config_path << std::endl;
|
|
|
|
ServerConfig config;
|
|
if (!simple_object_storage::load_config(config_path, config)) {
|
|
std::cout << "Config file at " << config_path << " is not valid." << std::endl;
|
|
return false;
|
|
}
|
|
|
|
std::cout << "Starting server at " << config.host << ":" << config.port << std::endl;
|
|
std::cout << "Object store path: " << config.object_store_path << std::endl;
|
|
|
|
g_server = std::make_unique<Server>(config);
|
|
|
|
// Start server in a separate thread
|
|
g_server_thread = std::thread([&]() {
|
|
if (!g_server->start()) {
|
|
std::cerr << "Failed to start server" << std::endl;
|
|
g_running = false;
|
|
}
|
|
});
|
|
|
|
return true;
|
|
}
|
|
|
|
void stop_server() {
|
|
if (g_server) {
|
|
std::cout << "Stopping server..." << std::endl;
|
|
g_server->stop();
|
|
if (g_server_thread.joinable()) {
|
|
g_server_thread.join();
|
|
}
|
|
g_server.reset();
|
|
}
|
|
}
|
|
|
|
void signal_handler(int signal) {
|
|
switch (signal) {
|
|
case SIGHUP:
|
|
std::cout << "Received SIGHUP signal - reloading configuration" << std::endl;
|
|
stop_server();
|
|
if (!initialize_server()) {
|
|
std::cerr << "Failed to restart server with new configuration" << std::endl;
|
|
g_running = false;
|
|
}
|
|
break;
|
|
case SIGTERM:
|
|
case SIGINT:
|
|
std::cout << "Received termination signal" << std::endl;
|
|
g_running = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
// Set up signal handlers
|
|
struct sigaction sa;
|
|
sa.sa_handler = signal_handler;
|
|
sigemptyset(&sa.sa_mask);
|
|
sa.sa_flags = 0;
|
|
|
|
if (sigaction(SIGHUP, &sa, nullptr) == -1) {
|
|
std::cerr << "Failed to set up SIGHUP handler" << std::endl;
|
|
return 1;
|
|
}
|
|
if (sigaction(SIGTERM, &sa, nullptr) == -1) {
|
|
std::cerr << "Failed to set up SIGTERM handler" << std::endl;
|
|
return 1;
|
|
}
|
|
if (sigaction(SIGINT, &sa, nullptr) == -1) {
|
|
std::cerr << "Failed to set up SIGINT handler" << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
// Initial server startup
|
|
if (!initialize_server()) {
|
|
return 1;
|
|
}
|
|
|
|
// Main loop - wait for signals
|
|
while (g_running) {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
|
}
|
|
|
|
// Graceful shutdown
|
|
stop_server();
|
|
return 0;
|
|
}
|
|
|
|
} // namespace simple_object_storage
|
|
|
|
int main(int argc, char* argv[]) {
|
|
return simple_object_storage::main(argc, argv);
|
|
} |