Fix function signature (#3070)

internal/cmd/commands/dev/dev.go

@@ -48,7 +48,7 @@ var (
 	_ cli.CommandAutocomplete = (*Command)(nil)
 )
 
-var extraSelfTerminationConditionFuncs []func(context.Context, chan struct{})
+var extraSelfTerminationConditionFuncs []func(*Command, chan struct{})
 
 type Command struct {
 	*base.Server
@@ -924,7 +924,7 @@ func (c *Command) Run(args []string) int {
 	}
 
 	for _, f := range extraSelfTerminationConditionFuncs {
-		f(c.Context, c.ServerSideShutdownCh)
+		f(c, c.ServerSideShutdownCh)
 	}
 
 	for !errorEncountered.Load() && !shutdownCompleted.Load() {

internal/cmd/commands/server/server.go

@@ -41,7 +41,7 @@ var (
 	_ cli.CommandAutocomplete = (*Command)(nil)
 )
 
-var extraSelfTerminationConditionFuncs []func(context.Context, chan struct{})
+var extraSelfTerminationConditionFuncs []func(*Command, chan struct{})
 
 type Command struct {
 	*base.Server
@@ -700,7 +700,7 @@ func (c *Command) WaitForInterrupt() int {
 	}
 
 	for _, f := range extraSelfTerminationConditionFuncs {
-		f(c.Context, c.ServerSideShutdownCh)
+		f(c, c.ServerSideShutdownCh)
 	}
 
 	for !shutdownCompleted.Load() {