diff --git a/engine/storage_mysql.go b/engine/storage_mysql.go index c49c88378..6e84dc10a 100644 --- a/engine/storage_mysql.go +++ b/engine/storage_mysql.go @@ -27,7 +27,7 @@ import ( ) type MySQLStorage struct { - *SQLStorage + SQLStorage } func NewMySQLStorage(host, port, name, user, password string, maxConn, maxIdleConn int) (*SQLStorage, error) { @@ -43,6 +43,8 @@ func NewMySQLStorage(host, port, name, user, password string, maxConn, maxIdleCo db.DB().SetMaxOpenConns(maxConn) //db.LogMode(true) mySQLStorage := new(MySQLStorage) + mySQLStorage.db = db + mySQLStorage.Db = db.DB() return &SQLStorage{db.DB(), db, mySQLStorage, mySQLStorage}, nil } diff --git a/engine/storage_postgres.go b/engine/storage_postgres.go index 0ab686960..4dbe81a65 100644 --- a/engine/storage_postgres.go +++ b/engine/storage_postgres.go @@ -40,11 +40,13 @@ func NewPostgresStorage(host, port, name, user, password string, maxConn, maxIdl db.DB().SetMaxOpenConns(maxConn) //db.LogMode(true) postgressStorage := new(PostgresStorage) + postgressStorage.db = db + postgressStorage.Db = db.DB() return &SQLStorage{db.DB(), db, postgressStorage, postgressStorage}, nil } type PostgresStorage struct { - *SQLStorage + SQLStorage } func (self *PostgresStorage) SetVersions(vrs Versions, overwrite bool) (err error) {