diff options
Diffstat (limited to 'server/guest/store.go')
-rw-r--r-- | server/guest/store.go | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/server/guest/store.go b/server/guest/store.go index a5b9374..4290f8f 100644 --- a/server/guest/store.go +++ b/server/guest/store.go @@ -157,7 +157,7 @@ func (store Store) Update(guest Guest) error { if err := store.updateGuest(guest); err != nil { return err } - if err := store.deleteOldParty(guest.ID); err != nil { + if err := store.deleteParty(guest.ID); err != nil { return err } return store.insertParty(guest) @@ -171,7 +171,7 @@ func (store Store) updateGuest(guest Guest) error { return err } -func (store Store) deleteOldParty(guestID int) error { +func (store Store) deleteParty(guestID int) error { statement := "delete from party where guest_id = $1" _, err := store.database.Exec(context.Background(), statement, guestID) return err @@ -189,3 +189,19 @@ func (store Store) insertParty(guest Guest) error { } return nil } + +func (store Store) Delete(guestID int) error { + if err := store.deleteGuest(guestID); err != nil { + return err + } + if err := store.deleteParty(guestID); err != nil { + return err + } + return nil +} + +func (store Store) deleteGuest(guestID int) error { + statement := "delete from guest where id = $1" + _, err := store.database.Exec(context.Background(), statement, guestID) + return err +} |