mirror of
https://github.com/glitch-soc/mastodon.git
synced 2024-11-24 09:04:00 -05:00
5db3a14388
* Rework tests * Add tests * Fix HomeFeed#get with min_id fetching from database * Minor code cleanup and optimizations * Add tests * Take DMs into account when fetching home TL from database * Fix not listing own DMs in Home timeline * Add tests * Please CodeClimate
63 lines
2.1 KiB
Ruby
63 lines
2.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class HomeFeed < Feed
|
|
def initialize(account)
|
|
@account = account
|
|
super(:home, account.id)
|
|
end
|
|
|
|
def regenerating?
|
|
redis.exists?("account:#{@account.id}:regeneration")
|
|
end
|
|
|
|
def get(limit, max_id = nil, since_id = nil, min_id = nil)
|
|
limit = limit.to_i
|
|
max_id = max_id.to_i if max_id.present?
|
|
since_id = since_id.to_i if since_id.present?
|
|
min_id = min_id.to_i if min_id.present?
|
|
|
|
if min_id.present?
|
|
redis_min_id = fetch_min_redis_id
|
|
return from_redis(limit, max_id, since_id, min_id) if redis_min_id && min_id >= redis_min_id
|
|
|
|
statuses = from_database(limit, redis_min_id, since_id, min_id)
|
|
return statuses if statuses.size >= limit
|
|
|
|
remaining_limit = limit - statuses.size
|
|
min_id = statuses.first.id unless statuses.empty?
|
|
from_redis(remaining_limit, max_id, since_id, min_id) + statuses
|
|
else
|
|
statuses = from_redis(limit, max_id, since_id, min_id)
|
|
return statuses if statuses.size >= limit
|
|
|
|
if since_id.present?
|
|
redis_min_id = fetch_min_redis_id
|
|
return statuses if redis_min_id.present? && since_id >= redis_min_id
|
|
end
|
|
|
|
remaining_limit = limit - statuses.size
|
|
max_id = statuses.last.id unless statuses.empty?
|
|
statuses + from_database(remaining_limit, max_id, since_id, min_id)
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
def from_database(limit, max_id, since_id, min_id)
|
|
scope = Status.where(account: @account.following)
|
|
scope = scope.left_outer_joins(:mentions)
|
|
scope = scope.where(visibility: %i(public unlisted private)).or(scope.where(mentions: { account_id: @account.id })).group(Status.arel_table[:id])
|
|
scope = scope.or(Status.where(account: @account))
|
|
scope
|
|
.to_a_paginated_by_id(limit, min_id: min_id, max_id: max_id, since_id: since_id)
|
|
.reject { |status| FeedManager.instance.filter?(:home, status, @account) }
|
|
.sort_by { |status| -status.id }
|
|
end
|
|
|
|
private
|
|
|
|
def fetch_min_redis_id
|
|
redis.zrangebyscore(key, '(0', '(+inf', limit: [0, 1], with_scores: true).first&.first&.to_i
|
|
end
|
|
end
|