👔 Update answer if the poll answered

This commit is contained in:
2025-02-13 22:19:37 +08:00
parent e16201a3ad
commit a2633e6494
3 changed files with 24 additions and 18 deletions

View File

@ -25,8 +25,14 @@ func AddPollAnswer(poll models.Poll, answer models.PollAnswer) (models.PollAnswe
answer.PollID = poll.ID
var count int64
if err := database.C.Model(&models.PollAnswer{}).Where("poll_id = ? AND account_id = ?", poll.ID, answer.AccountID).Count(&count).Error; err != nil {
return answer, fmt.Errorf("you already answered the poll")
if err := database.C.Model(&models.PollAnswer{}).
Where("poll_id = ? AND account_id = ?", poll.ID, answer.AccountID).
Count(&count).Error; err != nil {
if err := database.C.Model(&models.PollAnswer{}).Where("poll_id = ? AND account_id = ?", poll.ID, answer.AccountID).Update("answer", answer.Answer).Error; err != nil {
return answer, fmt.Errorf("failed to update your answer")
}
return answer, nil
}
if err := database.C.Create(&answer).Error; err != nil {
return answer, err