Adjust thread index type to uint16 for future expansion & Organize stellar directory files

2024-04-25 16:48:50 +08:00
parent 17ca476c24
commit 611fda598f
22 changed files with 232 additions and 222 deletions
--- a/src/stellar/main.cpp
+++ b/src/stellar/main.cpp
@@ -0,0 +1,130 @@
+#include <stdio.h>
+#include <assert.h>
+#include <errno.h>
+#include <unistd.h>
+#include <string.h>
+#include <signal.h>
+#include <stdlib.h>
+
+#include "logo.h"
+#include "config.h"
+#include "timestamp.h"
+#include "id_generator.h"
+#include "stellar_priv.h"
+
+static const char *log_config_file = "./conf/log.toml";
+static const char *stellar_config_file = "./conf/stellar.toml";
+
+static void signal_handler(int signo)
+{
+    if (signo == SIGINT)
+    {
+        STELLAR_LOG_STATE("SIGINT received, notify threads to exit !!!");
+        ATOMIC_SET(&runtime->need_exit, 1);
+    }
+
+    if (signo == SIGQUIT)
+    {
+        STELLAR_LOG_STATE("SIGQUIT received, notify threads to exit !!!");
+        ATOMIC_SET(&runtime->need_exit, 1);
+    }
+
+    if (signo == SIGTERM)
+    {
+        STELLAR_LOG_STATE("SIGTERM received, notify threads to exit !!!");
+        ATOMIC_SET(&runtime->need_exit, 1);
+    }
+
+    if (signo == SIGHUP)
+    {
+        STELLAR_LOG_STATE("SIGHUP received, reload log level !!!");
+        log_reload_level(log_config_file);
+    }
+}
+
+int main(int argc, char **argv)
+{
+    timestamp_update();
+
+    signal(SIGINT, signal_handler);
+    signal(SIGQUIT, signal_handler);
+    signal(SIGTERM, signal_handler);
+    signal(SIGHUP, signal_handler);
+
+    if (log_init(log_config_file) != 0)
+    {
+        STELLAR_LOG_ERROR("unable to init log");
+        goto error_out;
+    }
+    STELLAR_LOG_STATE("start stellar (version: %s)\n %s", __stellar_version, logo_str);
+
+    if (stellar_load_config(stellar_config_file, config) != 0)
+    {
+        STELLAR_LOG_ERROR("unable to load config file");
+        goto error_out;
+    }
+    stellar_print_config(config);
+    STELLAR_LOG_DEBUG("sizeof(struct session) = %lu bytes", sizeof(struct session));
+
+    if (id_generator_init(config->dev_opts.base, config->dev_opts.offset) != 0)
+    {
+        STELLAR_LOG_ERROR("unable to init id generator");
+        goto error_out;
+    }
+
+    runtime->stat = stellar_stat_new(config->io_opts.nr_threads);
+    if (runtime->stat == NULL)
+    {
+        STELLAR_LOG_ERROR("unable to create stellar stat");
+        goto error_out;
+    }
+
+    runtime->plug_mgr = plugin_manager_new();
+    if (runtime->plug_mgr == NULL)
+    {
+        STELLAR_LOG_ERROR("unable to create plugin manager");
+        goto error_out;
+    }
+
+    runtime->packet_io = packet_io_new(&config->io_opts);
+    if (runtime->packet_io == NULL)
+    {
+        STELLAR_LOG_ERROR("unable to create packet io");
+        goto error_out;
+    }
+
+    if (stellar_thread_init(runtime, config) != 0)
+    {
+        STELLAR_LOG_ERROR("unable to init thread context");
+        goto error_out;
+    }
+
+    if (stellar_thread_run(runtime, config) != 0)
+    {
+        STELLAR_LOG_ERROR("unable to create worker thread");
+        goto error_out;
+    }
+
+    runtime->stat_last_output_ts = timestamp_get_msec();
+    while (!ATOMIC_READ(&runtime->need_exit))
+    {
+        timestamp_update();
+        if (timestamp_get_msec() - runtime->stat_last_output_ts > 2000)
+        {
+            runtime->stat_last_output_ts = timestamp_get_msec();
+            stellar_stat_output(runtime->stat);
+        }
+        usleep(1000); // 1ms
+    }
+
+error_out:
+    stellar_thread_join(runtime, config);
+    stellar_thread_clean(runtime, config);
+    packet_io_free(runtime->packet_io);
+    plugin_manager_free(runtime->plug_mgr);
+    stellar_stat_free(runtime->stat);
+    STELLAR_LOG_STATE("stellar exit !!!\n");
+    log_free();
+
+    return 0;
+}