use defer cancel

pingcap · Mar 4, 2023 · 374802c · 374802c
1 parent 287e259
commit 374802c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cdc/sink/dmlsink/txn/mysql/mysql.go b/cdc/sink/dmlsink/txn/mysql/mysql.go
@@ -615,7 +615,8 @@ func (s *mysqlBackend) multiStmtExecute(
 		}
 		multiStmtArgs = append(multiStmtArgs, dmls.values[i]...)
 	}
-	ctx, cancelFunc := context.WithTimeout(ctx, writeTimeout)
+	ctx, cancel := context.WithTimeout(ctx, writeTimeout)
+	defer cancel()
 	_, execError := tx.ExecContext(ctx, multiStmtSQL, multiStmtArgs...)
 	if execError != nil {
 		err := logDMLTxnErr(
@@ -626,10 +627,8 @@ func (s *mysqlBackend) multiStmtExecute(
 				log.Warn("failed to rollback txn", zap.Error(rbErr))
 			}
 		}
-		cancelFunc()
 		return err
 	}
-	cancelFunc()
 	return nil
 }