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

refactor: Detecting nil in dbConnector.GetDBConn()

Browse Source
This commit is contained in:
KantaaHasegawa 2023-06-03 21:13:36 +09:00
parent 6444df8eee
commit 224c57f8f3
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gorm.go
View File

@ -376,7 +376,9 @@ func (db *DB) DB() (*sql.DB, error) {
connPool := db.ConnPool connPool := db.ConnPool
if dbConnector, ok := connPool.(GetDBConnector); ok && dbConnector != nil { if dbConnector, ok := connPool.(GetDBConnector); ok && dbConnector != nil {
return dbConnector.GetDBConn() if sqldb, err := dbConnector.GetDBConn(); sqldb != nil && err != nil {
return sqldb, err
}
} }
if sqldb, ok := connPool.(*sql.DB); ok && sqldb != nil { if sqldb, ok := connPool.(*sql.DB); ok && sqldb != nil {