Skip to content

Commit b9f3bf0

Browse files
committed
feat: refactor asset synchronization logic and enhance data source aggregation
1 parent 1f48e3c commit b9f3bf0

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

backend/src/main/java/com/park/utmstack/service/network_scan/AssetSynchronizationService.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,11 @@ public void syncDataInputsAndAssets() {
5353

5454
Map<String, AgentDTO> agentsMap = loadAgents();
5555
Map<String, UtmDataInputStatus> statusMap = buildDataInputStatusMap();
56-
Map<String, UtmNetworkScan> assetsMap = buildNetworkAssetsMap(new ArrayList<>(statsMap.keySet()));
56+
Map<String, UtmNetworkScan> assetsMap = buildNetworkAssetsMap(
57+
new ArrayList<>(statsMap.values()
58+
.stream()
59+
.map(StatisticDocument::getDataSource)
60+
.collect(Collectors.toSet())));
5761

5862
List<UtmDataInputStatus> statusToSave = new ArrayList<>();
5963
List<UtmNetworkScan> assetsToSave = new ArrayList<>();
@@ -65,7 +69,7 @@ public void syncDataInputsAndAssets() {
6569
statusToSave.add(status);
6670

6771
// Update network asset
68-
UtmNetworkScan asset = processNetworkAsset(sourceName, agentsMap, assetsMap, statusMap);
72+
UtmNetworkScan asset = processNetworkAsset(stat.getDataSource(), agentsMap, assetsMap, statusMap);
6973
assetsToSave.add(asset);
7074
}
7175

0 commit comments

Comments
 (0)