kafka-ui/kafka-ui-react-app/src/lib/hooks/api/brokers.ts

import { brokersApiClient as api } from 'lib/api';
import { useQuery } from 'react-query';
import { ClusterName } from 'redux/interfaces';

export function useBrokers(clusterName: ClusterName) {
  return useQuery(
    ['clusters', clusterName, 'brokers'],
    () => api.getBrokers({ clusterName }),
    { refetchInterval: 5000 }
  );
}

export function useBrokerMetrics(clusterName: ClusterName, brokerId: number) {
  return useQuery(
    ['clusters', clusterName, 'brokers', brokerId, 'metrics'],
    () =>
      api.getBrokersMetrics({
        clusterName,
        id: brokerId,
      }),
    { refetchInterval: 5000 }
  );
}

export function useBrokerLogDirs(clusterName: ClusterName, brokerId: number) {
  return useQuery(
    ['clusters', clusterName, 'brokers', brokerId, 'logDirs'],
    () =>
      api.getAllBrokersLogdirs({
        clusterName,
        broker: [brokerId],
      }),
    { refetchInterval: 5000 }
  );
}