Add endpoint to create allowance (#34)
Closes #11 Reviewed-on: #34
This commit was merged in pull request #34.
This commit is contained in:
@@ -3,6 +3,7 @@ package main
|
||||
import (
|
||||
"errors"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"gitea.seeseepuff.be/seeseemelk/mysqlite"
|
||||
)
|
||||
@@ -45,11 +46,11 @@ func (db *Db) GetUsers() ([]User, error) {
|
||||
return users, nil
|
||||
}
|
||||
|
||||
func (db *Db) GetUser(id int) (*User, error) {
|
||||
user := &User{}
|
||||
func (db *Db) GetUser(id int) (*UserWithAllowance, error) {
|
||||
user := &UserWithAllowance{}
|
||||
|
||||
err := db.db.Query("select id, name from users where id = ?").
|
||||
Bind(id).ScanSingle(&user.ID, &user.Name)
|
||||
err := db.db.Query("select u.id, u.name, sum(h.amount) from users u join history h on h.user_id = u.id where u.id = ?").
|
||||
Bind(id).ScanSingle(&user.ID, &user.Name, &user.Allowance)
|
||||
if errors.Is(err, mysqlite.ErrNoRows) {
|
||||
return nil, nil
|
||||
}
|
||||
@@ -237,3 +238,19 @@ func (db *Db) UpdateTask(id int, task *CreateTaskRequest) error {
|
||||
}
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
func (db *Db) AddAllowance(userId int, allowance *PostAllowance) error {
|
||||
tx, err := db.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.MustRollback()
|
||||
|
||||
err = tx.Query("insert into history (user_id, date, amount) values (?, ?, ?)").
|
||||
Bind(userId, time.Now().Unix(), allowance.Allowance).
|
||||
Exec()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user