delete old ssvc upon new one

SoftwareDesignLab · Nov 13, 2023 · a96a61e · a96a61e
1 parent 020cf5a
commit a96a61e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/reconciler/src/main/java/edu/rit/se/nvip/DatabaseHelper.java b/reconciler/src/main/java/edu/rit/se/nvip/DatabaseHelper.java
@@ -799,9 +799,15 @@ public boolean exploitExists(String cveId) {
     }
 
     public void insertSSVCSet(Set<CompositeVulnerability> vulns) {
-        try (Connection conn = getConnection(); PreparedStatement pstmt = conn.prepareStatement(INSERT_SSVC)) {
+        String deleteOldSSVC = "DELETE FROM ssvc WHERE cve_id = ?";
+        try (Connection conn = getConnection(); PreparedStatement pstmt = conn.prepareStatement(INSERT_SSVC);
+            PreparedStatement deleteStmt = conn.prepareStatement(deleteOldSSVC)) {
             conn.setAutoCommit(false);
             for (CompositeVulnerability vuln : vulns) {
+                deleteStmt.setString(1, vuln.getCveId());
+                deleteStmt.executeUpdate();
+
+
                 // Get SSVC data
                 final SSVC ssvc = vuln.getSSVC();