rebase onto develop-2.0

2 files changed, 11 insertions, 0 deletions

diff --git a/infra/session_manager/session_manager.c b/infra/session_manager/session_manager.c
index 21fb779..3691805 100644
--- a/infra/session_manager/session_manager.c
+++ b/infra/session_manager/session_manager.c
@@ -593,5 +593,15 @@ void session_manager_on_thread_exit(struct module_manager *mod_mgr, int thread_i
 {
      struct session_manager *sess_mgr = module_get_ctx(mod);
      assert(sess_mgr);
+     assert(thread_id < (int)sess_mgr->cfg->thread_num);
+
      session_manager_clean(sess_mgr, thread_id);
+}
+
+// temp add for show session command
+struct session_manager_rte *session_manager_get_runtime(struct session_manager *sess_mgr, uint16_t thread_id)
+{
+     assert(sess_mgr);
+     assert(thread_id < sess_mgr->cfg->thread_num);
+     return sess_mgr->rte[thread_id];
 }
\ No newline at end of file
diff --git a/infra/session_manager/session_manager_rte.h b/infra/session_manager/session_manager_rte.h
index de8c09c..f04dceb 100644
--- a/infra/session_manager/session_manager_rte.h
+++ b/infra/session_manager/session_manager_rte.h
@@ -7,6 +7,7 @@ extern "C"
 
 #include "tuple.h"
 #include "stellar/session.h"
+#include "session_manager_cfg.h"
 
 enum session_scan_flags
 {