From 14e93058f201af60f42102c85e5fe5daf4022980 Mon Sep 17 00:00:00 2001
From: Ryan Montgomery <ryan.montgomery@capitalone.com>
Date: Thu, 2 Sep 2021 13:33:54 -0400
Subject: [PATCH] Update widget infra endpoint

---
 pom.xml                                        |  2 +-
 .../service/InfraStructureServiceImpl.java     | 18 +++++++++++++++---
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/pom.xml b/pom.xml
index 04c5b847..2c8e436d 100644
--- a/pom.xml
+++ b/pom.xml
@@ -59,7 +59,7 @@
 
     <properties>
         <!-- Dependencies -->
-        <com.capitalone.dashboard.core.version>3.15.9</com.capitalone.dashboard.core.version>
+        <com.capitalone.dashboard.core.version>3.15.16</com.capitalone.dashboard.core.version>
         <spring-security.version>4.2.18.RELEASE</spring-security.version>
         <tomcat.version>8.5.57</tomcat.version>
         <commons-beanutils.version>1.9.4</commons-beanutils.version>
diff --git a/src/main/java/com/capitalone/dashboard/service/InfraStructureServiceImpl.java b/src/main/java/com/capitalone/dashboard/service/InfraStructureServiceImpl.java
index b40ec65b..4a0b198d 100644
--- a/src/main/java/com/capitalone/dashboard/service/InfraStructureServiceImpl.java
+++ b/src/main/java/com/capitalone/dashboard/service/InfraStructureServiceImpl.java
@@ -6,8 +6,9 @@
 import com.capitalone.dashboard.request.InfraStructureRequest;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-
+import java.util.ArrayList;
 import java.util.Collections;
+import java.util.List;
 
 @Service
 public class InfraStructureServiceImpl implements InfraStructureService {
@@ -21,7 +22,18 @@ public InfraStructureServiceImpl(InfrastructureScanRepository infrastructureScan
 
     @Override
     public DataResponse<Iterable<InfrastructureScan>> getInfraScanForWidget(InfraStructureRequest request) {
-        InfrastructureScan infrastructureScan = infrastructureScanRepository.findTopByCollectorItemIdOrderByTimestampDesc(request.getCollectorItemId());
-        return new DataResponse<>(Collections.singletonList(infrastructureScan), System.currentTimeMillis());
+        List<InfrastructureScan> tempInfraScanList = infrastructureScanRepository.findByCollectorItemIdOrderByTimestampDesc(request.getCollectorItemId());
+        List<InfrastructureScan> infrastructureScanList = new ArrayList<InfrastructureScan>();
+
+        ArrayList<String> instanceIdArray = new ArrayList<String>();
+        tempInfraScanList.forEach(scan -> {
+            String instanceId = scan.getInstanceId();
+            if(!instanceIdArray.contains(instanceId)){
+                instanceIdArray.add(instanceId);
+                infrastructureScanList.add(scan);
+            }
+        });
+
+        return new DataResponse<>(infrastructureScanList, System.currentTimeMillis());
     }
 }