Update CreateFollowingRelationships

This commit is contained in:
Egor Kislitsyn 2019-10-08 13:06:48 +07:00
parent 12ebeef130
commit 6291eaa590

View file

@ -18,7 +18,8 @@ defmodule Pleroma.Repo.Migrations.CreateFollowingRelationships do
create_if_not_exists(index(:following_relationships, :follower_id)) create_if_not_exists(index(:following_relationships, :follower_id))
create_if_not_exists(unique_index(:following_relationships, [:follower_id, :following_id])) create_if_not_exists(unique_index(:following_relationships, [:follower_id, :following_id]))
execute(insert_following_relationships_rows(), restore_following_column()) execute(import_following_from_users(), "")
execute(import_following_from_activities(), restore_following_column())
drop(index(:users, [:following], concurrently: true, using: :gin)) drop(index(:users, [:following], concurrently: true, using: :gin))
@ -27,7 +28,33 @@ defmodule Pleroma.Repo.Migrations.CreateFollowingRelationships do
end end
end end
defp insert_following_relationships_rows do defp import_following_from_users do
"""
INSERT INTO following_relationships (follower_id, following_id, state, inserted_at, updated_at)
SELECT
relations.follower_id,
following.id,
'accept',
now(),
now()
FROM (
SELECT
users.id AS follower_id,
unnest(users.following) AS following_ap_id
FROM
users
WHERE
users.following != '{}'
AND users.local = false OR users.local = true AND users.email IS NOT NULL -- Exclude `internal/fetch` and `relay`
) AS relations
JOIN users AS "following" ON "following".follower_address = relations.following_ap_id
WHERE relations.follower_id != following.id
ON CONFLICT DO NOTHING
"""
end
defp import_following_from_activities do
""" """
INSERT INTO INSERT INTO
following_relationships ( following_relationships (
@ -49,7 +76,9 @@ defmodule Pleroma.Repo.Migrations.CreateFollowingRelationships do
JOIN users AS following ON (activities.data ->> 'object' = following.ap_id) JOIN users AS following ON (activities.data ->> 'object' = following.ap_id)
WHERE WHERE
activities.data ->> 'type' = 'Follow' activities.data ->> 'type' = 'Follow'
AND activities.data ->> 'state' IN ('accept', 'pending', 'reject') ON CONFLICT DO NOTHING AND activities.data ->> 'state' IN ('accept', 'pending', 'reject')
ORDER BY activities.updated_at DESC
ON CONFLICT DO NOTHING
""" """
end end
@ -58,35 +87,25 @@ defmodule Pleroma.Repo.Migrations.CreateFollowingRelationships do
UPDATE UPDATE
users users
SET SET
following = following_query.following, following = following_query.following_array,
updated_at = now() updated_at = now()
FROM FROM (
( SELECT
SELECT follwer.id AS follower_id,
followers.id AS follower_id, CASE follwer.local
array_prepend( WHEN TRUE THEN
followers.follower_address, array_prepend(follwer.follower_address, array_agg(following.follower_address))
array_agg(DISTINCT following.ap_id) FILTER ( ELSE
WHERE array_agg(following.follower_address)
following.ap_id IS NOT NULL END AS following_array
) FROM
) as following following_relationships
FROM JOIN users AS follwer ON follwer.id = following_relationships.follower_id
users AS followers JOIN users AS FOLLOWING ON following.id = following_relationships.following_id
LEFT OUTER JOIN activities ON ( GROUP BY
followers.ap_id = activities.actor follwer.id) AS following_query
AND activities.data ->> 'type' = 'Follow'
AND activities.data ->> 'state' IN ('accept', 'pending', 'reject')
)
LEFT OUTER JOIN users AS following ON (activities.data ->> 'object' = following.ap_id)
WHERE
followers.email IS NOT NULL -- Exclude `internal/fetch` and `relay`
GROUP BY
followers.id,
followers.ap_id
) AS following_query
WHERE WHERE
following_query.follower_id = users.id; following_query.follower_id = users.id
""" """
end end
end end