fix: review

Author: artemmufazalov

src/components/ShardsTable/ShardsTable.tsx

@@ -19,10 +19,10 @@ export function ShardsTable({columnsIds, schemaPath, database, ...props}: Shards
     const columns = React.useMemo(
         () =>
             columnsIds
+                .filter((id) => id in shardsColumnIdToGetColumn)
                 .map((id) => {
-                    return shardsColumnIdToGetColumn[id]({database, schemaPath});
-                })
-                .map((column) => {
+                    const column = shardsColumnIdToGetColumn[id]({database, schemaPath});
+
                     return {
                         ...column,
                         sortable: isSortableTopShardsColumn(column.name),

src/components/ShardsTable/columns.tsx

@@ -9,6 +9,7 @@ import {LinkToSchemaObject} from '../LinkToSchemaObject/LinkToSchemaObject';
 import {TabletNameWrapper} from '../TabletNameWrapper/TabletNameWrapper';
 import {UsageLabel} from '../UsageLabel/UsageLabel';
 
+import type {TopShardsColumnId} from './constants';
 import {TOP_SHARDS_COLUMNS_IDS, TOP_SHARDS_COLUMNS_TITLES} from './constants';
 import type {GetShardsColumn} from './types';
 import {prepareDateTimeValue} from './utils';
@@ -100,7 +101,7 @@ export const getIntervalEndColumn: GetShardsColumn = () => {
     };
 };
 
-export const shardsColumnIdToGetColumn = {
+export const shardsColumnIdToGetColumn: Record<TopShardsColumnId, GetShardsColumn> = {
     [TOP_SHARDS_COLUMNS_IDS.Path]: getPathColumn,
     [TOP_SHARDS_COLUMNS_IDS.DataSize]: getDataSizeColumn,
     [TOP_SHARDS_COLUMNS_IDS.TabletId]: getTabletIdColumn,

src/components/ShardsTable/types.ts

@@ -2,7 +2,6 @@ import type {Column} from '@gravity-ui/react-data-table';
 
 import type {KeyValueRow} from '../../types/api/query';
 
-export type GetShardsColumn = (params: {
-    database: string;
-    schemaPath?: string;
-}) => Column<KeyValueRow>;
+export type ShardsColumn = Column<KeyValueRow>;
+
+export type GetShardsColumn = (params: {database: string; schemaPath?: string}) => ShardsColumn;