Added a context.Context to the plugins.Handle interface

Signed-off-by: Shmuel Kallner <[email protected]>

Author: shmuelk

cmd/epp/runner/register.go

@@ -17,6 +17,8 @@ limitations under the License.
 package runner
 
 import (
+	"context"
+
 	"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/plugins"
 	"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/scheduling/framework/plugins/filter"
 	"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/scheduling/framework/plugins/multi/prefix"
@@ -42,9 +44,15 @@ func RegisterAllPlugins() {
 
 // eppHandle is an implementation of the interface plugins.Handle
 type eppHandle struct {
+	ctx     context.Context
 	plugins plugins.HandlePlugins
 }
 
+// Context returns a context the plugins can use, if they need one
+func (h *eppHandle) Context() context.Context {
+	return h.ctx
+}
+
 // Plugins returns the sub-handle for working with instantiated plugins
 func (h *eppHandle) Plugins() plugins.HandlePlugins {
 	return h.plugins
@@ -79,8 +87,9 @@ func (h *eppHandlePlugins) GetAllPluginsWithNames() map[string]plugins.Plugin {
 	return h.thePlugins
 }
 
-func newEppHandle() *eppHandle {
+func newEppHandle(ctx context.Context) *eppHandle {
 	return &eppHandle{
+		ctx: ctx,
 		plugins: &eppHandlePlugins{
 			thePlugins: map[string]plugins.Plugin{},
 		},

pkg/epp/plugins/plugins.go

@@ -16,6 +16,8 @@ limitations under the License.
 
 package plugins
 
+import "context"
+
 // Plugin defines the interface for a plugin.
 // This interface should be embedded in all plugins across the code.
 type Plugin interface {
@@ -27,6 +29,9 @@ type Plugin interface {
 
 // Handle provides plugins a set of standard data and tools to work with
 type Handle interface {
+	// Context returns a context the plugins can use, if they need one
+	Context() context.Context
+
 	// Plugins returns the sub-handle for working with instantiated plugins
 	Plugins() HandlePlugins
 }

test/utils/handle.go

@@ -16,13 +16,23 @@ limitations under the License.
 
 package utils
 
-import "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/plugins"
+import (
+	"context"
+
+	"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/plugins"
+)
 
 // testHandle is an implmentation of plugins.Handle for test purposes
 type testHandle struct {
+	ctx     context.Context
 	plugins plugins.HandlePlugins
 }
 
+// Context returns a context the plugins can use, if they need one
+func (h *testHandle) Context() context.Context {
+	return h.ctx
+}
+
 func (h *testHandle) Plugins() plugins.HandlePlugins {
 	return h.plugins
 }
@@ -51,8 +61,9 @@ func (h *testHandlePlugins) GetAllPluginsWithNames() map[string]plugins.Plugin {
 	return h.thePlugins
 }
 
-func NewTestHandle() plugins.Handle {
+func NewTestHandle(ctx context.Context) plugins.Handle {
 	return &testHandle{
+		ctx: ctx,
 		plugins: &testHandlePlugins{
 			thePlugins: map[string]plugins.Plugin{},
 		},