forked from AkkomaGang/akkoma
Use atomic update for note count and follower count
This commit is contained in:
parent
9b63fda9c7
commit
af0039a3a0
1 changed files with 56 additions and 31 deletions
|
@ -662,23 +662,43 @@ def get_follow_requests(%User{} = user) do
|
||||||
end
|
end
|
||||||
|
|
||||||
def increase_note_count(%User{} = user) do
|
def increase_note_count(%User{} = user) do
|
||||||
info_cng = User.Info.add_to_note_count(user.info, 1)
|
User
|
||||||
|
|> where(id: ^user.id)
|
||||||
cng =
|
|> update([u],
|
||||||
change(user)
|
set: [
|
||||||
|> put_embed(:info, info_cng)
|
info:
|
||||||
|
fragment(
|
||||||
update_and_set_cache(cng)
|
"jsonb_set(?, '{note_count}', ((?->>'note_count')::int + 1)::varchar::jsonb, true)",
|
||||||
|
u.info,
|
||||||
|
u.info
|
||||||
|
)
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|> Repo.update_all([], returning: true)
|
||||||
|
|> case do
|
||||||
|
{1, [user]} -> set_cache(user)
|
||||||
|
_ -> {:error, user}
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
def decrease_note_count(%User{} = user) do
|
def decrease_note_count(%User{} = user) do
|
||||||
info_cng = User.Info.add_to_note_count(user.info, -1)
|
User
|
||||||
|
|> where(id: ^user.id)
|
||||||
cng =
|
|> update([u],
|
||||||
change(user)
|
set: [
|
||||||
|> put_embed(:info, info_cng)
|
info:
|
||||||
|
fragment(
|
||||||
update_and_set_cache(cng)
|
"jsonb_set(?, '{note_count}', (greatest(0, (?->>'note_count')::int - 1))::varchar::jsonb, true)",
|
||||||
|
u.info,
|
||||||
|
u.info
|
||||||
|
)
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|> Repo.update_all([], returning: true)
|
||||||
|
|> case do
|
||||||
|
{1, [user]} -> set_cache(user)
|
||||||
|
_ -> {:error, user}
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
def update_note_count(%User{} = user) do
|
def update_note_count(%User{} = user) do
|
||||||
|
@ -702,24 +722,29 @@ def update_note_count(%User{} = user) do
|
||||||
|
|
||||||
def update_follower_count(%User{} = user) do
|
def update_follower_count(%User{} = user) do
|
||||||
follower_count_query =
|
follower_count_query =
|
||||||
from(
|
User
|
||||||
u in User,
|
|> where([u], ^user.follower_address in u.following)
|
||||||
where: ^user.follower_address in u.following,
|
|> where([u], u.id != ^user.id)
|
||||||
where: u.id != ^user.id,
|
|> select([u], %{count: count(u.id)})
|
||||||
select: count(u.id)
|
|
||||||
|
User
|
||||||
|
|> where(id: ^user.id)
|
||||||
|
|> join(:inner, [u], s in subquery(follower_count_query))
|
||||||
|
|> update([u, s],
|
||||||
|
set: [
|
||||||
|
info:
|
||||||
|
fragment(
|
||||||
|
"jsonb_set(?, '{follower_count}', ?::varchar::jsonb, true)",
|
||||||
|
u.info,
|
||||||
|
s.count
|
||||||
)
|
)
|
||||||
|
]
|
||||||
follower_count = Repo.one(follower_count_query)
|
)
|
||||||
|
|> Repo.update_all([], returning: true)
|
||||||
info_cng =
|
|> case do
|
||||||
user.info
|
{1, [user]} -> set_cache(user)
|
||||||
|> User.Info.set_follower_count(follower_count)
|
_ -> {:error, user}
|
||||||
|
end
|
||||||
cng =
|
|
||||||
change(user)
|
|
||||||
|> put_embed(:info, info_cng)
|
|
||||||
|
|
||||||
update_and_set_cache(cng)
|
|
||||||
end
|
end
|
||||||
|
|
||||||
def get_users_from_set_query(ap_ids, false) do
|
def get_users_from_set_query(ap_ids, false) do
|
||||||
|
|
Loading…
Reference in a new issue