From 757814d8451ab934da91c0ba99187829ac435e0a Mon Sep 17 00:00:00 2001 From: Lunny Xiao Date: Tue, 2 Feb 2021 10:53:00 +0800 Subject: [PATCH] Add NewEngineWithDB --- engine.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/engine.go b/engine.go index 873fcdc1..f52148aa 100644 --- a/engine.go +++ b/engine.go @@ -105,6 +105,15 @@ func NewEngineWithParams(driverName string, dataSourceName string, params map[st return engine, err } +// NewEngineWithDB new a db manager with db. The params will be passed to db. +func NewEngineWithDB(driverName string, dataSourceName string, db *core.DB) (*Engine, error) { + dialect, err := dialects.OpenDialect(driverName, dataSourceName) + if err != nil { + return nil, err + } + return newEngine(driverName, dataSourceName, dialect, db) +} + // NewEngineWithDialectAndDB new a db manager according to the parameter. // If you do not want to use your own dialect or db, please use NewEngine. // For creating dialect, you can call dialects.OpenDialect. And, for creating db,