arno/clair
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

database: create notification during vulnerability insertion

Browse Source
This commit is contained in:
Quentin Machu 2016-01-23 22:02:51 -05:00 committed by Jimmy Zelinskie
parent baed60e19b
commit 2690800331
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
database/pgsql/vulnerability.go
View File

@ -196,12 +196,6 @@ func (pgSQL *pgSQL) insertVulnerability(vulnerability database.Vulnerability) er
tx.Rollback() tx.Rollback()
return handleError("i_vulnerability", err) return handleError("i_vulnerability", err)
} }
// Create NewVulnerabilityNotification.
notification := database.VulnerabilityNotification{NewVulnerability: vulnerability}
if err := pgSQL.insertNotification(tx, notification); err != nil {
return err
}
} else { } else {
// Update vulnerability // Update vulnerability
if vulnerability.Description != existingVulnerability.Description || if vulnerability.Description != existingVulnerability.Description ||
@ -228,6 +222,22 @@ func (pgSQL *pgSQL) insertVulnerability(vulnerability database.Vulnerability) er
return err return err
} }
// Create notification.
var notification database.VulnerabilityNotification
if existingVulnerability.ID == 0 {
notification = database.VulnerabilityNotification{
NewVulnerability: vulnerability,
}
} else {
notification = database.VulnerabilityNotification{
OldVulnerability: &existingVulnerability,
NewVulnerability: vulnerability,
}
}
if err := pgSQL.insertNotification(tx, notification); err != nil {
return err
}
// Commit transaction. // Commit transaction.
err = tx.Commit() err = tx.Commit()
if err != nil { if err != nil {