Refactor code

This commit is contained in:
wenxu12345
2021-12-29 10:52:52 +08:00
parent 348ce56776
commit f0c5b1e544
2 changed files with 4 additions and 4 deletions
@@ -65,11 +65,10 @@ func UpdateFriendApplication(friendRequest *FriendRequest) error {
return err
}
friendRequest.CreateTime = time.Now()
err = dbConn.Table("friend_request").Where("from_user_id=? and to_user_id=?", friendRequest.FromUserID, friendRequest.ToUserID).Update(&friendRequest).Error
if err != nil {
return err
if dbConn.Table("friend_request").Where("from_user_id=? and to_user_id=?",
friendRequest.FromUserID, friendRequest.ToUserID).Update(&friendRequest).RowsAffected == 0 {
return InsertFriendApplication(friendRequest)
}
return nil
}
func InsertFriendApplication(friendRequest *FriendRequest) error {