api: handle last page for notifications

quay · Feb 24, 2016 · 96e96d9 · 96e96d9
1 parent 3eaae47
commit 96e96d9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/api/v1/models.go b/api/v1/models.go
@@ -172,6 +172,11 @@ func NotificationFromDatabaseModel(dbNotification database.VulnerabilityNotifica
 		*oldVuln = VulnerabilityWithLayersFromDatabaseModel(*dbNotification.OldVulnerability)
 	}
 
+	var nextPageStr string
+	if nextPage != database.NoVulnerabilityNotificationPage {
+		nextPageStr = DBPageNumberToString(nextPage)
+	}
+
 	// TODO(jzelinskie): implement "changed" key
 	return Notification{
 		Name:     dbNotification.Name,
@@ -180,7 +185,7 @@ func NotificationFromDatabaseModel(dbNotification database.VulnerabilityNotifica
 		Deleted:  fmt.Sprintf("%d", dbNotification.Deleted.Unix()),
 		Limit:    limit,
 		Page:     DBPageNumberToString(page),
-		NextPage: DBPageNumberToString(nextPage),
+		NextPage: nextPageStr,
 		Old:      oldVuln,
 		New:      VulnerabilityWithLayersFromDatabaseModel(dbNotification.NewVulnerability),
 	}