3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-08-04 09:19:39 +02:00
Code Issues Projects Releases Packages Wiki Activity

zero/connect: add re-run health checks command (#5219)

Browse source 
* zero/connect: add run health checks and shutdown commands

* fix proto

* trigger re-run on command

* add handler

* rename runPeriodicHealthChecksLeased
This commit is contained in:
Denis Mishin 2024-08-22 16:17:53 -04:00 • committed by GitHub
parent 6e766233c7
commit 0503b41108
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 220 additions and 104 deletions
Download patch file Download diff file Expand all files Collapse all files

15
internal/zero/healthcheck/syncer.go
View file

@ -9,12 +9,12 @@ import (
"github.com/pomerium/pomerium/pkg/protoutil"
)
func (c *checker) ConfigSyncer(ctx context.Context) error {
func (c *Checker) ConfigSyncer(ctx context.Context) error {
syncer := databroker.NewSyncer("zero-health-check", c, databroker.WithTypeURL(protoutil.GetTypeURL(new(configpb.Config))))
return syncer.Run(ctx)
}
func (c *checker) GetConfigs() []*configpb.Config {
func (c *Checker) GetConfigs() []*configpb.Config {
configs := c.configs.Load()
if configs == nil {
return nil
@ -23,12 +23,12 @@ func (c *checker) GetConfigs() []*configpb.Config {
}
// ClearRecords implements databroker.Syncer interface
func (c *checker) ClearRecords(_ context.Context) {
func (c *Checker) ClearRecords(_ context.Context) {
c.configs.Store([]*configpb.Config{})
}
// UpdateRecords implements databroker.Syncer interface
func (c *checker) UpdateRecords(_ context.Context, _ uint64, records []*databroker.Record) {
func (c *Checker) UpdateRecords(_ context.Context, _ uint64, records []*databroker.Record) {
if len(records) == 0 {
return
}
@ -39,13 +39,10 @@ func (c *checker) UpdateRecords(_ context.Context, _ uint64, records []*databrok
return
}
c.configs.Store(cfgs)
select {
case c.forceCheck <- struct{}{}:
default:
}
c.ForceCheck()
}
// GetDataBrokerServiceClient implements databroker.Syncer interface
func (c *checker) GetDataBrokerServiceClient() databroker.DataBrokerServiceClient {
func (c *Checker) GetDataBrokerServiceClient() databroker.DataBrokerServiceClient {
return c.databrokerClient
}