@parca/profile 0.19.138 → 0.19.140

package/src/ProfileView/hooks/useVisualizationState.ts

@@ -13,13 +13,8 @@
 
 import {useCallback, useMemo} from 'react';
 
-import {
-  JSONParser,
-  JSONSerializer,
-  useURLState,
-  useURLStateBatch,
-  useURLStateCustom,
-} from '@parca/components';
+import {useQueryState} from 'nuqs';
+
 import {USER_PREFERENCES, useUserPreference} from '@parca/hooks';
 
 import {
@@ -30,12 +25,19 @@ import {
   FIELD_MAPPING_FILE,
 } from '../../ProfileFlameGraph/FlameGraphArrow';
 import {CurrentPathFrame} from '../../ProfileFlameGraph/FlameGraphArrow/utils';
+import {
+  flamechartDimensionParser,
+  groupByParser,
+  jsonParser,
+  stringParam,
+} from '../../hooks/urlParsers';
+import {useColorBy} from '../../hooks/useColorBy';
 import {useResetFlameGraphState} from './useResetFlameGraphState';
 
 export const useVisualizationState = (): {
   curPathArrow: CurrentPathFrame[];
   setCurPathArrow: (path: CurrentPathFrame[]) => void;
-  colorStackLegend: string | undefined;
+  colorStackLegend: string | null;
   colorBy: string;
   setColorBy: (colorBy: string) => void;
   groupBy: string[];
@@ -44,46 +46,52 @@ export const useVisualizationState = (): {
   setGroupByLabels: (labels: string[]) => void;
   flamechartDimension: string[];
   setFlamechartDimension: (labels: string[]) => void;
-  sandwichFunctionName: string | undefined;
-  setSandwichFunctionName: (sandwichFunctionName: string | undefined) => void;
+  sandwichFunctionName: string | null;
+  setSandwichFunctionName: (sandwichFunctionName: string | null) => void;
   resetSandwichFunctionName: () => void;
   alignFunctionName: string;
   setAlignFunctionName: (align: string) => void;
 } => {
-  const [colorByPreference, setColorByPreference] = useUserPreference<string>(
-    USER_PREFERENCES.COLOR_BY.key
-  );
   const [alignFunctionNamePreference, setAlignFunctionNamePreference] = useUserPreference<string>(
     USER_PREFERENCES.ALIGN_FUNCTION_NAME.key
   );
 
-  const [curPathArrow, setCurPathArrow] = useURLStateCustom<CurrentPathFrame[]>('cur_path', {
-    parse: JSONParser<CurrentPathFrame[]>,
-    stringify: JSONSerializer,
-    defaultValue: '[]',
-  });
-  const [colorStackLegend] = useURLState<string | undefined>('color_stack_legend');
-  const [colorBy, setStoreColorBy] = useURLState('color_by', {
-    defaultValue: colorByPreference,
-  });
-  const [alignFunctionName, setStoreAlignFunctionName] = useURLState('align_function_name', {
-    defaultValue: alignFunctionNamePreference,
-  });
-  const [groupBy, setStoreGroupBy] = useURLState<string[]>('group_by', {
-    defaultValue: [FIELD_FUNCTION_NAME],
-    alwaysReturnArray: true,
-  });
-  const [sandwichFunctionName, setSandwichFunctionName] = useURLState<string | undefined>(
-    'sandwich_function_name'
+  const [curPathArrow, setRawCurPathArrow] = useQueryState(
+    'cur_path',
+    jsonParser<CurrentPathFrame[]>().withDefault([])
+  );
+  const setCurPathArrow = useCallback(
+    (path: CurrentPathFrame[]) => {
+      void setRawCurPathArrow(path);
+    },
+    [setRawCurPathArrow]
+  );
+  const [colorStackLegend] = useQueryState('color_stack_legend', stringParam);
+  const {colorBy, setColorBy} = useColorBy();
+  const [alignFunctionNameRaw, setStoreAlignFunctionName] = useQueryState(
+    'align_function_name',
+    stringParam
+  );
+  const alignFunctionName = alignFunctionNameRaw ?? alignFunctionNamePreference ?? 'left';
+  const [groupBy, setStoreGroupBy] = useQueryState(
+    'group_by',
+    groupByParser.withDefault([FIELD_FUNCTION_NAME])
+  );
+  const [sandwichFunctionName, setRawSandwichFunctionName] = useQueryState(
+    'sandwich_function_name',
+    stringParam
   );
-  const [flamechartDimension, setStoreFlamechartDimension] = useURLState<string[]>(
+  const setSandwichFunctionName = useCallback(
+    (name: string | null) => {
+      void setRawSandwichFunctionName(name);
+    },
+    [setRawSandwichFunctionName]
+  );
+  const [flamechartDimension, setStoreFlamechartDimension] = useQueryState(
     'flamechart_dimension',
-    {
-      alwaysReturnArray: true,
-    }
+    flamechartDimensionParser.withDefault([])
   );
   const resetFlameGraphState = useResetFlameGraphState();
-  const batchUpdates = useURLStateBatch();
 
   const levelsOfProfiling = useMemo(
     () => [
@@ -97,62 +105,46 @@ export const useVisualizationState = (): {
 
   const setGroupBy = useCallback(
     (keys: string[]): void => {
-      setStoreGroupBy(keys);
+      void setStoreGroupBy(keys);
     },
     [setStoreGroupBy]
   );
 
   const toggleGroupBy = useCallback(
     (key: string): void => {
-      // Batch updates to combine setGroupBy + resetFlameGraphState into single URL navigation
-      batchUpdates(() => {
-        if (groupBy.includes(key)) {
-          setGroupBy(groupBy.filter(v => v !== key)); // remove
-        } else {
-          const filteredGroupBy = groupBy.filter(item => !levelsOfProfiling.includes(item));
-          setGroupBy([...filteredGroupBy, key]); // add
-        }
-
-        resetFlameGraphState();
-      });
+      if (groupBy.includes(key)) {
+        setGroupBy(groupBy.filter(v => v !== key));
+      } else {
+        const filteredGroupBy = groupBy.filter(item => !levelsOfProfiling.includes(item));
+        setGroupBy([...filteredGroupBy, key]);
+      }
+      resetFlameGraphState();
     },
-    [groupBy, setGroupBy, levelsOfProfiling, resetFlameGraphState, batchUpdates]
+    [groupBy, setGroupBy, levelsOfProfiling, resetFlameGraphState]
   );
 
   const setGroupByLabels = useCallback(
     (labels: string[]): void => {
-      // Batch updates to combine setGroupBy + resetFlameGraphState into single URL navigation
-      batchUpdates(() => {
-        setGroupBy(groupBy.filter(l => !l.startsWith(`${FIELD_LABELS}.`)).concat(labels));
-
-        resetFlameGraphState();
-      });
+      setGroupBy(groupBy.filter(l => !l.startsWith(`${FIELD_LABELS}.`)).concat(labels));
+      resetFlameGraphState();
     },
-    [groupBy, setGroupBy, resetFlameGraphState, batchUpdates]
+    [groupBy, setGroupBy, resetFlameGraphState]
   );
 
   const setFlamechartDimension = useCallback(
     (labels: string[]): void => {
-      setStoreFlamechartDimension(labels.filter(l => l.startsWith(`${FIELD_LABELS}.`)));
+      void setStoreFlamechartDimension(labels.filter(l => l.startsWith(`${FIELD_LABELS}.`)));
     },
     [setStoreFlamechartDimension]
   );
 
   const resetSandwichFunctionName = useCallback((): void => {
-    setSandwichFunctionName(undefined);
+    setSandwichFunctionName(null);
   }, [setSandwichFunctionName]);
 
-  const setColorBy = useCallback(
-    (value: string): void => {
-      setStoreColorBy(value);
-      setColorByPreference(value);
-    },
-    [setStoreColorBy, setColorByPreference]
-  );
-
   const setAlignFunctionName = useCallback(
     (value: string): void => {
-      setStoreAlignFunctionName(value);
+      void setStoreAlignFunctionName(value);
       setAlignFunctionNamePreference(value);
     },
     [setStoreAlignFunctionName, setAlignFunctionNamePreference]
@@ -162,7 +154,7 @@ export const useVisualizationState = (): {
     curPathArrow,
     setCurPathArrow,
     colorStackLegend,
-    colorBy: (colorBy as string) ?? '',
+    colorBy,
     setColorBy,
     groupBy,
     setGroupBy,
@@ -173,7 +165,7 @@ export const useVisualizationState = (): {
     sandwichFunctionName,
     setSandwichFunctionName,
     resetSandwichFunctionName,
-    alignFunctionName: (alignFunctionName as string) ?? 'left',
+    alignFunctionName,
     setAlignFunctionName,
   };
 };

package/src/ProfileViewWithData.tsx

@@ -13,15 +13,10 @@
 
 import {useEffect, useMemo, useState} from 'react';
 
+import {useQueryState} from 'nuqs';
+
 import {QueryRequest_ReportType, QueryServiceClient} from '@parca/client';
-import {
-  NumberParser,
-  NumberSerializer,
-  useGrpcMetadata,
-  useParcaContext,
-  useURLState,
-  useURLStateCustom,
-} from '@parca/components';
+import {useGrpcMetadata, useParcaContext} from '@parca/components';
 import {saveAsBlob} from '@parca/utilities';
 
 import {validateFlameChartQuery} from './ProfileFlameGraph';
@@ -35,6 +30,14 @@ import {MergedProfileSource, ProfileSource} from './ProfileSource';
 import {ProfileView} from './ProfileView';
 import {useProfileFilters} from './ProfileView/components/ProfileFilters/useProfileFilters';
 import type {SamplesSeries} from './ProfileView/types/visualization';
+import {
+  flamechartDimensionParser,
+  groupByParser,
+  intParam,
+  invertCallStackParser,
+  stringParam,
+} from './hooks/urlParsers';
+import {useDashboardItems} from './hooks/useDashboardItems';
 import {useQuery} from './useQuery';
 import {downloadPprof} from './utils';
 
@@ -53,22 +56,14 @@ export const ProfileViewWithData = ({
   onSwitchToFifteenMinutes,
 }: ProfileViewWithDataProps): JSX.Element => {
   const metadata = useGrpcMetadata();
-  const [dashboardItems, setDashboardItems] = useURLState<string[]>('dashboard_items', {
-    alwaysReturnArray: true,
-  });
-  const [sourceBuildID] = useURLState<string>('source_buildid');
-  const [sourceFilename] = useURLState<string>('source_filename');
-  const [groupBy] = useURLState<string[]>('group_by', {
-    defaultValue: [FIELD_FUNCTION_NAME],
-    alwaysReturnArray: true,
-  });
-  const [sandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
-  const [flamechartDimension] = useURLState<string[]>('flamechart_dimension', {
-    alwaysReturnArray: true,
-  });
+  const {dashboardItems, setDashboardItems} = useDashboardItems();
+  const [sourceBuildID] = useQueryState('source_buildid', stringParam);
+  const [sourceFilename] = useQueryState('source_filename', stringParam);
+  const [groupBy] = useQueryState('group_by', groupByParser.withDefault([FIELD_FUNCTION_NAME]));
+  const [sandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
+  const [flamechartDimension] = useQueryState('flamechart_dimension', flamechartDimensionParser);
 
-  const [invertStack] = useURLState('invert_call_stack');
-  const invertCallStack = invertStack === 'true';
+  const [invertCallStack] = useQueryState('invert_call_stack', invertCallStackParser);
 
   const [pprofDownloading, setPprofDownloading] = useState<boolean>(false);
 
@@ -110,7 +105,7 @@ export const ProfileViewWithData = ({
     skip: !dashboardItems.includes('flamegraph'),
     nodeTrimThreshold,
     groupBy,
-    invertCallStack,
+    invertCallStack: invertCallStack ?? false,
     protoFilters,
   });
 
@@ -134,11 +129,10 @@ export const ProfileViewWithData = ({
   );
 
   // Samples step count: 2px per data point for finer granularity in strips
-  const [samplesStepCount] = useURLStateCustom<number>('samples_step_count', {
-    defaultValue: String(getStepCountFromScreenWidth(2)),
-    parse: NumberParser,
-    stringify: NumberSerializer,
-  });
+  const [samplesStepCount] = useQueryState(
+    'samples_step_count',
+    intParam.withDefault(getStepCountFromScreenWidth(2))
+  );
 
   const {
     isLoading: samplesLoading,
@@ -203,8 +197,8 @@ export const ProfileViewWithData = ({
     error: sourceError,
   } = useQuery(queryClient, profileSource, QueryRequest_ReportType.SOURCE, {
     skip: !dashboardItems.includes('source'),
-    sourceBuildID,
-    sourceFilename,
+    sourceBuildID: sourceBuildID ?? undefined,
+    sourceFilename: sourceFilename ?? undefined,
     protoFilters,
   });
 
@@ -216,8 +210,8 @@ export const ProfileViewWithData = ({
     nodeTrimThreshold,
     groupBy: [FIELD_FUNCTION_NAME],
     invertCallStack: true,
-    sandwichByFunction: sandwichFunctionName,
-    skip: sandwichFunctionName === undefined && !dashboardItems.includes('sandwich'),
+    sandwichByFunction: sandwichFunctionName ?? undefined,
+    skip: sandwichFunctionName == null && !dashboardItems.includes('sandwich'),
     protoFilters,
   });
 
@@ -229,8 +223,8 @@ export const ProfileViewWithData = ({
     nodeTrimThreshold,
     groupBy: [FIELD_FUNCTION_NAME],
     invertCallStack: false,
-    sandwichByFunction: sandwichFunctionName,
-    skip: sandwichFunctionName === undefined && !dashboardItems.includes('sandwich'),
+    sandwichByFunction: sandwichFunctionName ?? undefined,
+    skip: sandwichFunctionName == null && !dashboardItems.includes('sandwich'),
     protoFilters,
   });
 

package/src/Sandwich/index.tsx

@@ -14,14 +14,15 @@
 import React, {useRef, useState} from 'react';
 
 import {AnimatePresence, motion} from 'framer-motion';
+import {useQueryState} from 'nuqs';
 
-import {useURLState} from '@parca/components';
 import {TEST_IDS, testId} from '@parca/test-utils';
 
 import {ProfileSource} from '../ProfileSource';
 import {useDashboard} from '../ProfileView/context/DashboardContext';
 import {useVisualizationState} from '../ProfileView/hooks/useVisualizationState';
 import {SandwichData} from '../ProfileView/types/visualization';
+import {stringParam} from '../hooks/urlParsers';
 import {CalleesSection} from './components/CalleesSection';
 import {CallersSection} from './components/CallersSection';
 
@@ -35,7 +36,7 @@ const Sandwich = React.memo(function Sandwich({
   profileSource,
 }: Props): React.JSX.Element {
   const {dashboardItems} = useDashboard();
-  const [sandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
+  const [sandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
 
   const callersRef = React.useRef<HTMLDivElement | null>(null);
   const calleesRef = React.useRef<HTMLDivElement | null>(null);
@@ -57,7 +58,7 @@ const Sandwich = React.memo(function Sandwich({
           transition={{duration: 0.5}}
         >
           <div className="relative flex flex-row">
-            {sandwichFunctionName !== undefined ? (
+            {sandwichFunctionName != null ? (
               <div className="w-full flex flex-col" ref={callersCalleesContainerRef}>
                 <CallersSection
                   callersRef={callersRef}

package/src/SourceView/index.tsx

@@ -15,12 +15,14 @@ import React, {useCallback, useEffect, useMemo} from 'react';
 
 import {tableFromIPC} from '@uwdata/flechette';
 import {AnimatePresence, motion} from 'framer-motion';
+import {useQueryState} from 'nuqs';
 import {Item, Menu, useContextMenu} from 'react-contexify';
 
 import {Source} from '@parca/client';
-import {SourceSkeleton, useParcaContext, useURLState, type ProfileData} from '@parca/components';
+import {SourceSkeleton, useParcaContext, type ProfileData} from '@parca/components';
 
 import {ExpandOnHover} from '../GraphTooltipArrow/ExpandOnHoverValue';
+import {stringParam} from '../hooks/urlParsers';
 import {alignedUint8Array, truncateStringReverse} from '../utils';
 import {Highlighter, profileAwareRenderer, type LineDataLookup} from './Highlighter';
 import useLineRange from './useSelectedLineRange';
@@ -42,7 +44,7 @@ export const SourceView = React.memo(function SourceView({
   filtered,
   setActionButtons,
 }: SourceViewProps): JSX.Element {
-  const [sourceFileName] = useURLState<string | undefined>('source_filename');
+  const [sourceFileName] = useQueryState('source_filename', stringParam);
   const {isDarkMode, sourceViewContextMenuItems = []} = useParcaContext();
 
   const sourceCode = useMemo(() => {

package/src/SourceView/useSelectedLineRange.ts

@@ -11,9 +11,25 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import {useMemo} from 'react';
+import {useCallback} from 'react';
 
-import {useURLState} from '@parca/components';
+import {createParser, useQueryState} from 'nuqs';
+
+interface SelectedLineRange {
+  start: number;
+  end: number;
+}
+
+const lineRangeParser = createParser<SelectedLineRange>({
+  parse: (value: string) => {
+    const [start, end] = value.split('-');
+    const startNum = parseInt(start, 10);
+    if (isNaN(startNum)) return null;
+    const endNum = end !== undefined ? parseInt(end, 10) : startNum;
+    return {start: startNum, end: isNaN(endNum) ? startNum : endNum};
+  },
+  serialize: (value: SelectedLineRange) => `${value.start}-${value.end}`,
+}).withOptions({history: 'replace'});
 
 interface LineRange {
   startLine: number;
@@ -22,24 +38,23 @@ interface LineRange {
 }
 
 const useLineRange = (): LineRange => {
-  const [sourceLine, setSourceLine] = useURLState<string | undefined>('source_line');
-  const [startLine, endLine] = useMemo(() => {
-    if (sourceLine == null) {
-      return [-1, -1];
-    }
-    const [start, end] = sourceLine.split('-');
-
-    if (end === undefined) {
-      return [parseInt(start, 10), parseInt(start, 10)];
-    }
-    return [parseInt(start, 10), parseInt(end, 10)];
-  }, [sourceLine]);
-
-  const setLineRange = (start: number, end: number): void => {
-    setSourceLine(`${start}-${end}`);
+  const [lineRange, setRawLineRange] = useQueryState(
+    'source_line',
+    lineRangeParser.withDefault({start: -1, end: -1})
+  );
+
+  const setLineRange = useCallback(
+    (start: number, end: number): void => {
+      void setRawLineRange({start, end});
+    },
+    [setRawLineRange]
+  );
+
+  return {
+    startLine: lineRange.start,
+    endLine: lineRange.end,
+    setLineRange,
   };
-
-  return {startLine, endLine, setLineRange};
 };
 
 export default useLineRange;

package/src/Table/MoreDropdown.tsx

@@ -13,23 +13,21 @@
 
 import {Menu} from '@headlessui/react';
 import {Icon} from '@iconify/react';
+import {useQueryState} from 'nuqs';
 
-import {useParcaContext, useURLState, useURLStateBatch} from '@parca/components';
+import {useParcaContext} from '@parca/components';
+
+import {stringParam} from '../hooks/urlParsers';
+import {useDashboardItems} from '../hooks/useDashboardItems';
 
 const MoreDropdown = ({functionName}: {functionName: string}): React.JSX.Element | null => {
-  const [_, setSandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
-  const [dashboardItems, setDashboardItems] = useURLState<string[]>('dashboard_items', {
-    alwaysReturnArray: true,
-  });
+  const [_, setSandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
+  const {dashboardItems, setDashboardItems} = useDashboardItems();
   const {enableSandwichView} = useParcaContext();
-  const batchUpdates = useURLStateBatch();
 
   const onSandwichViewSelect = (): void => {
-    // Batch updates to combine setSandwichFunctionName + setDashboardItems into single URL navigation
-    batchUpdates(() => {
-      setSandwichFunctionName(functionName.trim());
-      setDashboardItems([...dashboardItems, 'sandwich']);
-    });
+    void setSandwichFunctionName(functionName.trim());
+    setDashboardItems([...dashboardItems, 'sandwich']);
   };
 
   const menuItems: Array<{label: string; action: () => void}> = [];

package/src/Table/TableContextMenu.tsx

@@ -13,15 +13,18 @@
 
 import {Icon} from '@iconify/react';
 import cx from 'classnames';
+import {useQueryState} from 'nuqs';
 import {Item, Menu, Submenu} from 'react-contexify';
 
 import 'react-contexify/dist/ReactContexify.css';
 
-import {useParcaContext, useURLState, useURLStateBatch} from '@parca/components';
+import {useParcaContext} from '@parca/components';
 import {valueFormatter} from '@parca/utilities';
 
 import {type Row} from '.';
 import {getTextForCumulative} from '../ProfileFlameGraph/FlameGraphArrow/utils';
+import {stringParam} from '../hooks/urlParsers';
+import {useDashboardItems} from '../hooks/useDashboardItems';
 import {truncateString} from '../utils';
 import {type ColumnName} from './utils/functions';
 
@@ -42,22 +45,16 @@ const TableContextMenu = ({
   totalUnfiltered,
   columnVisibility,
 }: TableContextMenuProps): React.JSX.Element => {
-  const [_, setSandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
-  const [dashboardItems, setDashboardItems] = useURLState<string[]>('dashboard_items', {
-    alwaysReturnArray: true,
-  });
+  const [_, setSandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
+  const {dashboardItems, setDashboardItems} = useDashboardItems();
   const {enableSandwichView, isDarkMode} = useParcaContext();
-  const batchUpdates = useURLStateBatch();
 
   const onSandwichViewSelect = (): void => {
     if (row?.name != null && row.name.length > 0) {
-      // Batch updates to combine setSandwichFunctionName + setDashboardItems into single URL navigation
-      batchUpdates(() => {
-        setSandwichFunctionName(row.name.trim());
-        if (!dashboardItems.includes('sandwich')) {
-          setDashboardItems([...dashboardItems, 'sandwich']);
-        }
-      });
+      void setSandwichFunctionName(row.name.trim());
+      if (!dashboardItems.includes('sandwich')) {
+        setDashboardItems([...dashboardItems, 'sandwich']);
+      }
     }
   };
 

package/src/Table/hooks/useTableConfiguration.tsx

@@ -14,11 +14,12 @@
 import {useEffect, useMemo, useState} from 'react';
 
 import {createColumnHelper, type ColumnDef} from '@tanstack/table-core';
+import {useQueryState} from 'nuqs';
 
-import {useURLState} from '@parca/components';
 import {valueFormatter} from '@parca/utilities';
 
 import {type Row} from '..';
+import {tableColumnsParser} from '../../hooks/urlParsers';
 import {ColorCell} from '../ColorCell';
 import {addPlusSign, ratioString, type ColumnName} from '../utils/functions';
 
@@ -43,9 +44,7 @@ export function useTableConfiguration({
   compareMode,
 }: UseTableConfigurationProps): TableConfiguration {
   const columnHelper = createColumnHelper<Row>();
-  const [tableColumns] = useURLState<string[]>('table_columns', {
-    alwaysReturnArray: true,
-  });
+  const [tableColumns] = useQueryState('table_columns', tableColumnsParser);
 
   const [columnVisibility, setColumnVisibility] = useState(() => {
     return {

package/src/Table/index.tsx

@@ -16,14 +16,10 @@ import React, {useCallback, useEffect, useMemo, useRef} from 'react';
 import {RpcError} from '@protobuf-ts/runtime-rpc';
 import {tableFromIPC} from '@uwdata/flechette';
 import {AnimatePresence, motion} from 'framer-motion';
+import {useQueryState} from 'nuqs';
 import {useContextMenu} from 'react-contexify';
 
-import {
-  Table as TableComponent,
-  TableSkeleton,
-  useParcaContext,
-  useURLState,
-} from '@parca/components';
+import {Table as TableComponent, TableSkeleton, useParcaContext} from '@parca/components';
 import {useCurrentColorProfile} from '@parca/hooks';
 import {ProfileType} from '@parca/parser';
 
@@ -31,6 +27,9 @@ import useMappingList, {
   useFilenamesList,
 } from '../ProfileFlameGraph/FlameGraphArrow/useMappingList';
 import {useProfileViewContext} from '../ProfileView/context/ProfileViewContext';
+import {stringParam} from '../hooks/urlParsers';
+import {useColorBy} from '../hooks/useColorBy';
+import {useDashboardItems} from '../hooks/useDashboardItems';
 import {alignedUint8Array} from '../utils';
 import TableContextMenuWrapper, {TableContextMenuWrapperRef} from './TableContextMenuWrapper';
 import {useColorManagement} from './hooks/useColorManagement';
@@ -76,11 +75,9 @@ export const Table = React.memo(function Table({
   error,
 }: TableProps): React.JSX.Element {
   const currentColorProfile = useCurrentColorProfile();
-  const [dashboardItems] = useURLState<string[]>('dashboard_items', {
-    alwaysReturnArray: true,
-  });
-  const [_, setSandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
-  const [colorBy, setColorBy] = useURLState('color_by');
+  const {dashboardItems} = useDashboardItems();
+  const [_, setSandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
+  const {colorBy, setColorBy} = useColorBy();
   const {isDarkMode} = useParcaContext();
   const {compareMode} = useProfileViewContext();
 
@@ -108,7 +105,7 @@ export const Table = React.memo(function Table({
   // If there is only one mapping file, we want to color by filename by default.
   useEffect(() => {
     if (mappingsListCount === 1 && colorBy !== 'filename') {
-      setColorBy('filename');
+      void setColorBy('filename');
     }
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [mappingsListCount]);
@@ -118,7 +115,7 @@ export const Table = React.memo(function Table({
     currentColorProfile,
     mappingsList,
     filenamesList,
-    colorBy: colorBy as string,
+    colorBy,
   });
 
   unit = useMemo(() => unit ?? profileType?.sampleUnit ?? '', [unit, profileType?.sampleUnit]);
@@ -135,7 +132,7 @@ export const Table = React.memo(function Table({
   const selectSpan = useCallback(
     (span: string): void => {
       if (!dashboardItems.includes('flamegraph')) {
-        setSandwichFunctionName(span.trim());
+        void setSandwichFunctionName(span.trim());
       }
     },
     [setSandwichFunctionName, dashboardItems]
@@ -191,13 +188,7 @@ export const Table = React.memo(function Table({
       return {
         id: i,
         colorProperty: {
-          color: getRowColor(
-            colorByColors,
-            mappingFileColumn,
-            i,
-            functionFileNameColumn,
-            colorBy as string
-          ),
+          color: getRowColor(colorByColors, mappingFileColumn, i, functionFileNameColumn, colorBy),
           mappingFile,
         },
         name: RowName(mappingFileColumn, locationAddressColumn, functionNameColumn, i),