Merge pull request #736 from go-gorm/fix_rep_db

fix: replace db
go-gorm · Jan 4, 2023 · adde295 · adde295
2 parents 8b6f628 + 9cf24c3
commit adde295
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/do.go b/do.go
@@ -64,7 +64,9 @@ func (d *DO) UseDB(db *gorm.DB, opts ...DOOption) {
 }
 
 // ReplaceDB replace db connection
-func (d *DO) ReplaceDB(db *gorm.DB) { d.db = db }
+func (d *DO) ReplaceDB(db *gorm.DB) {
+	d.db = db.Session(&gorm.Session{})
+}
 
 // ReplaceConnPool replace db connection pool
 func (d *DO) ReplaceConnPool(pool gorm.ConnPool) {