diff --git a/database.go b/database.go index cd3eb52..5f309b8 100644 --- a/database.go +++ b/database.go @@ -16,6 +16,10 @@ func newDatabase(host string, port int, dbName, user, pass string) (dbase Databa dbase.db, err = sql.Open("postgres", dbase.sqlConnString()) return }// }}} +func databaseFromInstance(db *sql.DB) (dbase Database, err error) { + dbase.db = db + return +} func (dbase Database) sqlConnString() string {// {{{ return fmt.Sprintf( diff --git a/upgrader.go b/upgrader.go index 8b20287..6b662c3 100644 --- a/upgrader.go +++ b/upgrader.go @@ -145,6 +145,10 @@ func (upgrader Upgrader) AddDatabase(host string, port int, dbName, user, pass s err = db.verifySchemaEntry() return }// }}} +func (upgrader Upgrader) AddDatabaseInstance(sqlDB *sql.DB) (db Database, err error) {// {{{ + return databaseFromInstance(sqlDB) +}// }}} + // Run executes the actual schema updates until there are no more available. func (upgrader Upgrader) Run() (err error) {// {{{ var version int