Add informers and internal go routines in manager (#2870)

* used manager's Add function in more places
* exit when ctx.Done is received in archivePruner go routines
* fix manager tests
* fix data race
* added more gpm function in manager and removed manager from a util function
* closed unused channel and stopped ticker after context is done
* added log statements
* used context.Done inside function instead of stopper channel
This commit is contained in:
Vardhaman Surana
2023-11-10 12:42:21 +05:30
committed by GitHub
parent 2a40b4538c
commit 3fabf64b3c
28 changed files with 195 additions and 114 deletions
+13
View File
@@ -4,6 +4,8 @@ import (
"context"
"sync"
"time"
k8sCache "k8s.io/client-go/tools/cache"
)
var _ Interface = &GroupManager{}
@@ -15,6 +17,8 @@ type Interface interface {
// and will also remove the "function" from the list when it completes
Add(ctx context.Context, function func(context.Context))
AddInformers(ctx context.Context, informers map[string]k8sCache.SharedIndexInformer)
// Wait blocks the execution of the process until all the go routines in the manager are completed.
Wait()
@@ -39,6 +43,15 @@ func (g *GroupManager) Add(ctx context.Context, f func(context.Context)) {
}()
}
func (g *GroupManager) AddInformers(ctx context.Context, informers map[string]k8sCache.SharedIndexInformer) {
for _, informer := range informers {
informer := informer
g.Add(ctx, func(ctxArg context.Context) {
informer.Run(ctxArg.Done())
})
}
}
func (g *GroupManager) Wait() {
g.wg.Wait()
}
+2
View File
@@ -53,6 +53,8 @@ func TestAddAndWaitWithTimeout(t *testing.T) {
err := mgr.WaitWithTimeout(1 * time.Second)
require.NotNil(t, err, "manager WaitWithTimeout did not return an error when timeout exceeded")
mgr = New()
expectedValue := 11
mgr.Add(context.Background(), func(ctx context.Context) {
time.Sleep(100 * time.Millisecond)