Remove Repository and use Database instead
This commit is contained in:
@ -12,18 +12,26 @@ import (
|
||||
//go:embed schema/*.sql
|
||||
var migrations embed.FS
|
||||
|
||||
type Database struct {
|
||||
*Queries
|
||||
*sql.DB
|
||||
}
|
||||
|
||||
// Connect to a database
|
||||
func Connect(server string, user string, password string, database string) (*Queries, *sql.DB, error) {
|
||||
func Connect(server string, user string, password string, database string) (*Database, error) {
|
||||
connString := fmt.Sprintf("postgres://%s:%s@%s/%s", user, password, server, database)
|
||||
conn, err := sql.Open("pgx", connString)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
return nil, err
|
||||
}
|
||||
|
||||
goose.SetBaseFS(migrations)
|
||||
if err = goose.Up(conn, "schema"); err != nil {
|
||||
return nil, nil, err
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return New(conn), conn, nil
|
||||
return &Database{
|
||||
New(conn),
|
||||
conn,
|
||||
}, nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user