@tanstack/table-core 9.0.0-beta.51 → 9.0.0-beta.52

package/skills/core/SKILL.md

@@ -5,7 +5,7 @@ description: >
 metadata:
   type: core
   library: '@tanstack/table-core'
-  library_version: '9.0.0-beta.51'
+  library_version: '9.0.0-beta.52'
 sources:
   - 'TanStack/table:docs/overview.md'
   - 'TanStack/table:docs/guide/tables.md'

package/skills/custom-features/SKILL.md

@@ -5,7 +5,7 @@ description: >
 metadata:
   type: sub-skill
   library: '@tanstack/table-core'
-  library_version: '9.0.0-beta.51'
+  library_version: '9.0.0-beta.52'
 requires: ['core', 'table-features', 'typescript']
 sources:
   - 'TanStack/table:docs/framework/react/guide/custom-features.md'

package/skills/expanding/SKILL.md

@@ -6,7 +6,7 @@ metadata:
   {
     type: sub-skill,
     library: '@tanstack/table-core',
-    library_version: '9.0.0-beta.51',
+    library_version: '9.0.0-beta.52',
   }
 requires: ['core', 'table-features', 'client-vs-server']
 sources:

package/skills/global-filtering/SKILL.md

@@ -6,7 +6,7 @@ metadata:
   {
     type: sub-skill,
     library: '@tanstack/table-core',
-    library_version: '9.0.0-beta.51',
+    library_version: '9.0.0-beta.52',
   }
 requires: ['core', 'table-features', 'client-vs-server', 'column-filtering']
 sources:

package/skills/grouping/SKILL.md

@@ -6,7 +6,7 @@ metadata:
   {
     type: sub-skill,
     library: '@tanstack/table-core',
-    library_version: '9.0.0-beta.51',
+    library_version: '9.0.0-beta.52',
   }
 requires: ['core', 'table-features', 'client-vs-server']
 sources:

package/skills/migrate-v8-to-v9/SKILL.md

@@ -5,7 +5,7 @@ description: >
 metadata:
   type: lifecycle
   library: '@tanstack/table-core'
-  library_version: '9.0.0-beta.51'
+  library_version: '9.0.0-beta.52'
 requires: ['core', 'table-features', 'typescript']
 sources:
   - 'TanStack/table:docs/framework/react/guide/migrating.md'

package/skills/pagination/SKILL.md

@@ -6,7 +6,7 @@ metadata:
   {
     type: sub-skill,
     library: '@tanstack/table-core',
-    library_version: '9.0.0-beta.51',
+    library_version: '9.0.0-beta.52',
   }
 requires: ['core', 'table-features', 'client-vs-server']
 sources:

package/skills/row-pinning/SKILL.md

@@ -6,7 +6,7 @@ metadata:
   {
     type: sub-skill,
     library: '@tanstack/table-core',
-    library_version: '9.0.0-beta.51',
+    library_version: '9.0.0-beta.52',
   }
 requires: ['core', 'table-features']
 sources:

package/skills/row-selection/SKILL.md

@@ -6,7 +6,7 @@ metadata:
   {
     type: sub-skill,
     library: '@tanstack/table-core',
-    library_version: '9.0.0-beta.51',
+    library_version: '9.0.0-beta.52',
   }
 requires: ['core', 'table-features']
 sources:

package/skills/sorting/SKILL.md

@@ -6,7 +6,7 @@ metadata:
   {
     type: sub-skill,
     library: '@tanstack/table-core',
-    library_version: '9.0.0-beta.51',
+    library_version: '9.0.0-beta.52',
   }
 requires: ['core', 'table-features', 'client-vs-server']
 sources:

package/skills/table-features/SKILL.md

@@ -5,7 +5,7 @@ description: >
 metadata:
   type: sub-skill
   library: '@tanstack/table-core'
-  library_version: '9.0.0-beta.51'
+  library_version: '9.0.0-beta.52'
 requires: ['core']
 sources:
   - 'TanStack/table:docs/guide/row-models.md'

package/skills/typescript/SKILL.md

@@ -5,7 +5,7 @@ description: >
 metadata:
   type: sub-skill
   library: '@tanstack/table-core'
-  library_version: '9.0.0-beta.51'
+  library_version: '9.0.0-beta.52'
 requires: ['core', 'table-features']
 sources:
   - 'TanStack/table:docs/guide/helpers.md'

package/src/features/column-grouping/createGroupedRowModel.ts

@@ -5,10 +5,10 @@ import { table_autoResetExpanded } from '../row-expanding/rowExpandingFeature.ut
 import { table_autoResetPageIndex } from '../row-pagination/rowPaginationFeature.utils'
 import {
   aggregateColumnValue,
-  normalizeAggregationRows,
+  normalizeUniqueAggregationRows,
 } from '../row-aggregation/rowAggregationFeature.utils'
-import { row_getGroupingValue } from './columnGroupingFeature.utils'
 import type { Row_ColumnGrouping } from './columnGroupingFeature.types'
+import type { Column_Internal } from '../../types/Column'
 import type { TableFeatures } from '../../types/TableFeatures'
 import type { RowModel } from '../../core/row-models/coreRowModelsFeature.types'
 import type { Table, Table_Internal } from '../../types/Table'
@@ -103,7 +103,7 @@ function _createGroupedRowModel<
     const columnId = existingGrouping[depth] as string
 
     // Group the rows together for this level
-    const rowGroupsMap = groupBy(rows, columnId)
+    const rowGroupsMap = groupBy(table, rows, columnId)
 
     // Perform aggregations for each group
     const aggregatedGroupedRows = Array.from(rowGroupsMap.entries()).map(
@@ -118,7 +118,13 @@ function _createGroupedRowModel<
           subRow.parentId = id
         })
 
-        const leafRows = normalizeAggregationRows(groupedRows, Infinity)
+        // Rows produced by groupBy are disjoint members of the pre-grouped
+        // row tree, so the duplicate-id guard is unnecessary; flat groups
+        // reuse the partition array as-is.
+        const leafRows = normalizeUniqueAggregationRows(
+          groupedRows,
+          Infinity,
+        ) as Array<Row<TFeatures, TData>>
 
         const row = constructRow(
           table,
@@ -171,6 +177,7 @@ function _createGroupedRowModel<
               column,
               groupingRow: row,
               rows: groupedRows,
+              uniqueRows: true,
             })
             return cache[colId]
           },
@@ -203,19 +210,49 @@ function _createGroupedRowModel<
 }
 
 function groupBy<TFeatures extends TableFeatures, TData extends RowData = any>(
+  table: Table_Internal<TFeatures, TData>,
   rows: Array<Row<TFeatures, TData>>,
   columnId: string,
 ) {
   const groupMap = new Map<any, Array<Row<TFeatures, TData>>>()
 
-  return rows.reduce((map, row) => {
-    const resKey = `${row_getGroupingValue(row, columnId)}`
-    const previous = map.get(resKey)
+  // Resolve the column once instead of per row: `table.getColumn` goes
+  // through the memoized-API dispatcher, which is far too expensive to sit
+  // inside this per-row loop. The branches below mirror
+  // `row_getGroupingValue`'s caching contract exactly.
+  const column = table_getColumn(table, columnId) as
+    | Column_Internal<TFeatures, TData, unknown>
+    | undefined
+  const getGroupingValue = column?.columnDef.getGroupingValue
+
+  for (let i = 0; i < rows.length; i++) {
+    const row = rows[i]!
+    let groupingValue
+    if (getGroupingValue) {
+      const cache = (row as any)._groupingValuesCache as
+        | Record<string, unknown>
+        | undefined
+      if (cache && hasOwn(cache, columnId)) {
+        groupingValue = cache[columnId]
+      } else if (cache) {
+        groupingValue = cache[columnId] = getGroupingValue(
+          row.original,
+          row.index,
+          row,
+        )
+      }
+    } else {
+      groupingValue = row.getValue(columnId)
+    }
+
+    const resKey = `${groupingValue}`
+    const previous = groupMap.get(resKey)
     if (!previous) {
-      map.set(resKey, [row])
+      groupMap.set(resKey, [row])
     } else {
       previous.push(row)
     }
-    return map
-  }, groupMap)
+  }
+
+  return groupMap
 }

package/src/features/row-aggregation/aggregationFns.ts

@@ -1,5 +1,4 @@
 import { constructAggregationFn } from './rowAggregationFeature.types'
-import type { AggregationContext } from './rowAggregationFeature.types'
 
 type RangeValue = Date | number
 
@@ -23,19 +22,9 @@ function compareRangeValues(left: RangeValue, right: RangeValue) {
   return leftValue - rightValue
 }
 
-function collectRangeValues(context: AggregationContext<any, any, unknown>) {
-  const values: Array<RangeValue> = []
-  let kind: 'date' | 'number' | undefined
-
-  for (let i = 0; i < context.rows.length; i++) {
-    const value = context.getValue(context.rows[i]!)
-    const valueKind = getRangeKind(value)
-    if (!valueKind) continue
-    kind ??= valueKind
-    if (valueKind === kind) values.push(value as RangeValue)
-  }
-
-  return values
+/** Comparable representation of a range value; `Date`s compare by time. */
+function toRangeNumber(value: RangeValue): number {
+  return value instanceof Date ? value.getTime() : value
 }
 
 /**
@@ -49,9 +38,10 @@ export const aggregationFn_sum = constructAggregationFn<
   number
 >({
   aggregate: (context) => {
+    const rows = context.rows
     let sum = 0
-    for (let i = 0; i < context.rows.length; i++) {
-      const value = context.getValue(context.rows[i]!)
+    for (let i = 0; i < rows.length; i++) {
+      const value = context.getValue(rows[i]!)
       sum += typeof value === 'number' ? value : 0
     }
     return sum
@@ -77,10 +67,23 @@ export const aggregationFn_min = constructAggregationFn<
   RangeValue | undefined
 >({
   aggregate: (context) => {
-    const values = collectRangeValues(context)
-    let result = values[0]
-    for (let i = 1; i < values.length; i++) {
-      if (compareRangeValues(values[i]!, result!) < 0) result = values[i]
+    const rows = context.rows
+    let kind: 'date' | 'number' | undefined
+    let result: RangeValue | undefined
+    let resultNumber = 0
+    for (let i = 0; i < rows.length; i++) {
+      const value = context.getValue(rows[i]!)
+      const valueKind = getRangeKind(value)
+      if (!valueKind || (kind !== undefined && valueKind !== kind)) continue
+      const valueNumber = toRangeNumber(value as RangeValue)
+      if (kind === undefined) {
+        kind = valueKind
+        result = value as RangeValue
+        resultNumber = valueNumber
+      } else if (valueNumber - resultNumber < 0) {
+        result = value as RangeValue
+        resultNumber = valueNumber
+      }
     }
     return result
   },
@@ -113,10 +116,23 @@ export const aggregationFn_max = constructAggregationFn<
   RangeValue | undefined
 >({
   aggregate: (context) => {
-    const values = collectRangeValues(context)
-    let result = values[0]
-    for (let i = 1; i < values.length; i++) {
-      if (compareRangeValues(values[i]!, result!) > 0) result = values[i]
+    const rows = context.rows
+    let kind: 'date' | 'number' | undefined
+    let result: RangeValue | undefined
+    let resultNumber = 0
+    for (let i = 0; i < rows.length; i++) {
+      const value = context.getValue(rows[i]!)
+      const valueKind = getRangeKind(value)
+      if (!valueKind || (kind !== undefined && valueKind !== kind)) continue
+      const valueNumber = toRangeNumber(value as RangeValue)
+      if (kind === undefined) {
+        kind = valueKind
+        result = value as RangeValue
+        resultNumber = valueNumber
+      } else if (valueNumber - resultNumber > 0) {
+        result = value as RangeValue
+        resultNumber = valueNumber
+      }
     }
     return result
   },
@@ -150,15 +166,33 @@ export const aggregationFn_extent = constructAggregationFn<
   [RangeValue | undefined, RangeValue | undefined]
 >({
   aggregate: (context) => {
-    const values = collectRangeValues(context)
-    if (!values.length) return [undefined, undefined]
-    let min = values[0]!
-    let max = values[0]!
-    for (let i = 1; i < values.length; i++) {
-      const value = values[i]!
-      if (compareRangeValues(value, min) < 0) min = value
-      if (compareRangeValues(value, max) > 0) max = value
+    const rows = context.rows
+    let kind: 'date' | 'number' | undefined
+    let min: RangeValue | undefined
+    let max: RangeValue | undefined
+    let minNumber = 0
+    let maxNumber = 0
+    for (let i = 0; i < rows.length; i++) {
+      const value = context.getValue(rows[i]!)
+      const valueKind = getRangeKind(value)
+      if (!valueKind || (kind !== undefined && valueKind !== kind)) continue
+      const valueNumber = toRangeNumber(value as RangeValue)
+      if (kind === undefined) {
+        kind = valueKind
+        min = max = value as RangeValue
+        minNumber = maxNumber = valueNumber
+      } else {
+        if (valueNumber - minNumber < 0) {
+          min = value as RangeValue
+          minNumber = valueNumber
+        }
+        if (valueNumber - maxNumber > 0) {
+          max = value as RangeValue
+          maxNumber = valueNumber
+        }
+      }
     }
+    if (kind === undefined) return [undefined, undefined]
     return [min, max]
   },
   merge: ({ subRowResults }) => {
@@ -205,10 +239,11 @@ export const aggregationFn_mean = constructAggregationFn<
   number | undefined
 >({
   aggregate: (context) => {
+    const rows = context.rows
     let count = 0
     let sum = 0
-    for (let i = 0; i < context.rows.length; i++) {
-      const value = context.getValue(context.rows[i]!)
+    for (let i = 0; i < rows.length; i++) {
+      const value = context.getValue(rows[i]!)
       if (value == null) continue
       const numberValue = typeof value === 'number' ? value : +value
       if (!Number.isNaN(numberValue)) {
@@ -231,10 +266,11 @@ export const aggregationFn_median = constructAggregationFn<
   number | undefined
 >({
   aggregate: (context) => {
-    if (!context.rows.length) return undefined
-    const values = new Array<number>(context.rows.length)
-    for (let i = 0; i < context.rows.length; i++) {
-      const value = context.getValue(context.rows[i]!)
+    const rows = context.rows
+    if (!rows.length) return undefined
+    const values = new Array<number>(rows.length)
+    for (let i = 0; i < rows.length; i++) {
+      const value = context.getValue(rows[i]!)
       if (typeof value !== 'number') return undefined
       values[i] = value
     }
@@ -254,9 +290,10 @@ export const aggregationFn_unique = constructAggregationFn<
   Array<unknown>
 >({
   aggregate: (context) => {
+    const rows = context.rows
     const values = new Set<unknown>()
-    for (let i = 0; i < context.rows.length; i++) {
-      values.add(context.getValue(context.rows[i]!))
+    for (let i = 0; i < rows.length; i++) {
+      values.add(context.getValue(rows[i]!))
     }
     return Array.from(values)
   },
@@ -270,9 +307,10 @@ export const aggregationFn_uniqueCount = constructAggregationFn<
   number
 >({
   aggregate: (context) => {
+    const rows = context.rows
     const values = new Set<unknown>()
-    for (let i = 0; i < context.rows.length; i++) {
-      values.add(context.getValue(context.rows[i]!))
+    for (let i = 0; i < rows.length; i++) {
+      values.add(context.getValue(rows[i]!))
     }
     return values.size
   },

package/src/features/row-aggregation/rowAggregationFeature.ts

@@ -1,4 +1,4 @@
-import { assignPrototypeAPIs, makeObjectMap } from '../../utils'
+import { assignPrototypeAPIs } from '../../utils'
 import {
   cell_getIsAggregated,
   column_getAggregationFns,
@@ -53,8 +53,4 @@ export const rowAggregationFeature: TableFeature = {
     ;(column as any)._aggregationValueCache = undefined
     ;(column as any)._resolvedAggregationFnsCache = undefined
   },
-
-  initRowInstanceData: (row) => {
-    ;(row as any)._aggregationValuesCache = makeObjectMap()
-  },
 }

package/src/features/row-aggregation/rowAggregationFeature.types.ts

@@ -302,8 +302,8 @@ export interface Cell_Aggregation {
 
 /** Internal per-row cache used while grouped aggregates are evaluated. */
 export interface Row_Aggregation {
-  /** Cached aggregate results keyed by column id. */
-  _aggregationValuesCache: Record<string, unknown>
+  /** Cached aggregate results keyed by column id; created lazily on grouped rows. */
+  _aggregationValuesCache?: Record<string, unknown>
 }
 
 /** Values passed to a column-level aggregation-value provider. */

package/src/features/row-aggregation/rowAggregationFeature.utils.ts

@@ -95,6 +95,52 @@ export function normalizeAggregationRows<
   return result
 }
 
+/**
+ * Frontier selection for rows that are distinct nodes of a single row tree —
+ * the row models the table builds itself. Skips `normalizeAggregationRows`'
+ * duplicate-id guard (disjoint subtrees cannot revisit a row) and returns
+ * `rows` unchanged when no row descends, so the default `maxDepth: 0` case
+ * costs nothing per aggregation.
+ */
+export function normalizeUniqueAggregationRows<
+  TFeatures extends TableFeatures,
+  TData extends RowData,
+>(
+  rows: ReadonlyArray<Row<TFeatures, TData>>,
+  maxDepth = 0,
+): ReadonlyArray<Row<TFeatures, TData>> {
+  const normalizedMaxDepth = resolveMaxAggregationDepth(maxDepth)
+
+  let needsDescent = false
+  if (normalizedMaxDepth > 0) {
+    for (let i = 0; i < rows.length; i++) {
+      if (rows[i]!.subRows.length) {
+        needsDescent = true
+        break
+      }
+    }
+  }
+  if (!needsDescent) return rows
+
+  const result: Array<Row<TFeatures, TData>> = []
+
+  const visit = (row: Row<TFeatures, TData>, depth: number) => {
+    if (row.subRows.length && depth < normalizedMaxDepth) {
+      for (let i = 0; i < row.subRows.length; i++) {
+        visit(row.subRows[i]!, depth + 1)
+      }
+      return
+    }
+    result.push(row)
+  }
+
+  for (let i = 0; i < rows.length; i++) {
+    visit(rows[i]!, 0)
+  }
+
+  return result
+}
+
 function getAutoAggregationFnName(
   value: unknown,
 ): 'extent' | 'sum' | undefined {
@@ -267,13 +313,21 @@ export function aggregateColumnValue<
   column: Column<TFeatures, TData, unknown>
   groupingRow?: Row<TFeatures, TData>
   rows: ReadonlyArray<Row<TFeatures, TData>>
+  /**
+   * Marks `rows` as distinct nodes of a single row tree (rows the table's own
+   * row models produced), enabling frontier selection without the
+   * duplicate-id guard. Caller-supplied row arrays must omit this.
+   */
+  uniqueRows?: boolean
 }): unknown {
-  const { subRows, column, groupingRow, rows } = args
+  const { subRows, column, groupingRow, rows, uniqueRows } = args
   const internalColumn = column as Column_Internal<TFeatures, TData, unknown>
   const maxDepth = resolveMaxAggregationDepth(
     args.maxDepth ?? internalColumn.columnDef.maxAggregationDepth,
   )
-  const aggregationRows = normalizeAggregationRows(rows, maxDepth)
+  const aggregationRows = uniqueRows
+    ? normalizeUniqueAggregationRows(rows, maxDepth)
+    : normalizeAggregationRows(rows, maxDepth)
   const entries = column_getAggregationFns(internalColumn)
   const isMultiple = Array.isArray(internalColumn.columnDef.aggregationFn)
   const canMerge =
@@ -284,6 +338,8 @@ export function aggregateColumnValue<
         (row as any).groupingColumnId !== column.id,
     )
 
+  const getValue = (row: Row<TFeatures, TData>) => row.getValue(column.id)
+
   const execute = (entry: ResolvedAggregationFn<TFeatures, TData>) => {
     const definition = entry.aggregationFn
     if (!definition) return undefined
@@ -292,7 +348,7 @@ export function aggregateColumnValue<
       ...(subRows ? { subRows } : {}),
       column,
       columnId: column.id,
-      getValue: (row: Row<TFeatures, TData>) => row.getValue(column.id),
+      getValue,
       ...(groupingRow ? { groupingRow } : {}),
       maxDepth,
       rows: aggregationRows,
@@ -378,6 +434,7 @@ export function column_getAggregationValue<
     column: column as any,
     maxDepth: resolvedMaxDepth,
     rows: model.rows,
+    uniqueRows: true,
   })
   ;(column as any)._aggregationValueCache = {
     aggregationFnOption,