@@ -526,23 +526,20 @@ func (c *testCase) execParaDDLSQL(taskCh chan *ddlJobTask, num int) error {
526
526
defer wg .Done ()
527
527
opStart := time .Now ()
528
528
db := c .pickupDB ()
529
- err := c .executeWithTimeout (db , task )
530
- if ! ddlIgnoreError (err ) {
531
- log .Infof ("[ddl] [instance %d] TiDB execute %s , err %v, elapsed time:%v" , c .caseIndex , task .sql , err , time .Since (opStart ).Seconds ())
532
- task .err = err
533
- db := c .dbs [0 ]
534
529
conn , err := db .Conn (context .Background ())
530
+ if err != nil {
531
+ unExpectedErr = err
532
+ return
533
+ }
535
534
defer func () {
536
535
err := conn .Close ()
537
536
if err != nil {
538
537
log .Errorf ("error when closes conn %s" , err .Error ())
539
538
}
540
539
}()
541
- if err != nil {
542
- unExpectedErr = err
543
- return
544
- }
545
- _ , ddlErr := conn .ExecContext (context .Background (), task .sql )
540
+ ddlErr := c .executeWithTimeout (task , func (ctx context.Context , s string ) (sql.Result , error ) {
541
+ return conn .ExecContext (ctx , s )
542
+ })
546
543
if ! ddlIgnoreError (ddlErr ) {
547
544
log .Infof ("[ddl] [instance %d] TiDB execute %s , err %v, elapsed time:%v" , c .caseIndex , task .sql , err , time .Since (opStart ).Seconds ())
548
545
task .err = ddlErr
@@ -626,7 +623,9 @@ func (c *testCase) execSerialDDLSQL(taskCh chan *ddlJobTask) error {
626
623
task := <- taskCh
627
624
db := c .pickupDB ()
628
625
opStart := time .Now ()
629
- err := c .executeWithTimeout (db , task )
626
+ err := c .executeWithTimeout (task , func (ctx context.Context , s string ) (sql.Result , error ) {
627
+ return db .ExecContext (ctx , task .sql )
628
+ })
630
629
log .Infof ("[ddl] [instance %d] %s, err: %v, elapsed time:%v" , c .caseIndex , task .sql , err , time .Since (opStart ).Seconds ())
631
630
if err != nil {
632
631
if ddlIgnoreError (err ) {
0 commit comments