summaryrefslogtreecommitdiff
path: root/infra/packet_manager
diff options
context:
space:
mode:
authoryangwei <[email protected]>2024-11-05 09:39:10 +0800
committeryangwei <[email protected]>2024-11-05 10:22:22 +0800
commit7f81e465226e60caa516ae82556c9411339e77cd (patch)
tree550872b92d7ac79f8dc9175fe12d42c167a62fc0 /infra/packet_manager
parenta4157944287874d60378e4e48b9415eb44f0b0e2 (diff)
🦄 refactor(stellar_module to module): simplify stellar module to module
Diffstat (limited to 'infra/packet_manager')
-rw-r--r--infra/packet_manager/packet_manager.c46
1 files changed, 21 insertions, 25 deletions
diff --git a/infra/packet_manager/packet_manager.c b/infra/packet_manager/packet_manager.c
index f132536..0238ec0 100644
--- a/infra/packet_manager/packet_manager.c
+++ b/infra/packet_manager/packet_manager.c
@@ -5,7 +5,7 @@
#include "packet_manager.h"
#include "fieldstat/fieldstat_easy.h"
-#define PACKET_MANAGER_MODULE_NAME "packet_manager_module"
+
#define PACKET_MANAGER_LOG_ERROR(format, ...) STELLAR_LOG_ERROR(__thread_local_logger, "packet manager", format, ##__VA_ARGS__)
#define PACKET_MANAGER_LOG_FATAL(format, ...) STELLAR_LOG_FATAL(__thread_local_logger, "packet manager", format, ##__VA_ARGS__)
@@ -436,10 +436,10 @@ void packet_manager_print_stat(struct packet_manager *pkt_mgr, uint16_t thread_i
* packet manager module
******************************************************************************/
-static void on_polling(struct stellar_module_manager *mod_mgr, void *args)
+static void on_polling(struct module_manager *mod_mgr, void *args)
{
uint64_t now_ms = clock_get_real_time_ms();
- int thread_id = stellar_module_manager_get_thread_id(mod_mgr);
+ int thread_id = module_manager_get_thread_id(mod_mgr);
struct packet_manager *pkt_mgr = (struct packet_manager *)args;
static __thread uint64_t last_sync_stat_ms = 0;
@@ -457,61 +457,57 @@ static void on_polling(struct stellar_module_manager *mod_mgr, void *args)
}
}
-struct packet_manager *stellar_module_get_packet_manager(struct stellar_module_manager *mod_mgr)
+struct packet_manager *module_to_packet_manager(struct module *mod)
{
- assert(mod_mgr);
- struct stellar_module *pkt_mgr_mod = stellar_module_manager_get_module(mod_mgr, PACKET_MANAGER_MODULE_NAME);
- if (pkt_mgr_mod == NULL)
- {
- return NULL;
- }
- return (struct packet_manager *)stellar_module_get_ctx(pkt_mgr_mod);
+ assert(mod);
+ assert(strcmp(module_get_name(mod), PACKET_MANAGER_MODULE_NAME) == 0);
+ return (struct packet_manager *)module_get_ctx(mod);
}
-struct stellar_module *packet_manager_on_init(struct stellar_module_manager *mod_mgr)
+struct module *packet_manager_on_init(struct module_manager *mod_mgr)
{
assert(mod_mgr);
- struct mq_schema *mq_sche = stellar_module_manager_get_mq_schema(mod_mgr);
+ struct mq_schema *mq_sche = module_manager_get_mq_schema(mod_mgr);
assert(mq_sche);
- uint16_t thread_num = stellar_module_manager_get_max_thread_num(mod_mgr);
+ uint16_t thread_num = module_manager_get_max_thread_num(mod_mgr);
struct packet_manager *pkt_mgr = packet_manager_new(mq_sche, thread_num);
if (pkt_mgr == NULL)
{
return NULL;
}
- stellar_module_manager_polling_subscribe(mod_mgr, on_polling, pkt_mgr);
+ module_manager_polling_subscribe(mod_mgr, on_polling, pkt_mgr);
- struct stellar_module *pkt_mgr_mod = stellar_module_new(PACKET_MANAGER_MODULE_NAME, NULL);
+ struct module *pkt_mgr_mod = module_new(PACKET_MANAGER_MODULE_NAME, NULL);
if (pkt_mgr_mod == NULL)
{
PACKET_MANAGER_LOG_ERROR("failed to create packet_manager");
packet_manager_free(pkt_mgr);
return NULL;
}
- stellar_module_set_ctx(pkt_mgr_mod, pkt_mgr);
+ module_set_ctx(pkt_mgr_mod, pkt_mgr);
PACKET_MANAGER_LOG_FATAL("packet_manager init");
return pkt_mgr_mod;
}
-void packet_manager_on_exit(struct stellar_module_manager *mod_mgr __attribute__((unused)), struct stellar_module *mod)
+void packet_manager_on_exit(struct module_manager *mod_mgr __attribute__((unused)), struct module *mod)
{
if (mod)
{
- struct packet_manager *pkt_mgr = stellar_module_get_ctx(mod);
+ struct packet_manager *pkt_mgr = module_get_ctx(mod);
packet_manager_free(pkt_mgr);
- stellar_module_free(mod);
+ module_free(mod);
PACKET_MANAGER_LOG_FATAL("packet_manager exit");
}
}
-struct stellar_module *packet_manager_on_thread_init(struct stellar_module_manager *mod_mgr, int thread_id, struct stellar_module *mod)
+struct module *packet_manager_on_thread_init(struct module_manager *mod_mgr, int thread_id, struct module *mod)
{
- struct packet_manager *pkt_mgr = stellar_module_get_ctx(mod);
+ struct packet_manager *pkt_mgr = module_get_ctx(mod);
assert(pkt_mgr);
- struct mq_runtime *mq_rte = stellar_module_manager_get_mq_runtime(mod_mgr);
+ struct mq_runtime *mq_rte = module_manager_get_mq_runtime(mod_mgr);
assert(mq_rte);
assert(thread_id < pkt_mgr->thread_num);
@@ -526,9 +522,9 @@ struct stellar_module *packet_manager_on_thread_init(struct stellar_module_manag
}
}
-void packet_manager_on_thread_exit(struct stellar_module_manager *mod_mgr __attribute__((unused)), int thread_id, struct stellar_module *mod)
+void packet_manager_on_thread_exit(struct module_manager *mod_mgr __attribute__((unused)), int thread_id, struct module *mod)
{
- struct packet_manager *pkt_mgr = stellar_module_get_ctx(mod);
+ struct packet_manager *pkt_mgr = module_get_ctx(mod);
assert(pkt_mgr);
assert(thread_id < pkt_mgr->thread_num);