initial import

core/stores/sqlx/sqlmanager.go

@@ -0,0 +1,74 @@
+package sqlx
+
+import (
+	"database/sql"
+	"io"
+	"sync"
+	"time"
+
+	"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
+}