3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-06-28 15:38:05 +02:00
Code Issues Projects Releases Packages Wiki Activity

add name

Browse source
This commit is contained in:
Caleb Doxsey 2024-04-23 16:10:32 -06:00
parent 7e36c96884
commit 1f28934a94
2 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
internal/disabler/disabler.go
View file

@ -4,6 +4,8 @@ package disabler
import (
"context"
"errors"
"github.com/pomerium/pomerium/internal/log"
)
var errDisabled = errors.New("disabled")
@ -19,6 +21,7 @@ type Disabler interface {
}
type disabler struct {
name string
handler Handler
onChangeEnabled chan bool
}
@ -27,8 +30,8 @@ type disabler struct {
// RunEnabled will be called. If the Disabler is subsequently disabled the
// context passed to RunEnabled will be canceled. If the Disabler is subseqently
// enabled, RunEnabled will be started again.
func New(handler Handler, enabled bool) Disabler {
d := disabler{handler: handler, onChangeEnabled: make(chan bool, 1)}
func New(name string, handler Handler, enabled bool) Disabler {
d := disabler{name: name, handler: handler, onChangeEnabled: make(chan bool, 1)}
d.change(enabled)
return d
}
@ -47,8 +50,10 @@ func (d disabler) Run(ctx context.Context) error {
continue
}
log.Ctx(ctx).Info().Msgf("enabled %s", d.name)
err := d.runEnabledOnce(ctx)
if errors.Is(err, errDisabled) {
log.Ctx(ctx).Info().Msgf("disabled %s", d.name)
continue
}

2
internal/identity/manager/manager.go
View file

@ -64,7 +64,7 @@ func New(
sessionScheduler: scheduler.New(),
userScheduler: scheduler.New(),
}
mgr.Disabler = disabler.New(mgr, true)
mgr.Disabler = disabler.New("identity_manager", mgr, true)
mgr.reset()
mgr.UpdateConfig(options...)
return mgr