fix cvss2/cvss3 lost when update CVE info. (#115)

vulsio · Nov 23, 2018 · b083bed · b083bed
1 parent 9549cd3
commit b083bed
Showing 1 changed file with 20 additions and 10 deletions.
diff --git a/db/rdb.go b/db/rdb.go
@@ -551,8 +551,10 @@ func (r *RDBDriver) InsertJvn(cves []models.CveDetail) error {
 				if err != nil && err != gorm.ErrRecordNotFound {
 					return rollback(tx, err)
 				}
-				if err := tx.Unscoped().Delete(&cvss2).Error; err != nil {
-					return rollback(tx, err)
+				for _, cvss := range cvss2 {
+					if err := tx.Unscoped().Delete(cvss).Error; err != nil {
+						return rollback(tx, err)
+					}
 				}
 
 				// Delete Cvss3
@@ -561,8 +563,10 @@ func (r *RDBDriver) InsertJvn(cves []models.CveDetail) error {
 				if err != nil && err != gorm.ErrRecordNotFound {
 					return rollback(tx, err)
 				}
-				if err := tx.Unscoped().Delete(&cvss3).Error; err != nil {
-					return rollback(tx, err)
+				for _, cvss := range cvss3 {
+					if err := tx.Unscoped().Delete(cvss).Error; err != nil {
+						return rollback(tx, err)
+					}
 				}
 
 				// Delete old Cpes
@@ -730,8 +734,10 @@ func (r *RDBDriver) InsertNvdXML(cves []models.CveDetail) error {
 				if err != nil && err != gorm.ErrRecordNotFound {
 					return rollback(tx, err)
 				}
-				if err := tx.Unscoped().Delete(&cvss2).Error; err != nil {
-					return rollback(tx, err)
+				for _, cvss := range cvss2 {
+					if err := tx.Unscoped().Delete(cvss).Error; err != nil {
+						return rollback(tx, err)
+					}
 				}
 
 				// Delete old Nvd
@@ -842,8 +848,10 @@ func (r *RDBDriver) InsertNvdJSON(cves []models.CveDetail) (err error) {
 				if err != nil && err != gorm.ErrRecordNotFound {
 					return rollback(tx, err)
 				}
-				if err := tx.Unscoped().Delete(&cvss2).Error; err != nil {
-					return rollback(tx, err)
+				for _, cvss := range cvss2 {
+					if err := tx.Unscoped().Delete(cvss).Error; err != nil {
+						return rollback(tx, err)
+					}
 				}
 
 				// Delete Cvss3
@@ -852,8 +860,10 @@ func (r *RDBDriver) InsertNvdJSON(cves []models.CveDetail) (err error) {
 				if err != nil && err != gorm.ErrRecordNotFound {
 					return rollback(tx, err)
 				}
-				if err := tx.Unscoped().Delete(&cvss3).Error; err != nil {
-					return rollback(tx, err)
+				for _, cvss := range cvss3 {
+					if err := tx.Unscoped().Delete(cvss).Error; err != nil {
+						return rollback(tx, err)
+					}
 				}
 
 				// Delte old CWE