Unfloyded/Legamunt
0
0
Fork 0
Code Activity

Improve Account#triadic_closures (#3079)

Browse source
This commit is contained in:
alpaca-tc 2017-05-16 19:06:38 +09:00 • committed by Eugen Rochko
parent 09ec6e504b
commit 682b68438e
2 changed files with 44 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

20
app/models/account.rb
View file

@ -259,24 +259,30 @@ class Account < ApplicationRecord
nil
end
def triadic_closures(account, limit = 5)
def triadic_closures(account, limit: 5, offset: 0)
sql = <<-SQL.squish
WITH first_degree AS (
SELECT target_account_id
FROM follows
WHERE account_id = :account_id
)
SELECT target_account_id
FROM follows
WHERE account_id = :account_id
)
SELECT accounts.*
FROM follows
INNER JOIN accounts ON follows.target_account_id = accounts.id
WHERE account_id IN (SELECT * FROM first_degree) AND target_account_id NOT IN (SELECT * FROM first_degree) AND target_account_id <> :account_id
WHERE
account_id IN (SELECT * FROM first_degree)
AND target_account_id NOT IN (SELECT * FROM first_degree)
AND target_account_id NOT IN (:excluded_account_ids)
GROUP BY target_account_id, accounts.id
ORDER BY count(account_id) DESC
OFFSET :offset
LIMIT :limit
SQL
excluded_account_ids = account.excluded_from_timeline_account_ids + [account.id]
find_by_sql(
[sql, { account_id: account.id, limit: limit }]
[sql, { account_id: account.id, excluded_account_ids: excluded_account_ids, limit: limit, offset: offset }]
)
end