From 388a7ffcd7b93bfee5f88d037bd789884d150b20 Mon Sep 17 00:00:00 2001 From: Gonzalo Avalos Ribas Date: Thu, 25 Jul 2024 14:49:25 -0300 Subject: [PATCH 1/2] Check for duplicated keys --- src/steps/ms-defender/machine/index.ts | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/steps/ms-defender/machine/index.ts b/src/steps/ms-defender/machine/index.ts index 1618c66..f06b670 100644 --- a/src/steps/ms-defender/machine/index.ts +++ b/src/steps/ms-defender/machine/index.ts @@ -72,9 +72,11 @@ export async function fetchEndpoints({ try { const endpoint = await graphClient.fetchEndpointDetails(machine.id); if (endpoint) { - const endpointEntity = await jobState.addEntity( - createEndpointEntity(endpoint), - ); + const endpointEntity = createEndpointEntity(endpoint); + + if (jobState.hasKey(endpoint._key)) return; + + await jobState.addEntity(endpointEntity); await jobState.addRelationship( createMachineEndpointRelationship({ From bab56fb8fcc64a93eca70096e3421917376a9bc4 Mon Sep 17 00:00:00 2001 From: Gonzalo Avalos Ribas Date: Thu, 25 Jul 2024 14:50:23 -0300 Subject: [PATCH 2/2] Format --- src/steps/ms-defender/machine/index.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/steps/ms-defender/machine/index.ts b/src/steps/ms-defender/machine/index.ts index f06b670..99e357b 100644 --- a/src/steps/ms-defender/machine/index.ts +++ b/src/steps/ms-defender/machine/index.ts @@ -73,7 +73,7 @@ export async function fetchEndpoints({ const endpoint = await graphClient.fetchEndpointDetails(machine.id); if (endpoint) { const endpointEntity = createEndpointEntity(endpoint); - + if (jobState.hasKey(endpoint._key)) return; await jobState.addEntity(endpointEntity);