You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
package sqlx
|
|
|
|
import (
|
|
"database/sql"
|
|
"io"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/zeromicro/go-zero/core/syncx"
|
|
)
|
|
|
|
const (
|
|
maxIdleConns = 64
|
|
maxOpenConns = 64
|
|
maxLifetime = time.Minute
|
|
)
|
|
|
|
var connManager = syncx.NewResourceManager()
|
|
|
|
type pingedDB struct {
|
|
*sql.DB
|
|
once sync.Once
|
|
}
|
|
|
|
func getCachedSqlConn(driverName, server string) (*pingedDB, error) {
|
|
val, err := connManager.GetResource(server, func() (io.Closer, error) {
|
|
conn, err := newDBConnection(driverName, server)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &pingedDB{
|
|
DB: conn,
|
|
}, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return val.(*pingedDB), nil
|
|
}
|
|
|
|
func getSqlConn(driverName, server string) (*sql.DB, error) {
|
|
pdb, err := getCachedSqlConn(driverName, server)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
pdb.once.Do(func() {
|
|
err = pdb.Ping()
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return pdb.DB, nil
|
|
}
|
|
|
|
func newDBConnection(driverName, datasource string) (*sql.DB, error) {
|
|
conn, err := sql.Open(driverName, datasource)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// we need to do this until the issue https://github.com/golang/go/issues/9851 get fixed
|
|
// discussed here https://github.com/go-sql-driver/mysql/issues/257
|
|
// if the discussed SetMaxIdleTimeout methods added, we'll change this behavior
|
|
// 8 means we can't have more than 8 goroutines to concurrently access the same database.
|
|
conn.SetMaxIdleConns(maxIdleConns)
|
|
conn.SetMaxOpenConns(maxOpenConns)
|
|
conn.SetConnMaxLifetime(maxLifetime)
|
|
|
|
return conn, nil
|
|
}
|