45 lines
963 B
Go
45 lines
963 B
Go
package storage
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
)
|
|
|
|
func ArchiveMessage(db *sql.DB, userID, messageID int) error {
|
|
_, err := db.Exec(`
|
|
UPDATE user_messages
|
|
SET is_archived = TRUE, archived_at = CURRENT_TIMESTAMP
|
|
WHERE id = ? AND recipientId = ?
|
|
`, messageID, userID)
|
|
return err
|
|
}
|
|
|
|
func MarkMessageAsRead(db *sql.DB, messageID, userID int) error {
|
|
result, err := db.Exec(`
|
|
UPDATE user_messages
|
|
SET is_read = TRUE
|
|
WHERE id = ? AND recipientId = ?
|
|
`, messageID, userID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rowsAffected, err := result.RowsAffected()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if rowsAffected == 0 {
|
|
return fmt.Errorf("no matching message found for user_id=%d and message_id=%d", userID, messageID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func RestoreMessage(db *sql.DB, userID, messageID int) error {
|
|
_, err := db.Exec(`
|
|
UPDATE user_messages
|
|
SET is_archived = FALSE, archived_at = NULL
|
|
WHERE id = ? AND recipientId = ?
|
|
`, messageID, userID)
|
|
return err
|
|
}
|