/
cursor.rb
48 lines (45 loc) · 1.52 KB
/
cursor.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
module Slack
module Web
module Api
module Pagination
class Cursor
include Enumerable
attr_reader :client
attr_reader :verb
attr_reader :sleep_interval
attr_reader :max_retries
attr_reader :params
def initialize(client, verb, params = {})
@client = client
@verb = verb
@sleep_interval = params[:sleep_interval]
@max_retries = params[:max_retries] || client.default_max_retries
@params = params.reject { |k, _| [:sleep_interval, :max_retries].include?(k) }
end
def each
next_cursor = nil
retry_count = 0
loop do
query = { limit: client.default_page_size }.merge(params).merge(cursor: next_cursor)
begin
response = client.send(verb, query)
rescue Slack::Web::Api::Errors::TooManyRequestsError => e
raise e if retry_count >= max_retries
client.logger.debug("#{self.class}##{__method__}") { e.to_s }
retry_count += 1
sleep(e.retry_after.seconds)
next
end
yield response
break unless response.response_metadata
next_cursor = response.response_metadata.next_cursor
break if next_cursor.blank?
retry_count = 0
sleep(sleep_interval) if sleep_interval
end
end
end
end
end
end
end