Rewrite query timeout error

8 months ago · 83f6c501fd
parent 8e4c9c88d4
commit 83f6c501fd
1 changed files with 1 additions and 0 deletions
--- a/db/db.go
+++ b/db/db.go
@ -818,6 +818,7 @@ func (db *DB) queryStmtWithConn(ctx context.Context, stmt *command.Statement, xT
 	rs, err := q.QueryContext(ctx, stmt.Sql, parameters...)
 	if err != nil {
 		stats.Add(numQueryErrors, 1)
+		err = rewriteContextTimeout(err, ErrQueryTimeout)
 		rows.Error = err.Error()
 		return rows, nil
 	}