Add an initialization method in SchemaReader to pass the column map t…

…o FilterClass
y-scope · Jan 31, 2025 · 60e0ee3 · 60e0ee3
1 parent ebe30e1
commit 60e0ee3
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/components/core/src/clp_s/SchemaReader.cpp b/components/core/src/clp_s/SchemaReader.cpp
@@ -259,6 +259,10 @@ void SchemaReader::initialize_filter(FilterClass* filter) {
     filter->init(this, m_schema_id, m_columns);
 }
 
+void SchemaReader::initialize_filter_with_column_map(FilterClass* filter) {
+    filter->init(this, m_schema_id, m_column_map);
+}
+
 void SchemaReader::generate_local_tree(int32_t global_id) {
     std::stack<int32_t> global_id_stack;
     global_id_stack.emplace(global_id);

diff --git a/components/core/src/clp_s/SchemaReader.hpp b/components/core/src/clp_s/SchemaReader.hpp
@@ -32,6 +32,18 @@ class FilterClass {
             std::vector<BaseColumnReader*> const& column_readers
     ) = 0;
 
+    /**
+     * Initializes the filter with a column map
+     * @param reader
+     * @param schema_id
+     * @param column_map
+     */
+    virtual void init(
+            SchemaReader* reader,
+            int32_t schema_id,
+            std::unordered_map<int32_t, BaseColumnReader*> const& column_map
+    ) {}
+
     /**
      * Filters the message
      * @param cur_message
@@ -180,6 +192,12 @@ class SchemaReader {
      */
     void initialize_filter(FilterClass* filter);
 
+    /**
+     * Initializes the filter with a column map
+     * @param filter
+     */
+    void initialize_filter_with_column_map(FilterClass* filter);
+
     /**
      * Marks a column as timestamp
      * @param column_reader