Update CreateFollowingRelationships

This commit is contained in:
Egor Kislitsyn 2019-10-08 13:06:48 +07:00
parent 12ebeef130
commit 6291eaa590
1 changed files with 49 additions and 30 deletions

View File

@ -18,7 +18,8 @@ def change 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 @@ def change 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 @@ defp insert_following_relationships_rows 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 @@ defp restore_following_column 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
followers.id AS follower_id, follwer.id AS follower_id,
array_prepend( CASE follwer.local
followers.follower_address, WHEN TRUE THEN
array_agg(DISTINCT following.ap_id) FILTER ( array_prepend(follwer.follower_address, array_agg(following.follower_address))
WHERE ELSE
following.ap_id IS NOT NULL array_agg(following.follower_address)
) END AS following_array
) as following
FROM FROM
users AS followers following_relationships
LEFT OUTER JOIN activities ON ( JOIN users AS follwer ON follwer.id = following_relationships.follower_id
followers.ap_id = activities.actor JOIN users AS FOLLOWING ON following.id = following_relationships.following_id
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 GROUP BY
followers.id, follwer.id) AS following_query
followers.ap_id
) AS following_query
WHERE WHERE
following_query.follower_id = users.id; following_query.follower_id = users.id
""" """
end end
end end