Class: A2A::Server::Middleware::LimitPaginationSize

Inherits:
Object
  • Object
show all
Defined in:
lib/a2a/server/middleware/limit_pagination_size.rb

Overview

Clamps request.page_size to a valid range and sets env["a2a.page_size"] for the agent.

Accepts a single integer — the maximum page size (also used as the default when the client doesn't specify one). Clamps to [1, max].

Part of the A2A::Server middleware stack — the max comes from A2A::Server.new(page_size: 50).

Instance Method Summary collapse

Constructor Details

#initialize(app, max = 100) ⇒ LimitPaginationSize

Returns a new instance of LimitPaginationSize.



19
20
21
22
# File 'lib/a2a/server/middleware/limit_pagination_size.rb', line 19

def initialize(app, max = 100)
  @app = app
  @max = max
end

Instance Method Details

#call(env) ⇒ Object



24
25
26
27
28
29
30
31
32
33
34
35
36
# File 'lib/a2a/server/middleware/limit_pagination_size.rb', line 24

def call(env)
  request = env["a2a.request"]

  page_size = @max
  if request.respond_to?(:page_size) && request.page_size
    ps = request.page_size.to_i
    page_size = [[ps, 1].max, @max].min
  end

  env["a2a.page_size"] = page_size

  @app.call(env)
end