summaryrefslogtreecommitdiff
path: root/lib/throttled_api_request.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/throttled_api_request.rb')
-rw-r--r--lib/throttled_api_request.rb25
1 files changed, 25 insertions, 0 deletions
diff --git a/lib/throttled_api_request.rb b/lib/throttled_api_request.rb
new file mode 100644
index 0000000..3f30c56
--- /dev/null
+++ b/lib/throttled_api_request.rb
@@ -0,0 +1,25 @@
+class ThrottledApiRequest < Struct.new(:api_name, :unit_time, :requests_per)
+ def before(job)
+ loop do
+ sleep_for = -1
+ ActiveRecord::Base.transaction do
+ ApiRequests.create(:api_name => self.api_name)
+ recent_requests = ApiRequets.
+ where(:api_name => self.api_name).
+ where("updated_at > ?", Time.now.utc - self.unit_time).
+ order(:updated_at)
+ if (recent_requests.count > self.requests_per)
+ sleep_for = Time.now.utc - recent_requests[recent_requests.count-self.requests_per].updated_at
+ raise ActiveRecord::Rollback
+ else
+ sleep_for = -1
+ end
+ end
+ if sleep_for != -1
+ sleep(sleep_for)
+ else
+ break
+ end
+ end
+ end
+end