primevue 3.44.0 → 3.46.0

package/datatable/TableHeader.vue

@@ -128,7 +128,7 @@
 
 <script>
 import BaseComponent from 'primevue/basecomponent';
-import { ObjectUtils } from 'primevue/utils';
+import { HelperSet, ObjectUtils } from 'primevue/utils';
 import { mergeProps } from 'vue';
 import ColumnFilter from './ColumnFilter.vue';
 import HeaderCell from './HeaderCell.vue';
@@ -226,6 +226,22 @@ export default {
             default: null
         }
     },
+    provide() {
+        return {
+            $rows: this.d_headerRows,
+            $columns: this.d_headerColumns
+        };
+    },
+    data() {
+        return {
+            d_headerRows: new HelperSet({ type: 'Row' }),
+            d_headerColumns: new HelperSet({ type: 'Column' })
+        };
+    },
+    beforeUnmount() {
+        this.d_headerRows.clear();
+        this.d_headerColumns.clear();
+    },
     methods: {
         columnProp(col, prop) {
             return ObjectUtils.getVNodeProp(col, prop);
@@ -292,33 +308,10 @@ export default {
             return [this.columnProp(column, 'filterHeaderStyle'), this.columnProp(column, 'style')];
         },
         getHeaderRows() {
-            let rows = [];
-
-            let columnGroup = this.columnGroup;
-
-            if (columnGroup.children && columnGroup.children.default) {
-                for (let child of columnGroup.children.default()) {
-                    if (child.type.name === 'Row') {
-                        rows.push(child);
-                    } else if (child.children && child.children instanceof Array) {
-                        rows = child.children;
-                    }
-                }
-
-                return rows;
-            }
+            return this.d_headerRows?.get(this.columnGroup, this.columnGroup.children);
         },
         getHeaderColumns(row) {
-            let cols = [];
-
-            if (row.children && row.children.default) {
-                row.children.default().forEach((child) => {
-                    if (child.children && child.children instanceof Array) cols = [...cols, ...child.children];
-                    else if (child.type.name === 'Column') cols.push(child);
-                });
-
-                return cols;
-            }
+            return this.d_headerColumns?.get(row, row.children);
         }
     },
     computed: {