Adjust Stats plugin

Author: Zadamsa

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)

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);