api: handle last page for notifications

This commit is contained in:
Jimmy Zelinskie 2016-02-01 19:22:04 -05:00
parent 3eaae478f9
commit 96e96d948d

View File

@ -172,6 +172,11 @@ func NotificationFromDatabaseModel(dbNotification database.VulnerabilityNotifica
*oldVuln = VulnerabilityWithLayersFromDatabaseModel(*dbNotification.OldVulnerability) *oldVuln = VulnerabilityWithLayersFromDatabaseModel(*dbNotification.OldVulnerability)
} }
var nextPageStr string
if nextPage != database.NoVulnerabilityNotificationPage {
nextPageStr = DBPageNumberToString(nextPage)
}
// TODO(jzelinskie): implement "changed" key // TODO(jzelinskie): implement "changed" key
return Notification{ return Notification{
Name: dbNotification.Name, Name: dbNotification.Name,
@ -180,7 +185,7 @@ func NotificationFromDatabaseModel(dbNotification database.VulnerabilityNotifica
Deleted: fmt.Sprintf("%d", dbNotification.Deleted.Unix()), Deleted: fmt.Sprintf("%d", dbNotification.Deleted.Unix()),
Limit: limit, Limit: limit,
Page: DBPageNumberToString(page), Page: DBPageNumberToString(page),
NextPage: DBPageNumberToString(nextPage), NextPage: nextPageStr,
Old: oldVuln, Old: oldVuln,
New: VulnerabilityWithLayersFromDatabaseModel(dbNotification.NewVulnerability), New: VulnerabilityWithLayersFromDatabaseModel(dbNotification.NewVulnerability),
} }