From 723d2d4772ab4bca67a5e50f47fd4f862bed809f Mon Sep 17 00:00:00 2001 From: "Alexander \"Arav\" Andreev" Date: Thu, 23 Mar 2023 17:03:23 +0400 Subject: [PATCH] Make DeleteReply() return count of affected rows. --- database/sqlite/database.go | 12 ++++++------ guestbook/database.go | 4 ++-- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/database/sqlite/database.go b/database/sqlite/database.go index 4b3cf42..2e4cc70 100644 --- a/database/sqlite/database.go +++ b/database/sqlite/database.go @@ -279,26 +279,26 @@ func (d *SQLiteDatabase) UpdateReply(reply *guestbook.Reply) error { return nil } -func (d *SQLiteDatabase) DeleteReply(entryID int64) error { +func (d *SQLiteDatabase) DeleteReply(entryID int64) (int64, error) { tx, err := d.db.Begin() if err != nil { - return err + return -1, err } defer tx.Rollback() res, err := tx.Stmt(stmtDeleteReply).Exec(entryID) if err != nil { - return err + return -1, err } - _, err = res.RowsAffected() + ra, err := res.RowsAffected() if err != nil { - return err + return -1, err } tx.Commit() - return nil + return ra, nil } func (d *SQLiteDatabase) Close() error { diff --git a/guestbook/database.go b/guestbook/database.go index b9592bd..505f5a7 100644 --- a/guestbook/database.go +++ b/guestbook/database.go @@ -7,9 +7,9 @@ type Guestbook interface { Count() (int64, error) NewEntry(entry *Entry) error UpdateEntry(entry *Entry) error - DeleteEntry(entryID int64) error + DeleteEntry(entryID int64) (int64, error) NewReply(reply *Reply) error UpdateReply(reply *Reply) error - DeleteReply(entryID int64) error + DeleteReply(entryID int64) (int64, error) Close() error }