Use id to load budgets

This commit is contained in:
Jan Bader 2016-12-27 00:27:52 +01:00
parent 72fbec1063
commit f36fe1b1c0
3 changed files with 6 additions and 2 deletions

View File

@ -20,5 +20,5 @@ type UserBudget struct {
// BudgetService provides Methods for CRUD of Budgets // BudgetService provides Methods for CRUD of Budgets
type BudgetService interface { type BudgetService interface {
Budget(id string) (Budget, error) Budget(id string) (Budget, error)
BudgetsForUser(user User) ([]Budget, error) BudgetsForUser(id string) ([]Budget, error)
} }

View File

@ -12,7 +12,10 @@ func (s *Repository) Budget(id string) (*budgeteer.Budget, error) {
return b, nil 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). err := s.DB.Model(&user).
Column("user.*", "Budgets"). Column("user.*", "Budgets").
Select() Select()

View File

@ -5,6 +5,7 @@ type Token interface {
GetUsername() string GetUsername() string
GetName() string GetName() string
GetExpiry() float64 GetExpiry() float64
GetID() string
} }
// TokenVerifier verifies a Token // TokenVerifier verifies a Token