func GetExpense(db *sql.DB, id int) (error, *entities.Expense) {
query := "select " +
"category, amount, created_at, spent_at, username " +
"from expenses " +
"where id = $1 " +
"limit 1"
row := db.QueryRow(query, id)
var category string
var amount float32
var createdAt time.Time
var spentAt time.Time
var user string
if err := row.Scan(&category, &amount, &createdAt, &spentAt, &user); err != nil {
return err, nil
}
expense := &entities.Expense{
ID: id,
Category: entities.Category(category),
Amount: entities.Money(amount),
CreatedAt: createdAt,
SpentAt: spentAt,
User: entities.User(user),
}
return nil, expense
}
func GetTotalAmountForUser(db *sql.DB, user entities.User) entities.Money {
username := string(user)
row := db.QueryRow("select sum(amount) from expenses where username = $1", username)
var amount float32
row.Scan(&amount)
return entities.Money(amount)
}
func GetExpenses(db *sql.DB) (error, []entities.Expense) {
expenses := []entities.Expense{}
rows, err := db.Query("select " +
"id, category, amount, created_at, spent_at, username " +
"from expenses " +
"order by spent_at desc, id desc limit 25")
defer rows.Close()
if err != nil {
return err, nil
}
for rows.Next() {
var id int
var category string
var amount float32
var createdAt time.Time
var spentAt time.Time
var user string
if err = rows.Scan(&id, &category, &amount, &createdAt, &spentAt, &user); err != nil {
return err, nil
}
expense := entities.Expense{
ID: id,
Category: entities.Category(category),
Amount: entities.Money(amount),
CreatedAt: createdAt,
SpentAt: spentAt,
User: entities.User(user),
}
expenses = append(expenses, expense)
}
return nil, expenses
}