123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146 |
- package mysql
- import (
- "context"
- "database/sql/driver"
- "net"
- )
- type connector struct {
- cfg *Config
- }
- func (c *connector) Connect(ctx context.Context) (driver.Conn, error) {
- var err error
-
- mc := &mysqlConn{
- maxAllowedPacket: maxPacketSize,
- maxWriteSize: maxPacketSize - 1,
- closech: make(chan struct{}),
- cfg: c.cfg,
- }
- mc.parseTime = mc.cfg.ParseTime
-
- dialsLock.RLock()
- dial, ok := dials[mc.cfg.Net]
- dialsLock.RUnlock()
- if ok {
- dctx := ctx
- if mc.cfg.Timeout > 0 {
- var cancel context.CancelFunc
- dctx, cancel = context.WithTimeout(ctx, c.cfg.Timeout)
- defer cancel()
- }
- mc.netConn, err = dial(dctx, mc.cfg.Addr)
- } else {
- nd := net.Dialer{Timeout: mc.cfg.Timeout}
- mc.netConn, err = nd.DialContext(ctx, mc.cfg.Net, mc.cfg.Addr)
- }
- if err != nil {
- return nil, err
- }
-
- if tc, ok := mc.netConn.(*net.TCPConn); ok {
- if err := tc.SetKeepAlive(true); err != nil {
-
- mc.netConn.Close()
- mc.netConn = nil
- return nil, err
- }
- }
-
- mc.startWatcher()
- if err := mc.watchCancel(ctx); err != nil {
- mc.cleanup()
- return nil, err
- }
- defer mc.finish()
- mc.buf = newBuffer(mc.netConn)
-
- mc.buf.timeout = mc.cfg.ReadTimeout
- mc.writeTimeout = mc.cfg.WriteTimeout
-
- authData, plugin, err := mc.readHandshakePacket()
- if err != nil {
- mc.cleanup()
- return nil, err
- }
- if plugin == "" {
- plugin = defaultAuthPlugin
- }
-
- authResp, err := mc.auth(authData, plugin)
- if err != nil {
-
- errLog.Print("could not use requested auth plugin '"+plugin+"': ", err.Error())
- plugin = defaultAuthPlugin
- authResp, err = mc.auth(authData, plugin)
- if err != nil {
- mc.cleanup()
- return nil, err
- }
- }
- if err = mc.writeHandshakeResponsePacket(authResp, plugin); err != nil {
- mc.cleanup()
- return nil, err
- }
-
- if err = mc.handleAuthResult(authData, plugin); err != nil {
-
-
-
- mc.cleanup()
- return nil, err
- }
- if mc.cfg.MaxAllowedPacket > 0 {
- mc.maxAllowedPacket = mc.cfg.MaxAllowedPacket
- } else {
-
- maxap, err := mc.getSystemVar("max_allowed_packet")
- if err != nil {
- mc.Close()
- return nil, err
- }
- mc.maxAllowedPacket = stringToInt(maxap) - 1
- }
- if mc.maxAllowedPacket < maxPacketSize {
- mc.maxWriteSize = mc.maxAllowedPacket
- }
-
- err = mc.handleParams()
- if err != nil {
- mc.Close()
- return nil, err
- }
- return mc, nil
- }
- func (c *connector) Driver() driver.Driver {
- return &MySQLDriver{}
- }
|