mastodon/app/models/follow_suggestion.rb

49 lines
1.3 KiB
Ruby
Raw Normal View History

class FollowSuggestion
2016-10-23 09:56:04 +00:00
class << self
def get(for_account_id, limit = 10)
neo = Neography::Rest.new
2016-10-23 09:56:04 +00:00
query = <<END
MATCH (a {account_id: {id}})-[:follows]->(b)-[:follows]->(c)
WHERE a <> c
AND NOT (a)-[:follows]->(c)
2016-10-15 15:19:18 +00:00
RETURN DISTINCT c.account_id, count(b), c.nodeRank
ORDER BY count(b) DESC, c.nodeRank DESC
LIMIT {limit}
END
2016-10-23 09:56:04 +00:00
results = neo.execute_query(query, id: for_account_id, limit: limit)
2016-10-23 09:56:04 +00:00
if results.empty? || results['data'].empty?
results = fallback(for_account_id, limit)
elsif results['data'].size < limit
results['data'] = (results['data'] + fallback(for_account_id, limit - results['data'].size)['data']).uniq
end
2016-10-23 09:56:04 +00:00
account_ids = results['data'].map(&:first)
blocked_ids = Block.where(account_id: for_account_id).pluck(:target_account_id)
accounts_map = Account.where(id: account_ids - blocked_ids).with_counters.map { |a| [a.id, a] }.to_h
2016-10-23 09:56:04 +00:00
account_ids.map { |id| accounts_map[id] }.compact
rescue Neography::NeographyError, Excon::Error::Socket => e
Rails.logger.error e
return []
end
2016-10-23 09:56:04 +00:00
private
2016-10-23 09:56:04 +00:00
def fallback(for_account_id, limit)
neo = Neography::Rest.new
2016-10-23 09:56:04 +00:00
query = <<END
MATCH (b)
RETURN b.account_id
ORDER BY b.nodeRank DESC
LIMIT {limit}
END
2016-10-23 09:56:04 +00:00
neo.execute_query(query, id: for_account_id, limit: limit)
end
end
end