Also return sql.DB to be able to use goose

This commit is contained in:
2021-12-02 20:36:22 +00:00
parent a3df95a700
commit 4011f3cace
3 changed files with 14 additions and 15 deletions

View File

@ -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
}