| 
									
										
										
										
											2018-10-09 21:56:09 +01:00
										 |  |  | package object
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import (
 | 
					
						
							|  |  |  | 	"k8s.io/apimachinery/pkg/runtime"
 | 
					
						
							|  |  |  | 	"k8s.io/client-go/tools/cache"
 | 
					
						
							|  |  |  | )
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-10-04 17:48:43 +02:00
										 |  |  | // NewIndexerInformer is a copy of the cache.NewIndexerInformer function, but allows custom process function
 | 
					
						
							|  |  |  | func NewIndexerInformer(lw cache.ListerWatcher, objType runtime.Object, h cache.ResourceEventHandler, indexers cache.Indexers, builder ProcessorBuilder) (cache.Indexer, cache.Controller) {
 | 
					
						
							| 
									
										
										
										
											2018-10-09 21:56:09 +01:00
										 |  |  | 	clientState := cache.NewIndexer(cache.DeletionHandlingMetaNamespaceKeyFunc, indexers)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	cfg := &cache.Config{
 | 
					
						
							| 
									
										
										
										
											2019-10-04 17:48:43 +02:00
										 |  |  | 		Queue:            cache.NewDeltaFIFO(cache.MetaNamespaceKeyFunc, clientState),
 | 
					
						
							| 
									
										
										
										
											2018-10-09 21:56:09 +01:00
										 |  |  | 		ListerWatcher:    lw,
 | 
					
						
							|  |  |  | 		ObjectType:       objType,
 | 
					
						
							| 
									
										
										
										
											2019-07-04 01:09:12 +05:30
										 |  |  | 		FullResyncPeriod: defaultResyncPeriod,
 | 
					
						
							| 
									
										
										
										
											2018-10-09 21:56:09 +01:00
										 |  |  | 		RetryOnError:     false,
 | 
					
						
							| 
									
										
										
										
											2019-10-04 17:48:43 +02:00
										 |  |  | 		Process:          builder(clientState, h),
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 	return clientState, cache.New(cfg)
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // DefaultProcessor is a copy of Process function from cache.NewIndexerInformer except it does a conversion.
 | 
					
						
							|  |  |  | func DefaultProcessor(convert ToFunc) ProcessorBuilder {
 | 
					
						
							|  |  |  | 	return func(clientState cache.Indexer, h cache.ResourceEventHandler) cache.ProcessFunc {
 | 
					
						
							|  |  |  | 		return func(obj interface{}) error {
 | 
					
						
							| 
									
										
										
										
											2018-10-09 21:56:09 +01:00
										 |  |  | 			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
 | 
					
						
							| 
									
										
										
										
											2019-10-04 17:48:43 +02:00
										 |  |  | 		}
 | 
					
						
							| 
									
										
										
										
											2018-10-09 21:56:09 +01:00
										 |  |  | 	}
 | 
					
						
							|  |  |  | }
 | 
					
						
							| 
									
										
										
										
											2019-07-04 01:09:12 +05:30
										 |  |  | 
 | 
					
						
							|  |  |  | const defaultResyncPeriod = 0
 |