expunge: decrement user favorite counts.

This commit is contained in:
evazion
2017-08-03 22:45:10 -05:00
committed by r888888888
parent fd9dc6f647
commit af42740ca9
3 changed files with 11 additions and 15 deletions

View File

@@ -4,19 +4,6 @@ class Favorite < ApplicationRecord
scope :for_user, lambda {|user_id| where("user_id % 100 = #{user_id.to_i % 100} and user_id = #{user_id.to_i}")}
attr_accessible :user_id, :post_id
# this is necessary because there's no trigger for deleting favorites
def self.destroy_all(user_id: nil, post_id: nil)
if user_id && post_id
connection.execute("delete from favorites_#{user_id % 100} where user_id = #{user_id} and post_id = #{post_id}")
elsif user_id
connection.execute("delete from favorites_#{user_id % 100} where user_id = #{user_id}")
elsif post_id
0.upto(99) do |uid|
connection.execute("delete from favorites_#{uid} where post_id = #{post_id}")
end
end
end
def self.add(post:, user:)
Favorite.transaction do
User.where(:id => user.id).select("id").lock("FOR UPDATE NOWAIT").first
@@ -40,7 +27,7 @@ class Favorite < ApplicationRecord
User.where(:id => user.id).select("id").lock("FOR UPDATE NOWAIT").first
return unless Favorite.for_user(user.id).where(:user_id => user.id, :post_id => post_id).exists?
Favorite.destroy_all(user_id: user.id, post_id: post_id)
Favorite.for_user(user.id).delete_all(post_id: post_id)
Post.where(:id => post_id).update_all("fav_count = fav_count - 1")
post.delete_user_from_fav_string(user.id) if post
User.where(:id => user.id).update_all("favorite_count = favorite_count - 1")

View File

@@ -1038,7 +1038,9 @@ class Post < ApplicationRecord
end
def remove_from_favorites
Favorite.destroy_all(post_id: self.id)
favorites.find_each do |fav|
remove_favorite!(fav.user)
end
end
def remove_from_fav_groups

View File

@@ -80,6 +80,13 @@ class PostTest < ActiveSupport::TestCase
end
end
should "decrement the user's favorite count" do
@post.add_favorite!(@post.uploader)
assert_difference(["@post.uploader.reload.favorite_count"], -1) do
@post.expunge!
end
end
should "remove the post from iqdb" do
mock_iqdb_service!
Post.iqdb_sqs_service.expects(:send_message).with("remove\n#{@post.id}")