fdsdump: store output_limit in View

Author: sedmicha

src/tools/fdsdump/src/aggregator/view.cpp

@@ -77,6 +77,13 @@ View::find_field(const std::string &name)
     return nullptr;
 }
 
+
+void
+View::set_output_limit(size_t n)
+{
+    m_output_limit = n;
+}
+
 Value &
 View::access_field(const Field &field, uint8_t *record_ptr) const
 {

src/tools/fdsdump/src/aggregator/view.hpp

@@ -354,6 +354,11 @@ class View {
      *
      * @return true if key1 comes before key2 in an ordered sequence, else false
      */
+    void
+    set_output_limit(size_t n);
+
+    size_t output_limit() const { return m_output_limit; }
+
     bool
     ordered_before(uint8_t *key1, uint8_t *key2) const;
 
@@ -402,6 +407,7 @@ class View {
     size_t m_value_count = 0;
     size_t m_key_size = 0;
     size_t m_value_size = 0;
+    size_t m_output_limit = 0; // 0 = no limit
     std::vector<OrderField> m_order_fields;
     bool m_has_inout_fields = false;
     bool m_is_fixed_size = true;

src/tools/fdsdump/src/aggregator/viewFactory.cpp

@@ -288,10 +288,11 @@ View
 ViewFactory::create_view(
     const std::string &key_def,
     const std::string &value_def,
-    const std::string &order_def)
+    const std::string &order_def,
+    unsigned int output_limit)
 {
     View view;
-    create_view(view, key_def, value_def, order_def);
+    create_view(view, key_def, value_def, order_def, output_limit);
     return view;
 }
 
@@ -300,7 +301,8 @@ ViewFactory::create_view(
     View &view,
     const std::string &key_def,
     const std::string &value_def,
-    const std::string &order_def)
+    const std::string &order_def,
+    unsigned int output_limit)
 {
     for (auto def : split_args(key_def)) {
         string_trim(def);
@@ -368,6 +370,8 @@ ViewFactory::create_view(
             view.m_order_fields.push_back({field, dir});
         }
     }
+
+    view.m_output_limit = output_limit;
 }
 
 } // namespace aggregator

src/tools/fdsdump/src/aggregator/viewFactory.hpp

@@ -40,10 +40,15 @@ class ViewFactory {
      * @param key_def  The definition of the key fields as specified on the command line
      * @param value_def  The definition of the value fields as specified on the command line
      * @param order_def  The definition of the order-by fields as specified on the command line
+     * @param output_limit  The number of items that will be outputted using this view (0 = no limit)
      *
      * @return The view
      */
-    static View create_view(const std::string &key_def, const std::string &value_def, const std::string &order_def);
+    static View create_view(
+        const std::string &key_def,
+        const std::string &value_def,
+        const std::string &order_def,
+        unsigned int output_limit = 0);
 
 private:
     static std::unique_ptr<Field> create_elem_or_alias(const std::string &def);
@@ -56,7 +61,7 @@ class ViewFactory {
 
     static std::unique_ptr<Field> parse_dir_field(const std::string &def);
 
-    static void create_view(View &view, const std::string &key_def, const std::string &value_def, const std::string &order_def);
+    static void create_view(View &view, const std::string &key_def, const std::string &value_def, const std::string &order_def, unsigned int output_limit);
 };
 
 } // namespace aggregator