Better pagination for ActivityPub outbox (#7356)

This commit is contained in:
Eugen Rochko 2018-05-04 19:19:11 +02:00 committed by GitHub
parent d181aad033
commit 2c1f7b2ece
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 50 additions and 12 deletions

View File

@ -22,7 +22,7 @@ class ActivityPub::CollectionsController < Api::BaseController
end end
def set_statuses def set_statuses
@statuses = scope_for_collection.paginate_by_max_id(20, params[:max_id], params[:since_id]) @statuses = scope_for_collection
@statuses = cache_collection(@statuses, Status) @statuses = cache_collection(@statuses, Status)
end end

View File

@ -1,14 +1,14 @@
# frozen_string_literal: true # frozen_string_literal: true
class ActivityPub::OutboxesController < Api::BaseController class ActivityPub::OutboxesController < Api::BaseController
LIMIT = 20
include SignatureVerification include SignatureVerification
before_action :set_account before_action :set_account
before_action :set_statuses
def show def show
@statuses = @account.statuses.permitted_for(@account, signed_request_account).paginate_by_max_id(20, params[:max_id], params[:since_id])
@statuses = cache_collection(@statuses, Status)
render json: outbox_presenter, serializer: ActivityPub::OutboxSerializer, adapter: ActivityPub::Adapter, content_type: 'application/activity+json' render json: outbox_presenter, serializer: ActivityPub::OutboxSerializer, adapter: ActivityPub::Adapter, content_type: 'application/activity+json'
end end
@ -19,11 +19,47 @@ class ActivityPub::OutboxesController < Api::BaseController
end end
def outbox_presenter def outbox_presenter
if page_requested?
ActivityPub::CollectionPresenter.new(
id: account_outbox_url(@account, page_params),
type: :ordered,
part_of: account_outbox_url(@account),
prev: prev_page,
next: next_page,
items: @statuses
)
else
ActivityPub::CollectionPresenter.new( ActivityPub::CollectionPresenter.new(
id: account_outbox_url(@account), id: account_outbox_url(@account),
type: :ordered, type: :ordered,
size: @account.statuses_count, size: @account.statuses_count,
items: @statuses first: account_outbox_url(@account, page: true),
last: account_outbox_url(@account, page: true, min_id: 0)
) )
end end
end
def next_page
account_outbox_url(@account, page: true, max_id: @statuses.last.id) if @statuses.size == LIMIT
end
def prev_page
account_outbox_url(@account, page: true, min_id: @statuses.first.id) unless @statuses.empty?
end
def set_statuses
return unless page_requested?
@statuses = @account.statuses.permitted_for(@account, signed_request_account)
@statuses = params[:min_id].present? ? @statuses.paginate_by_min_id(LIMIT, params[:min_id]).reverse : @statuses.paginate_by_max_id(LIMIT, params[:max_id])
@statuses = cache_collection(@statuses, Status)
end
def page_requested?
params[:page] == 'true'
end
def page_params
{ page: true, max_id: params[:max_id], min_id: params[:min_id] }.compact
end
end end

View File

@ -1,5 +1,5 @@
# frozen_string_literal: true # frozen_string_literal: true
class ActivityPub::CollectionPresenter < ActiveModelSerializers::Model class ActivityPub::CollectionPresenter < ActiveModelSerializers::Model
attributes :id, :type, :size, :items, :part_of, :first, :next, :prev attributes :id, :type, :size, :items, :part_of, :first, :last, :next, :prev
end end

View File

@ -7,12 +7,14 @@ class ActivityPub::CollectionSerializer < ActiveModel::Serializer
super super
end end
attributes :id, :type, :total_items attributes :id, :type
attribute :total_items, if: -> { object.size.present? }
attribute :next, if: -> { object.next.present? } attribute :next, if: -> { object.next.present? }
attribute :prev, if: -> { object.prev.present? } attribute :prev, if: -> { object.prev.present? }
attribute :part_of, if: -> { object.part_of.present? } attribute :part_of, if: -> { object.part_of.present? }
has_one :first, if: -> { object.first.present? } has_one :first, if: -> { object.first.present? }
has_one :last, if: -> { object.last.present? }
has_many :items, key: :items, if: -> { (!object.items.nil? || page?) && !ordered? } has_many :items, key: :items, if: -> { (!object.items.nil? || page?) && !ordered? }
has_many :items, key: :ordered_items, if: -> { (!object.items.nil? || page?) && ordered? } has_many :items, key: :ordered_items, if: -> { (!object.items.nil? || page?) && ordered? }