fix: use count query for...bar instead of limit 100 (#3756)

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>

Author: Copilot

src/components/QueriesActivityBar/useQueriesActivityData.ts

@@ -2,7 +2,6 @@ import React from 'react';
 
 import {selectGraphShardExists} from '../../store/reducers/capabilities/capabilities';
 import {topQueriesApi} from '../../store/reducers/executeTopQueries/executeTopQueries';
-import type {KeyValueRow} from '../../types/api/query';
 import {useAutoRefreshInterval} from '../../utils/hooks';
 import {useTypedSelector} from '../../utils/hooks/useTypedSelector';
 import type {TimeFrame} from '../../utils/timeframes';
@@ -32,10 +31,9 @@ export function useQueriesActivityData(database: string): UseQueriesActivityData
     const graphShardExists = useTypedSelector((state) => selectGraphShardExists(state, database));
     const skipCharts = graphShardExists === false;
 
-    const {data: runningQueriesData} = topQueriesApi.useGetRunningQueriesQuery(
+    const {data: runningQueriesCountData} = topQueriesApi.useGetRunningQueriesCountQuery(
         {
             database,
-            filters: {},
         },
         {pollingInterval: shouldRefresh},
     );
@@ -64,7 +62,9 @@ export function useQueriesActivityData(database: string): UseQueriesActivityData
         {pollingInterval: shouldRefresh, skip: skipCharts},
     );
 
-    const runningQueriesCount = runningQueriesData?.resultSets?.[0]?.result?.length || 0;
+    const runningQueriesCount = runningQueriesCountData?.runningQueriesCount ?? 0;
+    const uniqueApplications = runningQueriesCountData?.uniqueApplications ?? 0;
+    const uniqueUsers = runningQueriesCountData?.uniqueUsers ?? 0;
 
     // Determine chart availability from queries API success/error state
     const areChartsAvailable = React.useMemo(() => {
@@ -90,26 +90,6 @@ export function useQueriesActivityData(database: string): UseQueriesActivityData
         [latencyData?.metrics?.[0]?.data],
     );
 
-    const uniqueApplications = React.useMemo(() => {
-        const apps = new Set<string>();
-        runningQueriesData?.resultSets?.[0]?.result?.forEach((row: KeyValueRow) => {
-            if (row.ApplicationName) {
-                apps.add(String(row.ApplicationName));
-            }
-        });
-        return apps.size;
-    }, [runningQueriesData]);
-
-    const uniqueUsers = React.useMemo(() => {
-        const users = new Set<string>();
-        runningQueriesData?.resultSets?.[0]?.result?.forEach((row: KeyValueRow) => {
-            if (row.UserSID) {
-                users.add(String(row.UserSID));
-            }
-        });
-        return users.size;
-    }, [runningQueriesData]);
-
     return {
         runningQueriesCount,
         uniqueApplications,

src/store/reducers/executeTopQueries/executeTopQueries.ts

@@ -99,6 +99,17 @@ AND QueryStartAt is not null ${orderBy}
 LIMIT ${limit || 100}`;
 }
 
+function getRunningQueriesCountText() {
+    return `${QUERY_TECHNICAL_MARK}
+SELECT
+    COUNT(*) as RunningQueriesCount,
+    COUNT(DISTINCT CASE WHEN ApplicationName = '' THEN NULL ELSE ApplicationName END) as UniqueApplications,
+    COUNT(DISTINCT CASE WHEN UserSID = '' THEN NULL ELSE UserSID END) as UniqueUsers
+FROM \`.sys/query_sessions\`
+WHERE Query NOT LIKE '%${QUERY_TECHNICAL_MARK}%'
+AND QueryStartAt is not null`;
+}
+
 interface QueriesRequestParams {
     database: string;
     filters?: TopQueriesFilters;
@@ -190,6 +201,42 @@ export const topQueriesApi = api.injectEndpoints({
             },
             providesTags: ['All'],
         }),
+        getRunningQueriesCount: build.query({
+            queryFn: async ({database}: {database: string}, {signal}) => {
+                try {
+                    const response = await window.api.viewer.sendQuery(
+                        {
+                            query: getRunningQueriesCountText(),
+                            database,
+                            action: 'execute-query',
+                            internal_call: true,
+                        },
+                        {signal, withRetries: true},
+                    );
+
+                    if (isQueryErrorResponse(response)) {
+                        throw response;
+                    }
+
+                    const data = parseQueryAPIResponse(response);
+                    const row = data?.resultSets?.[0]?.result?.[0];
+
+                    return {
+                        data: {
+                            runningQueriesCount: Number(row?.RunningQueriesCount) || 0,
+                            uniqueApplications: Number(row?.UniqueApplications) || 0,
+                            uniqueUsers: Number(row?.UniqueUsers) || 0,
+                        },
+                    };
+                } catch (error) {
+                    return {error};
+                }
+            },
+            forceRefetch() {
+                return true;
+            },
+            providesTags: ['All'],
+        }),
     }),
     overrideExisting: 'throw',
 });