fix: avoid coroutine leaks when the dialecter initialization fails. (#…

…6249) Co-authored-by: Kevin Lin <[email protected]>
go-gorm · Apr 21, 2023 · 32fc201 · 32fc201
1 parent ac20d9e
commit 32fc201
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/gorm.go b/gorm.go
@@ -179,6 +179,12 @@ func Open(dialector Dialector, opts ...Option) (db *DB, err error) {
 
 	if config.Dialector != nil {
 		err = config.Dialector.Initialize(db)
+
+		if err != nil {
+			if db, err := db.DB(); err == nil {
+				_ = db.Close()
+			}
+		}
 	}
 
 	preparedStmt := &PreparedStmtDB{