pomerium/internal/zero/bootstrap/source.go

package bootstrap

import (
	"context"
	"sync"

	"github.com/google/go-cmp/cmp"
	"github.com/google/go-cmp/cmp/cmpopts"

	"github.com/pomerium/pomerium/config"
	"github.com/pomerium/pomerium/internal/atomicutil"
	cluster_api "github.com/pomerium/pomerium/pkg/zero/cluster"
)

var _ = config.Source(new(source))

var cmpOpts = []cmp.Option{
	cmpopts.IgnoreUnexported(config.Options{}),
	cmpopts.EquateEmpty(),
}

type source struct {
	cfg atomicutil.Value[*config.Config]

	listenerLock sync.RWMutex
	listeners    []config.ChangeListener

	ready     chan struct{}
	markReady sync.Once
}

func (src *source) WaitReady(ctx context.Context) error {
	select {
	case <-ctx.Done():
		return ctx.Err()
	case <-src.ready:
		return nil
	}
}

// GetConfig implements config.Source
func (src *source) GetConfig() *config.Config {
	return src.cfg.Load()
}

// OnConfigChange implements config.Source
func (src *source) OnConfigChange(_ context.Context, l config.ChangeListener) {
	src.listenerLock.Lock()
	src.listeners = append(src.listeners, l)
	src.listenerLock.Unlock()
}

// UpdateBootstrap updates the underlying configuration options
func (src *source) UpdateBootstrap(ctx context.Context, cfg cluster_api.BootstrapConfig) bool {
	current := src.cfg.Load()
	incoming := current.Clone()
	applyBootstrapConfig(incoming.Options, &cfg)

	src.markReady.Do(func() { close(src.ready) })

	if cmp.Equal(incoming.Options, current.Options, cmpOpts...) {
		return false
	}

	src.cfg.Store(incoming)

	src.notifyListeners(ctx, incoming)

	return true
}

// notifyListeners notifies all listeners of a configuration change
func (src *source) notifyListeners(ctx context.Context, cfg *config.Config) {
	src.listenerLock.RLock()
	listeners := make([]config.ChangeListener, len(src.listeners))
	copy(listeners, src.listeners)
	src.listenerLock.RUnlock()

	for _, l := range listeners {
		l(ctx, cfg)
	}
}

func applyBootstrapConfig(dst *config.Options, src *cluster_api.BootstrapConfig) {
	if src.DatabrokerStorageConnection != nil {
		dst.DataBrokerStorageType = config.StoragePostgresName
		dst.DataBrokerStorageConnectionString = *src.DatabrokerStorageConnection
	} else {
		dst.DataBrokerStorageType = config.StorageInMemoryName
		dst.DataBrokerStorageConnectionString = ""
	}
}