🦄 refactor(polling manager): merge polling into module manager

1 files changed, 4 insertions, 8 deletions

diff --git a/infra/session_manager/session_manager.c b/infra/session_manager/session_manager.c
index 5f3820f..ecc55f6 100644
--- a/infra/session_manager/session_manager.c
+++ b/infra/session_manager/session_manager.c
@@ -5,7 +5,6 @@
 #include "stellar/packet_manager.h"
 #include "stellar/session_manager.h"
 #include "stellar/module_manager.h"
-#include "stellar/polling_manager.h"
 
 #include "utils.h"
 #include "session_internal.h"
@@ -174,10 +173,9 @@ static void clean_session(struct session_manager_runtime *sess_mgr_rt, uint64_t
     }
 }
 
-static void on_polling(struct stellar_polling_manager *poll_mgr, void *args)
+static void on_polling(struct stellar_module_manager *mod_mgr, void *args)
 {
     struct session_manager *sess_mgr = (struct session_manager *)args;
-    struct stellar_module_manager *mod_mgr = sess_mgr->mod_mgr;
     int thread_id = stellar_module_manager_get_thread_id(mod_mgr);
     struct session_manager_runtime *sess_mgr_rt = sess_mgr->runtime[thread_id];
     uint64_t now_ms = clock_get_real_time_ms();
@@ -316,7 +314,7 @@ void session_manager_free(struct session_manager *sess_mgr)
     }
 }
 
-struct session_manager *session_manager_new(struct stellar_polling_manager *poll_mgr, struct packet_manager *pkt_mgr, struct mq_schema *mq_schema, const char *toml_file)
+struct session_manager *session_manager_new(struct stellar_module_manager *mod_mgr, struct packet_manager *pkt_mgr, struct mq_schema *mq_schema, const char *toml_file)
 {
     assert(pkt_mgr);
     assert(mq_schema);
@@ -368,7 +366,7 @@ struct session_manager *session_manager_new(struct stellar_polling_manager *poll
         }
     }
 
-    stellar_polling_subscribe(poll_mgr, on_polling, sess_mgr);
+    stellar_module_manager_polling_subscribe(mod_mgr, on_polling, sess_mgr);
 
     return sess_mgr;
 
@@ -437,14 +435,12 @@ struct stellar_module *session_manager_on_init(struct stellar_module_manager *mo
     assert(mod_mgr);
     struct packet_manager *pkt_mgr = stellar_module_get_packet_manager(mod_mgr);
     assert(pkt_mgr);
-    struct stellar_polling_manager *poll_mgr = stellar_module_get_polling_manager(mod_mgr);
-    assert(poll_mgr);
     struct mq_schema *mq_schema = stellar_module_manager_get_mq_schema(mod_mgr);
     assert(mq_schema);
     const char *toml_file = stellar_module_manager_get_toml_path(mod_mgr);
     assert(toml_file);
 
-    struct session_manager *sess_mgr = session_manager_new(poll_mgr, pkt_mgr, mq_schema, toml_file);
+    struct session_manager *sess_mgr = session_manager_new(mod_mgr, pkt_mgr, mq_schema, toml_file);
     if (sess_mgr == NULL)
     {
         return NULL;