mirror of https://github.com/portainer/portainer
30 lines
748 B
TypeScript
30 lines
748 B
TypeScript
import { useQuery } from 'react-query';
|
|
|
|
import { EnvironmentId } from '@/portainer/environments/types';
|
|
import axios, { parseAxiosError } from '@/portainer/services/axios';
|
|
|
|
import { queryKeys } from './query-keys';
|
|
|
|
async function getContainerGpus(
|
|
environmentId: EnvironmentId,
|
|
containerId: string
|
|
) {
|
|
try {
|
|
const { data } = await axios.get<{ gpus: string }>(
|
|
`/docker/${environmentId}/containers/${containerId}/gpus`
|
|
);
|
|
return data.gpus;
|
|
} catch (err) {
|
|
throw parseAxiosError(err as Error);
|
|
}
|
|
}
|
|
|
|
export function useContainerGpus(
|
|
environmentId: EnvironmentId,
|
|
containerId: string
|
|
) {
|
|
return useQuery(queryKeys.gpus(environmentId, containerId), () =>
|
|
getContainerGpus(environmentId, containerId)
|
|
);
|
|
}
|