Add signal and abort controller to agent metadata and TakeAction button (#103217)

This commit is contained in:
Esteban Beltran 2021-06-29 08:58:37 +02:00 committed by GitHub
parent 09bd6301d6
commit ac17ab1436
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 3 deletions

View file

@ -178,8 +178,14 @@ export const getCaseIdsFromAlertId = async ({
*
* @param host id
*/
export const getHostMetadata = async ({ agentId }: { agentId: string }): Promise<HostInfo> =>
export const getHostMetadata = async ({
agentId,
signal,
}: {
agentId: string;
signal?: AbortSignal;
}): Promise<HostInfo> =>
KibanaServices.get().http.fetch<HostInfo>(
resolvePathVariables(HOST_METADATA_GET_ROUTE, { id: agentId }),
{ method: 'get' }
{ method: 'GET', signal }
);

View file

@ -38,18 +38,23 @@ export const useHostIsolationStatus = ({
const { addError } = useAppToasts();
useEffect(() => {
const abortCtrl = new AbortController();
// isMounted tracks if a component is mounted before changing state
let isMounted = true;
let fleetAgentId: string;
const fetchData = async () => {
try {
const metadataResponse = await getHostMetadata({ agentId });
const metadataResponse = await getHostMetadata({ agentId, signal: abortCtrl.signal });
if (isMounted) {
setIsIsolated(isEndpointHostIsolated(metadataResponse.metadata));
setAgentStatus(metadataResponse.host_status);
fleetAgentId = metadataResponse.metadata.elastic.agent.id;
}
} catch (error) {
// don't show self-aborted requests errors to the user
if (error.name === 'AbortError') {
return;
}
addError(error.message, { title: ISOLATION_STATUS_FAILURE });
}
@ -80,6 +85,7 @@ export const useHostIsolationStatus = ({
return () => {
// updates to show component is unmounted
isMounted = false;
abortCtrl.abort();
};
}, [addError, agentId]);
return { loading, isIsolated, agentStatus, pendingIsolation, pendingUnisolation };