[DataUsage][Serverless] Data usage metrics page enhancements

x-pack/plugins/data_usage/server/routes/internal/data_streams_handler.ts

@@ -5,36 +5,50 @@
  * 2.0.
  */
 
-import { RequestHandler } from '@kbn/core/server';
+import { type ElasticsearchClient, RequestHandler } from '@kbn/core/server';
 import { DataUsageContext, DataUsageRequestHandlerContext } from '../../types';
-
 import { errorHandler } from '../error_handler';
 
+export interface MeteringStats {
+  name: string;
+  num_docs: number;
+  size_in_bytes: number;
+}
+
+interface MeteringStatsResponse {
+  datastreams: MeteringStats[];
+}
+
+const getMeteringStats = (client: ElasticsearchClient) => {
+  return client.transport.request<MeteringStatsResponse>({
+    method: 'GET',
+    path: '/_metering/stats',
+  });
+};
+
 export const getDataStreamsHandler = (
   dataUsageContext: DataUsageContext
 ): RequestHandler<never, unknown, DataUsageRequestHandlerContext> => {
   const logger = dataUsageContext.logFactory.get('dataStreamsRoute');
 
   return async (context, _, response) => {
-    logger.debug(`Retrieving user data streams`);
+    logger.debug('Retrieving user data streams');
 
     try {
       const core = await context.core;
-      const esClient = core.elasticsearch.client.asCurrentUser;
-
-      const { data_streams: dataStreamsResponse } = await esClient.indices.dataStreamsStats({
-        name: '*',
-        expand_wildcards: 'all',
-      });
+      const { datastreams: meteringStats } = await getMeteringStats(
+        core.elasticsearch.client.asSecondaryAuthUser
+      );
 
-      const sorted = dataStreamsResponse
-        .sort((a, b) => b.store_size_bytes - a.store_size_bytes)
-        .map((dataStream) => ({
-          name: dataStream.data_stream,
-          storageSizeBytes: dataStream.store_size_bytes,
+      const body = meteringStats
+        .sort((a, b) => b.size_in_bytes - a.size_in_bytes)
+        .map((stat) => ({
+          name: stat.name,
+          storageSizeBytes: stat.size_in_bytes,
         }));
+
       return response.ok({
-        body: sorted,
+        body,
       });
     } catch (error) {
       return errorHandler(logger, response, error);