Use goose for migrations
This commit is contained in:
@ -2,23 +2,19 @@ package postgres
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
|
||||
"git.javil.eu/jacob1123/budgeteer"
|
||||
)
|
||||
|
||||
// Budget returns a budget for a given id.
|
||||
func (s *Repository) Budget(id string) (*budgeteer.Budget, error) {
|
||||
b := &budgeteer.Budget{ID: id}
|
||||
err := s.DB.Select(&b)
|
||||
func (s *Repository) Budget(id string) (*Budget, error) {
|
||||
budget, err := s.DB.GetBudget(context.Background(), id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return b, nil
|
||||
return &budget, nil
|
||||
}
|
||||
|
||||
func (s *Repository) BudgetsForUser(id string) ([]Budget, error) {
|
||||
budgets, err := s.DB.GetBudgetsForUser(context.Background(), sql.NullString{id, true})
|
||||
budgets, err := s.DB.GetBudgetsForUser(context.Background(), id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@ -26,7 +22,7 @@ func (s *Repository) BudgetsForUser(id string) ([]Budget, error) {
|
||||
|
||||
}
|
||||
|
||||
func (s *Repository) NewBudget(name string, userID string) (Budget, error) {
|
||||
func (s *Repository) NewBudget(name string, userID string) (*Budget, error) {
|
||||
b := CreateBudgetParams{ID: s.IDGenerator.New(), Name: name}
|
||||
budget, err := s.DB.CreateBudget(context.Background(), b)
|
||||
if err != nil {
|
||||
@ -39,5 +35,5 @@ func (s *Repository) NewBudget(name string, userID string) (Budget, error) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return budget, nil
|
||||
return &budget, nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user