51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
package object
|
|
|
|
import (
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/client-go/tools/cache"
|
|
)
|
|
|
|
// NewIndexerInformer is a copy of the cache.NewIndexInformer function, but allows Process to have a conversion function (ToFunc).
|
|
func NewIndexerInformer(lw cache.ListerWatcher, objType runtime.Object, h cache.ResourceEventHandler, indexers cache.Indexers, convert ToFunc) (cache.Indexer, cache.Controller) {
|
|
clientState := cache.NewIndexer(cache.DeletionHandlingMetaNamespaceKeyFunc, indexers)
|
|
|
|
fifo := cache.NewDeltaFIFO(cache.MetaNamespaceKeyFunc, clientState)
|
|
|
|
cfg := &cache.Config{
|
|
Queue: fifo,
|
|
ListerWatcher: lw,
|
|
ObjectType: objType,
|
|
FullResyncPeriod: defaultResyncPeriod,
|
|
RetryOnError: false,
|
|
Process: func(obj interface{}) error {
|
|
for _, d := range obj.(cache.Deltas) {
|
|
|
|
obj := convert(d.Object)
|
|
|
|
switch d.Type {
|
|
case cache.Sync, cache.Added, cache.Updated:
|
|
if old, exists, err := clientState.Get(obj); err == nil && exists {
|
|
if err := clientState.Update(obj); err != nil {
|
|
return err
|
|
}
|
|
h.OnUpdate(old, obj)
|
|
} else {
|
|
if err := clientState.Add(obj); err != nil {
|
|
return err
|
|
}
|
|
h.OnAdd(obj)
|
|
}
|
|
case cache.Deleted:
|
|
if err := clientState.Delete(obj); err != nil {
|
|
return err
|
|
}
|
|
h.OnDelete(obj)
|
|
}
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
return clientState, cache.New(cfg)
|
|
}
|
|
|
|
const defaultResyncPeriod = 0
|