Skip to content
Snippets Groups Projects
Select Git revision
  • 7aa2e062013ab02889d18de8d22186444a39cb39
  • master default protected
  • 1.31
  • 4.26.0
  • 4.25.5
  • 4.25.4
  • 4.25.3
  • 4.25.2
  • 4.25.1
  • 4.25.0
  • 4.24.3
  • 4.24.2
  • 4.24.1
  • 4.24.0
  • 4.23.6
  • 4.23.5
  • 4.23.4
  • 4.23.3
  • 4.23.2
  • 4.23.1
  • 4.23.0
  • 4.22.3
  • 4.22.2
23 results

tests.js

Blame
  • driver.go 3.18 KiB
    // Copyright 2012 The Go-MySQL-Driver Authors. All rights reserved.
    //
    // This Source Code Form is subject to the terms of the Mozilla Public
    // License, v. 2.0. If a copy of the MPL was not distributed with this file,
    // You can obtain one at http://mozilla.org/MPL/2.0/.
    
    // Package mysql provides a MySQL driver for Go's database/sql package.
    //
    // The driver should be used via the database/sql package:
    //
    //	import "database/sql"
    //	import _ "github.com/go-sql-driver/mysql"
    //
    //	db, err := sql.Open("mysql", "user:password@/dbname")
    //
    // See https://github.com/go-sql-driver/mysql#usage for details
    package mysql
    
    import (
    	"context"
    	"database/sql"
    	"database/sql/driver"
    	"net"
    	"sync"
    )
    
    // MySQLDriver is exported to make the driver directly accessible.
    // In general the driver is used via the database/sql package.
    type MySQLDriver struct{}
    
    // DialFunc is a function which can be used to establish the network connection.
    // Custom dial functions must be registered with RegisterDial
    //
    // Deprecated: users should register a DialContextFunc instead
    type DialFunc func(addr string) (net.Conn, error)
    
    // DialContextFunc is a function which can be used to establish the network connection.
    // Custom dial functions must be registered with RegisterDialContext
    type DialContextFunc func(ctx context.Context, addr string) (net.Conn, error)
    
    var (
    	dialsLock sync.RWMutex
    	dials     map[string]DialContextFunc
    )
    
    // RegisterDialContext registers a custom dial function. It can then be used by the
    // network address mynet(addr), where mynet is the registered new network.
    // The current context for the connection and its address is passed to the dial function.
    func RegisterDialContext(net string, dial DialContextFunc) {
    	dialsLock.Lock()
    	defer dialsLock.Unlock()
    	if dials == nil {
    		dials = make(map[string]DialContextFunc)
    	}
    	dials[net] = dial
    }
    
    // RegisterDial registers a custom dial function. It can then be used by the
    // network address mynet(addr), where mynet is the registered new network.
    // addr is passed as a parameter to the dial function.
    //
    // Deprecated: users should call RegisterDialContext instead
    func RegisterDial(network string, dial DialFunc) {
    	RegisterDialContext(network, func(_ context.Context, addr string) (net.Conn, error) {
    		return dial(addr)
    	})
    }
    
    // Open new Connection.
    // See https://github.com/go-sql-driver/mysql#dsn-data-source-name for how
    // the DSN string is formatted
    func (d MySQLDriver) Open(dsn string) (driver.Conn, error) {
    	cfg, err := ParseDSN(dsn)
    	if err != nil {
    		return nil, err
    	}
    	c := &connector{
    		cfg: cfg,
    	}
    	return c.Connect(context.Background())
    }
    
    func init() {
    	sql.Register("mysql", &MySQLDriver{})
    }
    
    // NewConnector returns new driver.Connector.
    func NewConnector(cfg *Config) (driver.Connector, error) {
    	cfg = cfg.Clone()
    	// normalize the contents of cfg so calls to NewConnector have the same
    	// behavior as MySQLDriver.OpenConnector
    	if err := cfg.normalize(); err != nil {
    		return nil, err
    	}
    	return &connector{cfg: cfg}, nil
    }
    
    // OpenConnector implements driver.DriverContext.
    func (d MySQLDriver) OpenConnector(dsn string) (driver.Connector, error) {
    	cfg, err := ParseDSN(dsn)
    	if err != nil {
    		return nil, err
    	}
    	return &connector{
    		cfg: cfg,
    	}, nil
    }