mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
101 lines
3 KiB
Go
101 lines
3 KiB
Go
package informers
|
|
|
|
import (
|
|
"reflect"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/docker/cli/kubernetes/client/clientset"
|
|
"github.com/docker/cli/kubernetes/client/informers/compose"
|
|
"github.com/docker/cli/kubernetes/client/informers/internalinterfaces"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/client-go/tools/cache"
|
|
)
|
|
|
|
type sharedInformerFactory struct {
|
|
client clientset.Interface
|
|
lock sync.Mutex
|
|
defaultResync time.Duration
|
|
|
|
informers map[reflect.Type]cache.SharedIndexInformer
|
|
// startedInformers is used for tracking which informers have been started.
|
|
// This allows Start() to be called multiple times safely.
|
|
startedInformers map[reflect.Type]bool
|
|
}
|
|
|
|
// NewSharedInformerFactory constructs a new instance of sharedInformerFactory
|
|
func NewSharedInformerFactory(client clientset.Interface, defaultResync time.Duration) SharedInformerFactory {
|
|
return &sharedInformerFactory{
|
|
client: client,
|
|
defaultResync: defaultResync,
|
|
informers: make(map[reflect.Type]cache.SharedIndexInformer),
|
|
startedInformers: make(map[reflect.Type]bool),
|
|
}
|
|
}
|
|
|
|
// Start initializes all requested informers.
|
|
func (f *sharedInformerFactory) Start(stopCh <-chan struct{}) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
|
|
for informerType, informer := range f.informers {
|
|
if !f.startedInformers[informerType] {
|
|
go informer.Run(stopCh)
|
|
f.startedInformers[informerType] = true
|
|
}
|
|
}
|
|
}
|
|
|
|
// WaitForCacheSync waits for all started informers' cache were synced.
|
|
func (f *sharedInformerFactory) WaitForCacheSync(stopCh <-chan struct{}) map[reflect.Type]bool {
|
|
informers := func() map[reflect.Type]cache.SharedIndexInformer {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
|
|
informers := map[reflect.Type]cache.SharedIndexInformer{}
|
|
for informerType, informer := range f.informers {
|
|
if f.startedInformers[informerType] {
|
|
informers[informerType] = informer
|
|
}
|
|
}
|
|
return informers
|
|
}()
|
|
|
|
res := map[reflect.Type]bool{}
|
|
for informType, informer := range informers {
|
|
res[informType] = cache.WaitForCacheSync(stopCh, informer.HasSynced)
|
|
}
|
|
return res
|
|
}
|
|
|
|
// InternalInformerFor returns the SharedIndexInformer for obj using an internal
|
|
// client.
|
|
func (f *sharedInformerFactory) InformerFor(obj runtime.Object, newFunc internalinterfaces.NewInformerFunc) cache.SharedIndexInformer {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
|
|
informerType := reflect.TypeOf(obj)
|
|
informer, exists := f.informers[informerType]
|
|
if exists {
|
|
return informer
|
|
}
|
|
informer = newFunc(f.client, f.defaultResync)
|
|
f.informers[informerType] = informer
|
|
|
|
return informer
|
|
}
|
|
|
|
// SharedInformerFactory provides shared informers for resources in all known
|
|
// API group versions.
|
|
type SharedInformerFactory interface {
|
|
internalinterfaces.SharedInformerFactory
|
|
ForResource(resource schema.GroupVersionResource) (GenericInformer, error)
|
|
WaitForCacheSync(stopCh <-chan struct{}) map[reflect.Type]bool
|
|
|
|
Compose() compose.Interface
|
|
}
|
|
|
|
func (f *sharedInformerFactory) Compose() compose.Interface {
|
|
return compose.New(f)
|
|
}
|