tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add *sql.DB connector that uses database context (#6366)

Browse Source 
* feat: add SQLConnector

* rename
This commit is contained in:
Lev Zakharov 2023-06-05 11:25:05 +03:00 committed by GitHub
parent 5eaccaa624
commit 661781a3d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gorm.go
View File

@ -375,6 +375,10 @@ func (db *DB) AddError(err error) error {
func (db *DB) DB() (*sql.DB, error) { func (db *DB) DB() (*sql.DB, error) {
connPool := db.ConnPool connPool := db.ConnPool
if connector, ok := connPool.(GetDBConnectorWithContext); ok && connector != nil {
return connector.GetDBConnWithContext(db)
}
if dbConnector, ok := connPool.(GetDBConnector); ok && dbConnector != nil { if dbConnector, ok := connPool.(GetDBConnector); ok && dbConnector != nil {
if sqldb, err := dbConnector.GetDBConn(); sqldb != nil || err != nil { if sqldb, err := dbConnector.GetDBConn(); sqldb != nil || err != nil {
return sqldb, err return sqldb, err

6
interfaces.go
View File

@ -77,6 +77,12 @@ type GetDBConnector interface {
GetDBConn() (*sql.DB, error) GetDBConn() (*sql.DB, error)
} }
// GetDBConnectorWithContext represents SQL db connector which takes into
// account the current database context
type GetDBConnectorWithContext interface {
GetDBConnWithContext(db *DB) (*sql.DB, error)
}
// Rows rows interface // Rows rows interface
type Rows interface { type Rows interface {
Columns() ([]string, error) Columns() ([]string, error)