Merge pull request #100 from jortel/support-middleware

Web handler enhancements.
konveyor · Sep 27, 2021 · 858e8b9 · 858e8b9
2 parents 85e6c0e + a20c937
commit 858e8b9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/pkg/condition/zz_generated.deepcopy.go b/pkg/condition/zz_generated.deepcopy.go
diff --git a/pkg/inventory/web/handler.go b/pkg/inventory/web/handler.go
@@ -20,10 +20,6 @@ import (
 type RequestHandler interface {
 	// Add routes to the `gin` router.
 	AddRoutes(*gin.Engine)
-	// List resources in a REST collection.
-	List(*gin.Context)
-	// Get a specific REST resource.
-	Get(*gin.Context)
 }
 
 //

diff --git a/pkg/inventory/web/web.go b/pkg/inventory/web/web.go
@@ -42,7 +42,7 @@ type WebServer struct {
 // Start the web-server.
 // Initializes `gin` with routes and CORS origins.
 // Creates an http server to handle TLS
-func (w *WebServer) Start() {
+func (w *WebServer) Start(middleware ...gin.HandlerFunc) {
 	router := gin.Default()
 	router.Use(cors.New(cors.Config{
 		AllowMethods:     []string{"GET"},
@@ -51,6 +51,9 @@ func (w *WebServer) Start() {
 		AllowCredentials: true,
 		MaxAge:           12 * time.Hour,
 	}))
+	for _, h := range middleware {
+		router.Use(h)
+	}
 	w.buildOrigins()
 	w.addRoutes(router)
 	if w.TLS.Enabled {