expunge: decrement user favorite counts.
This commit is contained in:
@@ -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}")}
|
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
|
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:)
|
def self.add(post:, user:)
|
||||||
Favorite.transaction do
|
Favorite.transaction do
|
||||||
User.where(:id => user.id).select("id").lock("FOR UPDATE NOWAIT").first
|
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
|
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?
|
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.where(:id => post_id).update_all("fav_count = fav_count - 1")
|
||||||
post.delete_user_from_fav_string(user.id) if post
|
post.delete_user_from_fav_string(user.id) if post
|
||||||
User.where(:id => user.id).update_all("favorite_count = favorite_count - 1")
|
User.where(:id => user.id).update_all("favorite_count = favorite_count - 1")
|
||||||
|
|||||||
@@ -1038,7 +1038,9 @@ class Post < ApplicationRecord
|
|||||||
end
|
end
|
||||||
|
|
||||||
def remove_from_favorites
|
def remove_from_favorites
|
||||||
Favorite.destroy_all(post_id: self.id)
|
favorites.find_each do |fav|
|
||||||
|
remove_favorite!(fav.user)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
def remove_from_fav_groups
|
def remove_from_fav_groups
|
||||||
|
|||||||
@@ -80,6 +80,13 @@ class PostTest < ActiveSupport::TestCase
|
|||||||
end
|
end
|
||||||
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
|
should "remove the post from iqdb" do
|
||||||
mock_iqdb_service!
|
mock_iqdb_service!
|
||||||
Post.iqdb_sqs_service.expects(:send_message).with("remove\n#{@post.id}")
|
Post.iqdb_sqs_service.expects(:send_message).with("remove\n#{@post.id}")
|
||||||
|
|||||||
Reference in New Issue
Block a user