rancher/norman
1
0
Fork 0
mirror of https://github.com/rancher/norman.git synced 2025-09-01 15:18:20 +00:00
Code Issues Releases Wiki Activity

Add ability to add handlers transactionally across controllers

Browse Source
This commit is contained in:
Darren Shepherd
2020-02-05 20:47:25 -07:00
parent 9dd0f76a7e
commit 899cfdf49b
2 changed files with 61 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
controller/generic_controller.go
View File

@@ -123,6 +123,20 @@ func (g *genericController) Enqueue(namespace, name string) {
}
func (g *genericController) AddHandler(ctx context.Context, name string, handler HandlerFunc) {
t := getHandlerTransaction(ctx)
if t == nil {
g.addHandler(ctx, name, handler)
return
}
go func() {
if t.shouldContinue() {
g.addHandler(ctx, name, handler)
}
}()
}
func (g *genericController) addHandler(ctx context.Context, name string, handler HandlerFunc) {
g.Lock()
defer g.Unlock()