From 906b8c92aa5ea1cd098da1e6ace00177d84aa524 Mon Sep 17 00:00:00 2001
From: root <liuchang@geedgenetworks.com>
Date: Thu, 8 Aug 2024 08:25:51 +0000
Subject: [PATCH] use fieldstat4_easy auto output instead of manually write
 file

---
 src/inc_internal/maat_stat.h |  2 +-
 src/maat_rule.c              |  2 +-
 src/maat_stat.c              | 25 +++++--------------------
 3 files changed, 7 insertions(+), 22 deletions(-)

diff --git a/src/inc_internal/maat_stat.h b/src/inc_internal/maat_stat.h
index f76d76b..733cdc0 100644
--- a/src/inc_internal/maat_stat.h
+++ b/src/inc_internal/maat_stat.h
@@ -26,7 +26,7 @@ void maat_stat_free(struct maat_stat *stat);
 int maat_stat_init(struct maat_stat *stat, const char *stat_name, struct table_manager *tbl_mgr,
                    struct maat_garbage_bin *garbage_bin);
 
-void maat_stat_output(struct maat_stat *stat, struct table_manager *tbl_mgr,
+void maat_stat_refresh(struct maat_stat *stat, struct table_manager *tbl_mgr,
                       long long maat_version, int perf_on);
 
 #ifdef __cplusplus
diff --git a/src/maat_rule.c b/src/maat_rule.c
index 9bd289d..d87c5e6 100644
--- a/src/maat_rule.c
+++ b/src/maat_rule.c
@@ -532,7 +532,7 @@ void *rule_monitor_loop(void *arg)
         maat_plugin_table_garbage_collect_routine(maat_inst->tbl_mgr);
 
         if ((1 == maat_inst->opts.stat_on) && (time(NULL) % 2 == 0)) {
-            maat_stat_output(maat_inst->stat, maat_inst->tbl_mgr,
+            maat_stat_refresh(maat_inst->stat, maat_inst->tbl_mgr,
                              maat_inst->maat_version, maat_inst->opts.perf_on);
         }
     }
diff --git a/src/maat_stat.c b/src/maat_stat.c
index ee9e8db..8ed28e8 100644
--- a/src/maat_stat.c
+++ b/src/maat_stat.c
@@ -233,10 +233,12 @@ int maat_stat_init(struct maat_stat *stat, const char *stat_name, struct table_m
     fs_table_column_register(stat);
     fs_table_row_register(stat);
 
+    fieldstat_easy_enable_auto_output(stat->fs_handle, stat->stat_file, 2);
+
     return 0;
 }
 
-static void fs_table_row_output(FILE *fp, struct maat_stat *stat, int perf_on)
+static void fs_table_row_refresh(struct maat_stat *stat, int perf_on)
 {
     long long plugin_cache_num = 0, plugin_rule_num = 0;
     long long total_rule_num = 0, total_scan_bytes = 0, total_update_err = 0;
@@ -420,29 +422,14 @@ static void fs_table_row_output(FILE *fp, struct maat_stat *stat, int perf_on)
                                stat->g_metric_id[STATUS_GROUP_REF_EXCL_NUM],
                                NULL, 0, g2g_excl_rule_num); 
 
-    char *buff = NULL;
-    size_t buff_len = 0;
-
-    fieldstat_easy_output(stat->fs_handle, &buff, &buff_len);
-    fwrite(buff, buff_len, 1, fp);
-    fflush(fp);
-
-    FREE(buff);
 }
 
-void maat_stat_output(struct maat_stat *stat, struct table_manager *tbl_mgr,
+void maat_stat_refresh(struct maat_stat *stat, struct table_manager *tbl_mgr,
                       long long maat_version, int perf_on)
 {
     if (NULL == stat || NULL == stat->fs_handle) {
         return;
     }
-    
-    FILE *fp = fopen(stat->stat_file, "w+");
-    if (NULL == fp) {
-        log_fatal(stat->logger, MODULE_MAAT_STAT, "open stat_file:%s failed.",
-                  stat->stat_file);
-        return;
-    }
 
     long long active_thread_num =
         alignment_int64_array_cnt(stat->thread_call_cnt, stat->nr_worker_thread);
@@ -507,7 +494,5 @@ void maat_stat_output(struct maat_stat *stat, struct table_manager *tbl_mgr,
                                stat->g_metric_id[STATUS_ZOMBIE_STREAM_NUM],
                                NULL, 0, stat->zombie_rs_stream); 
 
-    fs_table_row_output(fp, stat, perf_on);
-
-    fclose(fp);
+    fs_table_row_refresh(stat, perf_on);
 }
\ No newline at end of file