weiguoqiang/cert-manager
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

BUGFIX: run pprof server on non-leaderelected replicas

Browse Source 
Signed-off-by: Tim Ramlot <42113979+inteon@users.noreply.github.com>
This commit is contained in:
Tim Ramlot 2023-11-16 11:21:34 +01:00
parent 8c7615f896
commit 05de994587
No known key found for this signature in database
GPG Key ID: 47428728E0C2878D
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
cmd/cainjector/app/controller.go
View File

@ -100,7 +100,7 @@ func Run(opts *config.CAInjectorConfiguration, ctx context.Context) error {
Handler: profilerMux,
}
mgr.Add(manager.RunnableFunc(func(ctx context.Context) error {
mgr.Add(runnableNoLeaderElectionFunc(func(ctx context.Context) error {
<-ctx.Done()
// allow a timeout for graceful shutdown
@ -110,7 +110,7 @@ func Run(opts *config.CAInjectorConfiguration, ctx context.Context) error {
return server.Shutdown(shutdownCtx)
}))
mgr.Add(manager.RunnableFunc(func(ctx context.Context) error {
mgr.Add(runnableNoLeaderElectionFunc(func(ctx context.Context) error {
if err := server.Serve(pprofListener); err != http.ErrServerClosed {
return err
}
@ -172,3 +172,19 @@ func Run(opts *config.CAInjectorConfiguration, ctx context.Context) error {
return nil
}
type runnableNoLeaderElectionFunc func(context.Context) error
func (r runnableNoLeaderElectionFunc) Start(ctx context.Context) error {
return r(ctx)
}
func (runnableNoLeaderElectionFunc) NeedLeaderElection() bool {
// By default, a runnable in c/r is leader election aware.
// Since we need to run this runnable for all replicas, this runnable must NOT be leader election aware.
return false
}
var _ manager.Runnable = runnableNoLeaderElectionFunc(nil)
var _ manager.LeaderElectionRunnable = runnableNoLeaderElectionFunc(nil)