add FlushAll for xlog

landuochong · Aug 10, 2022 · 6adbe2c · 6adbe2c
1 parent 45e8a74
commit 6adbe2c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/mars/log/src/xlogger_interface.cc b/mars/log/src/xlogger_interface.cc
@@ -144,6 +144,17 @@ void Flush(uintptr_t _instance_ptr, bool _is_sync) {
     }
 }
 
+void FlushAll(bool _is_sync) {
+    _is_sync ? appender_flush_sync() : appender_flush();
+    ScopedLock lock(sg_mutex);
+    // loop through all categories
+    for (auto it = sg_map.begin(); it != sg_map.end(); ++it) {
+        XloggerCategory* category  = it->second;
+        XloggerAppender* appender = reinterpret_cast<XloggerAppender*>(category->GetAppender());
+        _is_sync ? appender->FlushSync() : appender->Flush();
+    }
+}
+
 void SetConsoleLogOpen(uintptr_t _instance_ptr, bool _is_open) {
     if (0 == _instance_ptr) {
        appender_set_console_log(_is_open);

diff --git a/mars/log/xlogger_interface.h b/mars/log/xlogger_interface.h
@@ -56,6 +56,8 @@ void SetAppenderMode(uintptr_t _instance_ptr, TAppenderMode _mode);
 
 void Flush(uintptr_t _instance_ptr, bool _is_sync);
 
+void FlushAll(bool _is_sync);
+
 void SetConsoleLogOpen(uintptr_t _instance_ptr, bool _is_open);
 
 void SetMaxFileSize(uintptr_t _instance_ptr, long _max_file_size);