diff --git a/budget.go b/budget.go index c0084a3..4bf8a22 100644 --- a/budget.go +++ b/budget.go @@ -20,5 +20,5 @@ type UserBudget struct { // BudgetService provides Methods for CRUD of Budgets type BudgetService interface { Budget(id string) (Budget, error) - BudgetsForUser(user User) ([]Budget, error) + BudgetsForUser(id string) ([]Budget, error) } diff --git a/postgres/budgetservice.go b/postgres/budgetservice.go index d3d8d5b..3da9cb3 100644 --- a/postgres/budgetservice.go +++ b/postgres/budgetservice.go @@ -12,7 +12,10 @@ func (s *Repository) Budget(id string) (*budgeteer.Budget, error) { return b, nil } -func (s *Repository) BudgetsForUser(user budgeteer.User) ([]*budgeteer.Budget, error) { +func (s *Repository) BudgetsForUser(id string) ([]*budgeteer.Budget, error) { + user := &budgeteer.User{ + ID: id, + } err := s.DB.Model(&user). Column("user.*", "Budgets"). Select() diff --git a/token.go b/token.go index b5eea8c..8f40ba9 100644 --- a/token.go +++ b/token.go @@ -5,6 +5,7 @@ type Token interface { GetUsername() string GetName() string GetExpiry() float64 + GetID() string } // TokenVerifier verifies a Token