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

feat: add SQLConnector

Browse Source
This commit is contained in:
Lev Zakharov 2023-05-31 14:35:32 +03:00
parent 740f2be453
commit 8c92f70527
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.(SQLConnector); ok && connector != nil {
return connector.GetSQLConn(db)
}
if dbConnector, ok := connPool.(GetDBConnector); ok && dbConnector != nil { if dbConnector, ok := connPool.(GetDBConnector); ok && dbConnector != nil {
return dbConnector.GetDBConn() return dbConnector.GetDBConn()
} }

6
interfaces.go
View File

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