Also return sql.DB to be able to use goose
This commit is contained in:
@ -1,13 +1,17 @@
|
||||
package postgres
|
||||
|
||||
import "database/sql"
|
||||
|
||||
// Repository represents a PostgreSQL implementation of all ModelServices
|
||||
type Repository struct {
|
||||
DB *Queries
|
||||
DB *Queries
|
||||
LegacyDB *sql.DB
|
||||
}
|
||||
|
||||
func NewRepository(queries *Queries) (*Repository, error) {
|
||||
func NewRepository(queries *Queries, db *sql.DB) (*Repository, error) {
|
||||
repo := &Repository{
|
||||
DB: queries,
|
||||
DB: queries,
|
||||
LegacyDB: db,
|
||||
}
|
||||
return repo, nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user