PonyMirrors/philomena
1
0
Fork 0
mirror of https://github.com/philomena-dev/philomena.git synced 2024-11-27 13:47:58 +01:00
Code Issues Projects Releases Packages Wiki Activity

#44 unwrap tag reversions from transaction

Browse source
This commit is contained in:
byte[] 2020-05-06 10:40:23 -04:00
parent 6f9955bac1
commit b54704fba5
1 changed files with 33 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

73
lib/philomena/tag_changes.ex
View file

@ -44,59 +44,52 @@ defmodule Philomena.TagChanges do
to_add = Enum.map(removed, &%{image_id: &1.image_id, tag_id: &1.tag_id}) to_add = Enum.map(removed, &%{image_id: &1.image_id, tag_id: &1.tag_id})
Repo.transaction(fn -> {_count, inserted} =
{_count, inserted} = Repo.insert_all(Tagging, to_add, on_conflict: :nothing, returning: [:image_id, :tag_id])
Repo.insert_all(Tagging, to_add, on_conflict: :nothing, returning: [:image_id, :tag_id])
{_count, deleted} = Repo.delete_all(to_remove) {_count, deleted} = Repo.delete_all(to_remove)
inserted = Enum.map(inserted, &[&1.image_id, &1.tag_id]) inserted = Enum.map(inserted, &[&1.image_id, &1.tag_id])
added_changes = added_changes =
Enum.map(inserted, fn [image_id, tag_id] -> Enum.map(inserted, fn [image_id, tag_id] ->
Map.merge(tag_change_attributes, %{image_id: image_id, tag_id: tag_id, added: true}) Map.merge(tag_change_attributes, %{image_id: image_id, tag_id: tag_id, added: true})
end) end)
removed_changes = removed_changes =
Enum.map(deleted, fn [image_id, tag_id] -> Enum.map(deleted, fn [image_id, tag_id] ->
Map.merge(tag_change_attributes, %{image_id: image_id, tag_id: tag_id, added: false}) Map.merge(tag_change_attributes, %{image_id: image_id, tag_id: tag_id, added: false})
end) end)
Repo.insert_all(TagChange, added_changes ++ removed_changes) Repo.insert_all(TagChange, added_changes ++ removed_changes)
# In order to merge into the existing tables here in one go, insert_all # In order to merge into the existing tables here in one go, insert_all
# is used with a query that is guaranteed to conflict on every row by # is used with a query that is guaranteed to conflict on every row by
# using the primary key. # using the primary key.
added_upserts = added_upserts =
inserted inserted
|> Enum.group_by(fn [_image_id, tag_id] -> tag_id end) |> Enum.group_by(fn [_image_id, tag_id] -> tag_id end)
|> Enum.map(fn {tag_id, instances} -> |> Enum.map(fn {tag_id, instances} ->
Map.merge(tag_attributes, %{id: tag_id, images_count: length(instances)}) Map.merge(tag_attributes, %{id: tag_id, images_count: length(instances)})
end) end)
removed_upserts = removed_upserts =
deleted deleted
|> Enum.group_by(fn [_image_id, tag_id] -> tag_id end) |> Enum.group_by(fn [_image_id, tag_id] -> tag_id end)
|> Enum.map(fn {tag_id, instances} -> |> Enum.map(fn {tag_id, instances} ->
Map.merge(tag_attributes, %{id: tag_id, images_count: -length(instances)}) Map.merge(tag_attributes, %{id: tag_id, images_count: -length(instances)})
end) end)
update_query = update(Tag, inc: [images_count: fragment("EXCLUDED.images_count")]) update_query = update(Tag, inc: [images_count: fragment("EXCLUDED.images_count")])
upserts = added_upserts ++ removed_upserts upserts = added_upserts ++ removed_upserts
Repo.insert_all(Tag, upserts, on_conflict: update_query, conflict_target: [:id]) Repo.insert_all(Tag, upserts, on_conflict: update_query, conflict_target: [:id])
end)
|> case do
{:ok, _result} ->
Images.reindex_images(image_ids)
{:ok, tag_changes} Images.reindex_images(image_ids)
error -> {:ok, tag_changes}
error
end
end end
@doc """ @doc """