kill queries on timeout

connection.go

@@ -48,6 +48,27 @@ type mysqlConn struct {
 	finished chan<- struct{}
 	canceled atomicError // set non-nil if conn is canceled
 	closed   atomicBool  // set when conn is closed, before closech is closed
+
+	// for killing query after timeout
+	id int
+	d  MySQLDriver
+}
+
+func (mc *mysqlConn) kill() error {
+	t := 50 * time.Millisecond
+	killCfg := *mc.cfg
+	killCfg.Timeout = t
+	killCfg.ReadTimeout = t
+	killCfg.WriteTimeout = t
+
+	conn, err := mc.d.Open(killCfg.FormatDSN())
+	if err != nil {
+		return err
+	}
+	defer conn.Close()
+	query := "KILL QUERY " + strconv.Itoa(mc.id)
+	_, err = conn.(*mysqlConn).Exec(query, []driver.Value{})
+	return err
 }
 
 // Handles parameters set in DSN after the connection is established
@@ -445,6 +466,10 @@ func (mc *mysqlConn) getSystemVar(name string) ([]byte, error) {
 // finish is called when the query has canceled.
 func (mc *mysqlConn) cancel(err error) {
 	mc.canceled.Set(err)
+	if mc.cfg.KillQueryOnTimeout {
+		// do not put kill to cleanup to prevent cyclic kills
+		mc.kill()
+	}
 	mc.cleanup()
 }
 

driver.go

@@ -150,6 +150,7 @@ func (d MySQLDriver) Open(dsn string) (driver.Conn, error) {
 		return nil, err
 	}
 
+	mc.d = d
 	return mc, nil
 }