Ping reads its result; obeys the context's Done channel

Author: edsrzf

connection_go18.go

@@ -7,5 +7,18 @@ import (
 )
 
 func (mc *mysqlConn) Ping(ctx context.Context) error {
-	return mc.writeCommandPacket(comPing)
+	err := mc.writeCommandPacket(comPing)
+	if err != nil {
+		return err
+	}
+
+	select {
+	case <-ctx.Done():
+		mc.Close()
+		return ctx.Err()
+	default:
+	}
+
+	_, err = mc.readResultOK()
+	return err
 }