Fix context cancelling without sending COMM_QUIT for authenticated connections

Author: minhquang4334

connection.go

@@ -40,6 +40,7 @@ type mysqlConn struct {
 
 	// for context support (Go 1.8+)
 	watching bool
+	authed   bool
 	watcher  chan<- context.Context
 	closech  chan struct{}
 	finished chan<- struct{}
@@ -436,9 +437,15 @@ func (mc *mysqlConn) getSystemVar(name string) ([]byte, error) {
 }
 
 // cancel is called when the query has canceled.
-func (mc *mysqlConn) cancel(err error) {
+func (mc *mysqlConn) cancel(err error) error {
 	mc.canceled.Set(err)
+	if mc.authed {
+		return mc.Close()
+	}
+
 	mc.cleanup()
+
+	return nil
 }
 
 // finish is called when the query has succeeded.
@@ -624,7 +631,9 @@ func (mc *mysqlConn) startWatcher() {
 
 			select {
 			case <-ctx.Done():
-				mc.cancel(ctx.Err())
+				if err := mc.cancel(ctx.Err()); err != nil {
+					mc.log(err)
+				}
 			case <-finished:
 			case <-mc.closech:
 				return

connector.go

@@ -169,6 +169,7 @@ func (c *connector) Connect(ctx context.Context) (driver.Conn, error) {
 		mc.cleanup()
 		return nil, err
 	}
+	mc.authed = true
 
 	if mc.cfg.MaxAllowedPacket > 0 {
 		mc.maxAllowedPacket = mc.cfg.MaxAllowedPacket