mirror of https://github.com/k3s-io/k3s
141 lines
3.9 KiB
Go
141 lines
3.9 KiB
Go
/*
|
|
Copyright 2014 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package config
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"k8s.io/apimachinery/pkg/util/wait"
|
|
)
|
|
|
|
type Merger interface {
|
|
// Invoked when a change from a source is received. May also function as an incremental
|
|
// merger if you wish to consume changes incrementally. Must be reentrant when more than
|
|
// one source is defined.
|
|
Merge(source string, update interface{}) error
|
|
}
|
|
|
|
// MergeFunc implements the Merger interface
|
|
type MergeFunc func(source string, update interface{}) error
|
|
|
|
func (f MergeFunc) Merge(source string, update interface{}) error {
|
|
return f(source, update)
|
|
}
|
|
|
|
// Mux is a class for merging configuration from multiple sources. Changes are
|
|
// pushed via channels and sent to the merge function.
|
|
type Mux struct {
|
|
// Invoked when an update is sent to a source.
|
|
merger Merger
|
|
|
|
// Sources and their lock.
|
|
sourceLock sync.RWMutex
|
|
// Maps source names to channels
|
|
sources map[string]chan interface{}
|
|
}
|
|
|
|
// NewMux creates a new mux that can merge changes from multiple sources.
|
|
func NewMux(merger Merger) *Mux {
|
|
mux := &Mux{
|
|
sources: make(map[string]chan interface{}),
|
|
merger: merger,
|
|
}
|
|
return mux
|
|
}
|
|
|
|
// Channel returns a channel where a configuration source
|
|
// can send updates of new configurations. Multiple calls with the same
|
|
// source will return the same channel. This allows change and state based sources
|
|
// to use the same channel. Different source names however will be treated as a
|
|
// union.
|
|
func (m *Mux) Channel(source string) chan interface{} {
|
|
if len(source) == 0 {
|
|
panic("Channel given an empty name")
|
|
}
|
|
m.sourceLock.Lock()
|
|
defer m.sourceLock.Unlock()
|
|
channel, exists := m.sources[source]
|
|
if exists {
|
|
return channel
|
|
}
|
|
newChannel := make(chan interface{})
|
|
m.sources[source] = newChannel
|
|
go wait.Until(func() { m.listen(source, newChannel) }, 0, wait.NeverStop)
|
|
return newChannel
|
|
}
|
|
|
|
func (m *Mux) listen(source string, listenChannel <-chan interface{}) {
|
|
for update := range listenChannel {
|
|
m.merger.Merge(source, update)
|
|
}
|
|
}
|
|
|
|
// Accessor is an interface for retrieving the current merge state.
|
|
type Accessor interface {
|
|
// MergedState returns a representation of the current merge state.
|
|
// Must be reentrant when more than one source is defined.
|
|
MergedState() interface{}
|
|
}
|
|
|
|
// AccessorFunc implements the Accessor interface.
|
|
type AccessorFunc func() interface{}
|
|
|
|
func (f AccessorFunc) MergedState() interface{} {
|
|
return f()
|
|
}
|
|
|
|
type Listener interface {
|
|
// OnUpdate is invoked when a change is made to an object.
|
|
OnUpdate(instance interface{})
|
|
}
|
|
|
|
// ListenerFunc receives a representation of the change or object.
|
|
type ListenerFunc func(instance interface{})
|
|
|
|
func (f ListenerFunc) OnUpdate(instance interface{}) {
|
|
f(instance)
|
|
}
|
|
|
|
type Broadcaster struct {
|
|
// Listeners for changes and their lock.
|
|
listenerLock sync.RWMutex
|
|
listeners []Listener
|
|
}
|
|
|
|
// NewBroadcaster registers a set of listeners that support the Listener interface
|
|
// and notifies them all on changes.
|
|
func NewBroadcaster() *Broadcaster {
|
|
return &Broadcaster{}
|
|
}
|
|
|
|
// Add registers listener to receive updates of changes.
|
|
func (b *Broadcaster) Add(listener Listener) {
|
|
b.listenerLock.Lock()
|
|
defer b.listenerLock.Unlock()
|
|
b.listeners = append(b.listeners, listener)
|
|
}
|
|
|
|
// Notify notifies all listeners.
|
|
func (b *Broadcaster) Notify(instance interface{}) {
|
|
b.listenerLock.RLock()
|
|
listeners := b.listeners
|
|
b.listenerLock.RUnlock()
|
|
for _, listener := range listeners {
|
|
listener.OnUpdate(instance)
|
|
}
|
|
}
|