Merge branch 'improve_home_feed_query' into 'develop'

Improve home feed query

See merge request gab/social/gab-social!68
This commit is contained in:
admin 2021-03-01 01:58:36 +00:00
commit 74b3700f58
1 changed files with 36 additions and 43 deletions

View File

@ -1,6 +1,41 @@
# frozen_string_literal: true # frozen_string_literal: true
class HomeFeed < Feed class HomeFeed < Feed
QUERY = <<-SQL
with cte as
(
select
row_number() over (partition by sid.reblog_of_id order by sid.id desc) as rn_dupe,
sid.*
FROM
(select
s.id,
s.reblog_of_id
from statuses s
left join statuses r
on s.reblog_of_id = r.id
where
s.created_at > NOW() - INTERVAL '7 days'
and s.reply is false
and (exists(select ff.target_account_id from follows ff where ff.account_id = :id and ff.target_account_id = s.account_id)
or s.account_id = :id)
and not exists(select mm.target_account_id from mutes mm where mm.account_id = :id and mm.target_account_id in (s.account_id, r.account_id))
and not exists(select bb.target_account_id from blocks bb where bb.account_id = :id and bb.target_account_id in (s.account_id, r.account_id))
and (:max_id is null or s.id < :max_id)
and (:min_id is null or s.id > :min_id)
order by s.created_at desc
limit :limit
) sid
)
select
so.*
from cte
inner join statuses so on cte.id = so.id
where
cte.rn_dupe = 1 or cte.reblog_of_id is null
order by so.created_at desc
SQL
def initialize(account) def initialize(account)
@type = :home @type = :home
@id = account.id @id = account.id
@ -9,50 +44,8 @@ class HomeFeed < Feed
def get(limit = 20, max_id = nil, since_id = nil, min_id = nil) def get(limit = 20, max_id = nil, since_id = nil, min_id = nil)
ActiveRecord::Base.connected_to(role: :reading) do ActiveRecord::Base.connected_to(role: :reading) do
from_database(limit, max_id, since_id, min_id) Status.find_by_sql([QUERY, { id: @id, limit: limit, min_id: min_id, max_id: max_id }])
end end
end end
private
def from_database(limit, max_id, since_id, min_id)
pagination_max = ""
pagination_min = ""
pagination_max = "and s.id < #{max_id}" unless max_id.nil?
pagination_min = "and s.id > #{min_id}" unless min_id.nil?
Status.find_by_sql "
with cte as
(
select
row_number() over (partition by sid.reblog_of_id order by sid.id desc) as rn_dupe,
sid.*
FROM
(select
s.id,
s.reblog_of_id
from statuses s
left join statuses r
on s.reblog_of_id = r.id
where
s.created_at > NOW() - INTERVAL '7 days'
and s.reply is false
and (exists(select ff.target_account_id from follows ff where ff.account_id = #{@id} and ff.target_account_id = s.account_id)
or s.account_id = #{@id})
and not exists(select mm.target_account_id from mutes mm where mm.account_id = #{@id} and mm.target_account_id in (s.account_id, r.account_id))
and not exists(select bb.target_account_id from blocks bb where bb.account_id = #{@id} and bb.target_account_id in (s.account_id, r.account_id))
#{pagination_max}
#{pagination_min}
order by s.created_at desc
limit #{limit}
) sid
)
select
so.*
from cte
inner join statuses so on cte.id = so.id
where
cte.rn_dupe = 1 or cte.reblog_of_id is null
order by so.created_at desc
"
end
end end