func (s *postgresStore) UpdateExpense(e *models.Expense, userIDs []int64) error {
if e.ID == 0 {
return models.ErrStructNotSaved
}
eas, err := e.Assign(userIDs)
if err != nil {
return errors.Annotate(err, "Could not assign expense")
}
tx, err := s.db.Beginx()
if err != nil {
return errors.Annotate(err, "Could not create transaction")
}
stmt, err := tx.PrepareNamed(deleteExpenseAssignmentsStr)
if err != nil {
_ = tx.Rollback()
return errors.Annotate(err, "Error preparing delete expense assignments statement")
}
r, err := stmt.Exec(e)
if err != nil {
_ = tx.Rollback()
return errors.Annotate(err, "Error deleting expense assignments")
}
n, _ := r.RowsAffected()
if n == 0 {
_ = tx.Rollback()
return errors.New("expense does not have any associated assignments")
}
stmt, err = tx.PrepareNamed(updateExpenseStr)
if err != nil {
_ = tx.Rollback()
return errors.Annotate(err, "Error preparing update expense statement")
}
r, err = stmt.Exec(e)
if err != nil {
_ = tx.Rollback()
return errors.Annotate(err, "Error updating expense")
}
err = s.insertExpenseAssignments(eas, tx)
if err != nil {
_ = tx.Rollback()
return errors.Trace(err)
}
// updated expense and created new assignments
err = tx.Commit()
if err != nil {
return errors.Annotate(err, "error committing expense update")
}
e.Assignments = eas
return nil
}
func (s *postgresStore) InsertExpense(e *models.Expense, userIDs []int64) error {
// Assign expense and commit everything to the db within the same transaction
if e.ID != 0 {
return models.ErrAlreadySaved
}
tx, err := s.db.Beginx()
if err != nil {
return errors.Annotate(err, "Could not create transaction")
}
stmt, err := tx.PrepareNamed(insertExpeseStr)
if err != nil {
_ = tx.Rollback()
return errors.Annotate(err, "Error preparing insert expense statement")
}
err = stmt.Get(e, e)
if err != nil {
_ = tx.Rollback()
return errors.Annotate(err, "Error inserting expense")
}
eas, err := e.Assign(userIDs)
if err != nil {
_ = tx.Rollback()
return errors.Annotate(err, "Error assigning expense")
}
err = s.insertExpenseAssignments(eas, tx)
if err != nil {
_ = tx.Rollback()
return errors.Trace(err)
}
// Sucessfully inserted expense and assignments.
err = tx.Commit()
if err != nil {
return errors.Annotate(err, "Error committing to database")
}
e.Assignments = eas
return nil
}