fix(util): fix deadlock

Signed-off-by: linzhecheng <linzhecheng@bytedance.com>
kubewharf · Jun 11, 2024 · ddfa0c4 · ddfa0c4
1 parent eb4d284
commit ddfa0c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/util/asyncworker/async_limited_workers.go b/pkg/util/asyncworker/async_limited_workers.go
@@ -109,12 +109,12 @@ func (alw *AsyncLimitedWorkers) AddWork(work *Work, policy DuplicateWorkPolicy)
 
 func (alw *AsyncLimitedWorkers) poll(stopCh <-chan struct{}) (context.Context, *Work, error) {
 	alw.workLock.Lock()
+	defer alw.workLock.Unlock()
 	work, err := alw.doPoll(stopCh)
 	if err != nil {
 		return nil, nil, err
 	}
 	ctx := alw.prepareWork(work)
-	alw.workLock.Unlock()
 	return ctx, work, nil
 }