2023-06-22 14:14:53 +02:00
|
|
|
package informers
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
|
|
|
|
"github.com/go-logr/logr"
|
|
|
|
"k8s.io/client-go/tools/cache"
|
|
|
|
)
|
|
|
|
|
|
|
|
type informer interface {
|
|
|
|
Informer() cache.SharedIndexInformer
|
|
|
|
}
|
|
|
|
|
|
|
|
func StartInformers(ctx context.Context, informers ...informer) {
|
|
|
|
for i := range informers {
|
|
|
|
go func(informer cache.SharedIndexInformer) {
|
|
|
|
informer.Run(ctx.Done())
|
|
|
|
}(informers[i].Informer())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func WaitForCacheSync(ctx context.Context, logger logr.Logger, informers ...informer) bool {
|
2024-05-20 17:16:35 +08:00
|
|
|
cacheSyncs := make([]cache.InformerSynced, 0, len(informers))
|
2023-06-22 14:14:53 +02:00
|
|
|
for i := range informers {
|
|
|
|
cacheSyncs = append(cacheSyncs, informers[i].Informer().HasSynced)
|
|
|
|
}
|
|
|
|
return cache.WaitForCacheSync(ctx.Done(), cacheSyncs...)
|
|
|
|
}
|
|
|
|
|
|
|
|
func StartInformersAndWaitForCacheSync(ctx context.Context, logger logr.Logger, informers ...informer) bool {
|
|
|
|
StartInformers(ctx, informers...)
|
|
|
|
return WaitForCacheSync(ctx, logger, informers...)
|
|
|
|
}
|