diff options
| author | yangwei <[email protected]> | 2024-11-05 09:39:10 +0800 |
|---|---|---|
| committer | yangwei <[email protected]> | 2024-11-05 10:22:22 +0800 |
| commit | 7f81e465226e60caa516ae82556c9411339e77cd (patch) | |
| tree | 550872b92d7ac79f8dc9175fe12d42c167a62fc0 /infra/packet_manager | |
| parent | a4157944287874d60378e4e48b9415eb44f0b0e2 (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.c | 46 |
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); |
