save only ids of vulns

Cornul11 · Nov 24, 2023 · 75f9402 · 75f9402
1 parent 16e60f2
commit 75f9402
Showing 1 changed file with 7 additions and 14 deletions.
diff --git a/src/main/java/nl/tudelft/cornul11/thesis/corpus/service/VulnerabilityAnalyzer.java b/src/main/java/nl/tudelft/cornul11/thesis/corpus/service/VulnerabilityAnalyzer.java
@@ -45,18 +45,11 @@ public void checkForVulnerability(Map<String, Map<String, Object>> libraryVersio
  for (LibraryVersion libraryVersion : libraryVersions) {
  ArrayList<Document> vulnerabilitiesList = mongoDbClient.getVulnerabilities(libraryVersion.getLibrary(), libraryVersion.getVersion());
 
- List<Map<String, Object>> simplifiedVulnerabilities = vulnerabilitiesList.stream()
- .map(Document::toJson)
- .map(jsonString -> {
- try {
- return new ObjectMapper().readValue(jsonString, new TypeReference<Map<String, Object>>() {});
- } catch (IOException e) {
- logger.error("Error parsing vulnerability document to JSON", e);
- return new HashMap<String, Object>();
- }
- })
+ List<String> vulnerabilityIds = vulnerabilitiesList.stream()
+ .map(doc -> doc.getString("id"))
+ .filter(Objects::nonNull)
  .collect(Collectors.toList());
- libraryVersion.setVulnerabilities(simplifiedVulnerabilities);
+ libraryVersion.setVulnerabilities(vulnerabilityIds);
  logResult(libraryVersion, vulnerabilitiesList);
  }
 
@@ -129,7 +122,7 @@ public static class LibraryVersion {
  private final long count;
  private final long total;
  private final double ratio;
- private List<Map<String, Object>> vulnerabilities;
+ private List<String> vulnerabilities;
 
  public LibraryVersion(String library, String version, long count, long total, double ratio) {
  this.library = library;
@@ -140,11 +133,11 @@ public LibraryVersion(String library, String version, long count, long total, do
  this.vulnerabilities = new ArrayList<>();
  }
 
- public void setVulnerabilities(List<Map<String, Object>> vulnerabilities) {
+ public void setVulnerabilities(List<String> vulnerabilities) {
  this.vulnerabilities = vulnerabilities;
  }
 
- public List<Map<String, Object>> getVulnerabilities() {
+ public List<String> getVulnerabilities() {
  return vulnerabilities;
  }