small enhancements

inexio · Apr 15, 2020 · 3f4efe2 · 3f4efe2
1 parent 38b36b1
commit 3f4efe2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/cve-alert-restapi/request-handler/requestHandler.go b/cve-alert-restapi/request-handler/requestHandler.go
@@ -885,7 +885,7 @@ func getCategoriesOfOther(c echo.Context) error {
 }
 
 func getPlatforms(c echo.Context) error {
-	var platforms []string
+	var platforms []alertmanager.Platform
 	err := db.Select(&platforms, "SELECT DISTINCT platform FROM case_platforms")
 	if err != nil {
 		log.Error().

diff --git a/cvecase.go b/cvecase.go
@@ -257,6 +257,16 @@ type Category struct {
 	Category string `db:"category" json:"category"`
 }
 
+// Model for a platform
+//
+// swagger:model Platform
+type Platform struct {
+	// The platform name
+	//
+	// example: Linux
+	Platform string `db:"platform" json:"platform"`
+}
+
 // Model for exporting filter_config.yaml
 //
 // swagger:model FilterConfig