Adjust Stats plugin

Zadamsa · Oct 3, 2024 · f3f64ac · f3f64ac
1 parent 6963127
commit f3f64ac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/process/stats.cpp b/process/stats.cpp
@@ -82,21 +82,21 @@ ProcessPlugin *StatsPlugin::copy()
    return new StatsPlugin(*this);
 }
 
-int StatsPlugin::post_create(Flow &rec, const Packet &pkt)
+ProcessPlugin::FlowAction StatsPlugin::post_create(Flow &rec, const Packet &pkt)
 {
    m_packets += 1;
    m_new_flows += 1;
    m_flows_in_cache += 1;
    check_timestamp(pkt);
-   return 0;
+   return ProcessPlugin::FlowAction::GET_ALL_DATA;
 }
 
-int StatsPlugin::post_update(Flow &rec, const Packet &pkt)
+ProcessPlugin::FlowAction StatsPlugin::post_update(Flow &rec, const Packet &pkt)
 {
    m_packets += 1;
    m_cache_hits += 1;
    check_timestamp(pkt);
-   return 0;
+   return ProcessPlugin::FlowAction::GET_ALL_DATA;
 }
 
 void StatsPlugin::pre_export(Flow &rec)

diff --git a/process/stats.hpp b/process/stats.hpp
@@ -71,8 +71,8 @@ class StatsPlugin : public ProcessPlugin
    std::string get_name() const { return "stats"; }
    ProcessPlugin *copy();
 
-   int post_create(Flow &rec, const Packet &pkt);
-   int post_update(Flow &rec, const Packet &pkt);
+   ProcessPlugin::FlowAction post_create(Flow &rec, const Packet &pkt);
+   ProcessPlugin::FlowAction post_update(Flow &rec, const Packet &pkt);
    void pre_export(Flow &rec);
    void finish(bool print_stats);