mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-06 22:06:03 +02:00
d7ab817de7
* authorize: add databroker server and record version to result, force sync via polling * wrap inmem store to take read lock when grabbing databroker versions * address code review comments * reset max to 0
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/rs/zerolog"
|
|
|
|
"github.com/pomerium/pomerium/pkg/grpc/databroker"
|
|
)
|
|
|
|
type dataBrokerSyncer struct {
|
|
cfg *atomicConfig
|
|
log zerolog.Logger
|
|
|
|
update chan<- updateRecordsMessage
|
|
clear chan<- struct{}
|
|
|
|
syncer *databroker.Syncer
|
|
}
|
|
|
|
func newDataBrokerSyncer(
|
|
cfg *atomicConfig,
|
|
log zerolog.Logger,
|
|
update chan<- updateRecordsMessage,
|
|
clear chan<- struct{},
|
|
) *dataBrokerSyncer {
|
|
syncer := &dataBrokerSyncer{
|
|
cfg: cfg,
|
|
log: log,
|
|
|
|
update: update,
|
|
clear: clear,
|
|
}
|
|
syncer.syncer = databroker.NewSyncer(syncer)
|
|
return syncer
|
|
}
|
|
|
|
func (syncer *dataBrokerSyncer) Run(ctx context.Context) (err error) {
|
|
return syncer.syncer.Run(ctx)
|
|
}
|
|
|
|
func (syncer *dataBrokerSyncer) ClearRecords(ctx context.Context) {
|
|
select {
|
|
case <-ctx.Done():
|
|
case syncer.clear <- struct{}{}:
|
|
}
|
|
}
|
|
|
|
func (syncer *dataBrokerSyncer) GetDataBrokerServiceClient() databroker.DataBrokerServiceClient {
|
|
return syncer.cfg.Load().dataBrokerClient
|
|
}
|
|
|
|
func (syncer *dataBrokerSyncer) UpdateRecords(ctx context.Context, serverVersion uint64, records []*databroker.Record) {
|
|
select {
|
|
case <-ctx.Done():
|
|
case syncer.update <- updateRecordsMessage{records: records}:
|
|
}
|
|
}
|