txn: fix goroutine leak (#784)

Signed-off-by: disksing <[email protected]>
tikv · Apr 27, 2023 · aee2f31 · aee2f31
1 parent c619a50
commit aee2f31
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/txnkv/transaction/txn.go b/txnkv/transaction/txn.go
@@ -1335,7 +1335,9 @@ func (txn *KVTxn) asyncPessimisticRollback(ctx context.Context, keys [][]byte, s
 	}
 	wg := new(sync.WaitGroup)
 	wg.Add(1)
+	txn.store.WaitGroup().Add(1)
 	go func() {
+		defer txn.store.WaitGroup().Done()
 		if val, err := util.EvalFailpoint("beforeAsyncPessimisticRollback"); err == nil {
 			if s, ok := val.(string); ok {
 				if s == "skip" {