blob: 570105e2a05f136edb7452a21c13179b9c5d29ab (
plain)
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
|
# frozen_string_literal: true
require_relative "compact_index"
Artifice.deactivate
class CompactIndexRateLimited < CompactIndexAPI
class RequestCounter
def self.queue
@queue ||= Thread::Queue.new
end
def self.size
@queue.size
end
def self.enq(name)
@queue.enq(name)
end
def self.deq
@queue.deq
end
end
configure do
RequestCounter.queue
end
get "/info/:name" do
RequestCounter.enq(params[:name])
begin
if RequestCounter.size == 1
etag_response do
gem = gems.find {|g| g.name == params[:name] }
CompactIndex.info(gem ? gem.versions : [])
end
else
status 429
end
ensure
RequestCounter.deq
end
end
end
Artifice.activate_with(CompactIndexRateLimited)
|
